hanzo-mcp 0.8.8__py3-none-any.whl → 0.9.0__py3-none-any.whl

hanzo_mcp/tools/mcp/mcp_add.py

@@ -189,9 +189,7 @@ Use 'mcp_stats' to see all added servers and their status.
 
                 full_command = shlex.split(command)
 
-        await tool_ctx.info(
-            f"Adding MCP server '{name}' with command: {' '.join(full_command)}"
-        )
+        await tool_ctx.info(f"Adding MCP server '{name}' with command: {' '.join(full_command)}")
 
         # Create server configuration
         server_config = {
@@ -221,8 +219,8 @@ Use 'mcp_stats' to see all added servers and their status.
                     if not shutil.which("uvx"):
                         return "Error: uvx not found. Install uv first."
 
-                # MCP server connection handled by runtime
-                # Store configuration for runtime to use
+                # TODO: Actually start and connect to the MCP server
+                # For now, we just store the configuration
                 server_config["status"] = "ready"
 
             except Exception as e:

hanzo_mcp/tools/mcp/mcp_remove.py

@@ -103,7 +103,7 @@ Use 'mcp_stats' to see all servers before removing.
             if not force:
                 return f"Error: Server '{name}' is currently running. Use --force to remove anyway."
             else:
-                # Server process stopping handled by MCP runtime
+                # TODO: Stop the server process
                 await tool_ctx.info(f"Stopping running server '{name}'")
 
         # Remove from registry

hanzo_mcp/tools/mcp/mcp_stats.py

@@ -68,9 +68,7 @@ Example:
         servers = McpAddTool.get_servers()
 
         if not servers:
-            return (
-                "No MCP servers have been added yet.\n\nUse 'mcp_add' to add servers."
-            )
+            return "No MCP servers have been added yet.\n\nUse 'mcp_add' to add servers."
 
         output = []
         output.append("=== MCP Server Statistics ===")

hanzo_mcp/tools/mcp/mcp_tool.py

@@ -173,9 +173,7 @@ class MCPTool(BaseTool):
     def _auto_start_servers(self):
         """Auto-start servers configured for auto-start."""
         for name, server_config in self.config.get("servers", {}).items():
-            if server_config.get("enabled", False) and server_config.get(
-                "auto_start", False
-            ):
+            if server_config.get("enabled", False) and server_config.get("auto_start", False):
                 self._start_server(name, server_config)
 
     def _start_server(self, name: str, config: Dict[str, Any]) -> bool:
@@ -202,9 +200,7 @@ class MCPTool(BaseTool):
             cmd = [config["command"]] + config.get("args", [])
 
             # Create log directory
-            log_dir = Path(
-                self.config.get("log_dir", str(Path.home() / ".hanzo" / "mcp" / "logs"))
-            )
+            log_dir = Path(self.config.get("log_dir", str(Path.home() / ".hanzo" / "mcp" / "logs")))
             log_dir.mkdir(parents=True, exist_ok=True)
 
             # Start process
@@ -307,11 +303,11 @@ Status: {enabled} enabled, {running} running"""
         elif action == "restart":
             return self._handle_restart(params.get("name"))
         elif action == "config":
-            return self._handle_config(
-                params.get("config_key"), params.get("config_value")
-            )
+            return self._handle_config(params.get("config_key"), params.get("config_value"))
         else:
-            return f"Error: Unknown action '{action}'. Valid actions: list, add, remove, enable, disable, restart, config"
+            return (
+                f"Error: Unknown action '{action}'. Valid actions: list, add, remove, enable, disable, restart, config"
+            )
 
     def _handle_list(self) -> str:
         """List all MCP servers."""
@@ -355,9 +351,7 @@ Status: {enabled} enabled, {running} running"""
             output.append(f"{name}: {status}")
             if config.get("description"):
                 output.append(f"  Description: {config['description']}")
-            output.append(
-                f"  Command: {config['command']} {' '.join(config.get('args', []))}"
-            )
+            output.append(f"  Command: {config['command']} {' '.join(config.get('args', []))}")
 
             if config.get("env"):
                 env_str = ", ".join([f"{k}={v}" for k, v in config["env"].items()])
@@ -498,11 +492,7 @@ Status: {enabled} enabled, {running} running"""
                 else:
                     return f"Configuration key '{key}' not found"
 
-            return (
-                json.dumps(current, indent=2)
-                if isinstance(current, (dict, list))
-                else str(current)
-            )
+            return json.dumps(current, indent=2) if isinstance(current, (dict, list)) else str(current)
         else:
             # Set value
             # Navigate to parent
@@ -513,11 +503,7 @@ Status: {enabled} enabled, {running} running"""
                 current = current[k]
 
             # Parse value if it looks like JSON
-            if (
-                isinstance(value, str)
-                and value.startswith("{")
-                or value.startswith("[")
-            ):
+            if isinstance(value, str) and value.startswith("{") or value.startswith("["):
                 try:
                     value = json.loads(value)
                 except Exception:

hanzo_mcp/tools/memory/__init__.py

@@ -4,19 +4,26 @@ from mcp.server import FastMCP
 
 from hanzo_mcp.tools.common.base import BaseTool, ToolRegistry
 from hanzo_mcp.tools.common.permissions import PermissionManager
-from hanzo_mcp.tools.memory.memory_tools import (
-    CreateMemoriesTool,
-    DeleteMemoriesTool,
-    ManageMemoriesTool,
-    RecallMemoriesTool,
-    UpdateMemoriesTool,
-)
-from hanzo_mcp.tools.memory.knowledge_tools import (
-    StoreFactsTool,
-    RecallFactsTool,
-    SummarizeToMemoryTool,
-    ManageKnowledgeBasesTool,
-)
+
+# Import memory tools if available
+try:
+    from hanzo_mcp.tools.memory.memory_tools import (
+        CreateMemoriesTool,
+        DeleteMemoriesTool,
+        ManageMemoriesTool,
+        RecallMemoriesTool,
+        UpdateMemoriesTool,
+    )
+    from hanzo_mcp.tools.memory.knowledge_tools import (
+        StoreFactsTool,
+        RecallFactsTool,
+        SummarizeToMemoryTool,
+        ManageKnowledgeBasesTool,
+    )
+
+    MEMORY_TOOLS_AVAILABLE = True
+except ImportError:
+    MEMORY_TOOLS_AVAILABLE = False
 
 
 def register_memory_tools(
@@ -38,36 +45,22 @@ def register_memory_tools(
     Returns:
         List of registered tools
     """
+    if not MEMORY_TOOLS_AVAILABLE:
+        print("Warning: Memory tools not available (hanzo-memory package not found)")
+        return []
+
     # Create memory tools
-    recall_tool = RecallMemoriesTool(
-        user_id=user_id, project_id=project_id, **memory_config
-    )
-    create_tool = CreateMemoriesTool(
-        user_id=user_id, project_id=project_id, **memory_config
-    )
-    update_tool = UpdateMemoriesTool(
-        user_id=user_id, project_id=project_id, **memory_config
-    )
-    delete_tool = DeleteMemoriesTool(
-        user_id=user_id, project_id=project_id, **memory_config
-    )
-    manage_tool = ManageMemoriesTool(
-        user_id=user_id, project_id=project_id, **memory_config
-    )
+    recall_tool = RecallMemoriesTool(user_id=user_id, project_id=project_id, **memory_config)
+    create_tool = CreateMemoriesTool(user_id=user_id, project_id=project_id, **memory_config)
+    update_tool = UpdateMemoriesTool(user_id=user_id, project_id=project_id, **memory_config)
+    delete_tool = DeleteMemoriesTool(user_id=user_id, project_id=project_id, **memory_config)
+    manage_tool = ManageMemoriesTool(user_id=user_id, project_id=project_id, **memory_config)
 
     # Create knowledge tools
-    recall_facts_tool = RecallFactsTool(
-        user_id=user_id, project_id=project_id, **memory_config
-    )
-    store_facts_tool = StoreFactsTool(
-        user_id=user_id, project_id=project_id, **memory_config
-    )
-    summarize_tool = SummarizeToMemoryTool(
-        user_id=user_id, project_id=project_id, **memory_config
-    )
-    manage_kb_tool = ManageKnowledgeBasesTool(
-        user_id=user_id, project_id=project_id, **memory_config
-    )
+    recall_facts_tool = RecallFactsTool(user_id=user_id, project_id=project_id, **memory_config)
+    store_facts_tool = StoreFactsTool(user_id=user_id, project_id=project_id, **memory_config)
+    summarize_tool = SummarizeToMemoryTool(user_id=user_id, project_id=project_id, **memory_config)
+    manage_kb_tool = ManageKnowledgeBasesTool(user_id=user_id, project_id=project_id, **memory_config)
 
     # Register tools
     ToolRegistry.register_tool(mcp_server, recall_tool)