Function handle_memory_tool

src/mcp_cli/memory/tools.py:103–143

Execute a memory tool and return the result as a string.

(
    store: MemoryScopeStore, tool_name: str, arguments: dict
)

async def handle_memory_tool(
    store: MemoryScopeStore, tool_name: str, arguments: dict
) -> str:
    """Execute a memory tool and return the result as a string."""
    try:
        if tool_name == "remember":
            scope = MemoryScope(arguments["scope"])
            entry = store.remember(scope, arguments["key"], arguments["content"])
            return f"Remembered '{entry.key}' in {scope.value} scope."

        if tool_name == "recall":
            scope_str = arguments.get("scope")
            recall_scope: MemoryScope | None = (
                MemoryScope(scope_str) if scope_str else None
            )
            key = arguments.get("key")
            query = arguments.get("query")

            entries = store.recall(scope=recall_scope, key=key, query=query)
            if not entries:
                return "No memories found."

            lines = []
            for e in entries:
                lines.append(f"- [{e.key}]: {e.content}")
            return "\n".join(lines)

        if tool_name == "forget":
            scope = MemoryScope(arguments["scope"])
            removed = store.forget(scope, arguments["key"])
            if removed:
                return f"Forgot '{arguments['key']}' from {scope.value} scope."
            return (
                f"No memory with key '{arguments['key']}' found in {scope.value} scope."
            )

        return f"Unknown memory tool: {tool_name}"

    except Exception as exc:
        logger.warning("Memory tool %s failed: %s", tool_name, exc)
        return f"Memory tool error: {exc}"