hanzo-mcp 0.7.7__py3-none-any.whl → 0.8.1__py3-none-any.whl

hanzo_mcp/tools/mcp/__init__.py

@@ -1,11 +1,10 @@
 """MCP management tools."""
 
-from hanzo_mcp.tools.mcp.mcp_tool import MCPTool
-
 # Legacy imports
 from hanzo_mcp.tools.mcp.mcp_add import McpAddTool
-from hanzo_mcp.tools.mcp.mcp_remove import McpRemoveTool
+from hanzo_mcp.tools.mcp.mcp_tool import MCPTool
 from hanzo_mcp.tools.mcp.mcp_stats import McpStatsTool
+from hanzo_mcp.tools.mcp.mcp_remove import McpRemoveTool
 
 __all__ = [
     "MCPTool",

hanzo_mcp/tools/mcp/mcp_add.py

@@ -1,18 +1,16 @@
 """Add MCP servers dynamically."""
 
 import json
-import subprocess
 import shutil
-from typing import Annotated, Optional, TypedDict, Unpack, final, override, Dict, Any
+from typing import Any, Dict, Unpack, Optional, Annotated, TypedDict, final, override
 from pathlib import Path
 
-from mcp.server.fastmcp import Context as MCPContext
 from pydantic import Field
+from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 
-
 ServerCommand = Annotated[
     str,
     Field(
@@ -67,7 +65,7 @@ class McpAddParams(TypedDict, total=False):
 @final
 class McpAddTool(BaseTool):
     """Tool for adding MCP servers dynamically."""
-    
+
     # Class variable to store added servers
     _mcp_servers: Dict[str, Dict[str, Any]] = {}
     _config_file = Path.home() / ".hanzo" / "mcp" / "servers.json"
@@ -82,7 +80,7 @@ class McpAddTool(BaseTool):
         """Load servers from config file."""
         if cls._config_file.exists():
             try:
-                with open(cls._config_file, 'r') as f:
+                with open(cls._config_file, "r") as f:
                     cls._mcp_servers = json.load(f)
             except Exception:
                 cls._mcp_servers = {}
@@ -91,7 +89,7 @@ class McpAddTool(BaseTool):
     def _save_servers(cls):
         """Save servers to config file."""
         cls._config_file.parent.mkdir(parents=True, exist_ok=True)
-        with open(cls._config_file, 'w') as f:
+        with open(cls._config_file, "w") as f:
             json.dump(cls._mcp_servers, f, indent=2)
 
     @classmethod
@@ -171,9 +169,7 @@ Use 'mcp_stats' to see all added servers and their status.
         server_type = "unknown"
         if command.startswith("npx"):
             server_type = "node"
-        elif command.startswith("uvx"):
-            server_type = "python"
-        elif command.startswith("python"):
+        elif command.startswith("uvx") or command.startswith("python"):
             server_type = "python"
         elif command.startswith("node"):
             server_type = "node"
@@ -182,16 +178,20 @@ Use 'mcp_stats' to see all added servers and their status.
         full_command = [command]
         if args:
             import shlex
+
             # If command contains spaces, split it first
-            if ' ' in command:
+            if " " in command:
                 full_command = shlex.split(command)
             full_command.extend(shlex.split(args))
         else:
-            if ' ' in command:
+            if " " in command:
                 import shlex
+
                 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 = {
@@ -203,7 +203,7 @@ Use 'mcp_stats' to see all added servers and their status.
             "process_id": None,
             "tools": [],
             "resources": [],
-            "prompts": []
+            "prompts": [],
         }
 
         # Test if command is valid
@@ -211,7 +211,7 @@ Use 'mcp_stats' to see all added servers and their status.
             try:
                 # Try to start the server briefly to validate
                 test_env = {**env} if env else {}
-                
+
                 # Quick test to see if command exists
                 test_cmd = full_command[0]
                 if test_cmd == "npx":
@@ -220,11 +220,11 @@ Use 'mcp_stats' to see all added servers and their status.
                 elif test_cmd == "uvx":
                     if not shutil.which("uvx"):
                         return "Error: uvx not found. Install uv first."
-                
+
                 # TODO: Actually start and connect to the MCP server
                 # For now, we just store the configuration
                 server_config["status"] = "ready"
-                
+
             except Exception as e:
                 await tool_ctx.error(f"Failed to validate server: {str(e)}")
                 server_config["status"] = "error"
@@ -240,22 +240,24 @@ Use 'mcp_stats' to see all added servers and their status.
             f"  Command: {' '.join(full_command)}",
             f"  Status: {server_config['status']}",
         ]
-        
+
         if env:
             output.append(f"  Environment: {list(env.keys())}")
-        
-        output.extend([
-            "",
-            "Use 'mcp_stats' to see server details.",
-            f"Use 'mcp_remove --name {name}' to remove this server."
-        ])
+
+        output.extend(
+            [
+                "",
+                "Use 'mcp_stats' to see server details.",
+                f"Use 'mcp_remove --name {name}' to remove this server.",
+            ]
+        )
 
         # Note: In a real implementation, we would:
         # 1. Start the MCP server process
         # 2. Connect to it via stdio/HTTP
         # 3. Query its capabilities (tools, resources, prompts)
         # 4. Register those with our MCP server
-        
+
         return "\n".join(output)
 
     def register(self, mcp_server) -> None:

hanzo_mcp/tools/mcp/mcp_remove.py

@@ -1,14 +1,13 @@
 """Remove MCP servers."""
 
-from typing import Annotated, TypedDict, Unpack, final, override
+from typing import Unpack, Annotated, TypedDict, final, override
 
-from mcp.server.fastmcp import Context as MCPContext
 from pydantic import Field
+from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import BaseTool
-from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.mcp.mcp_add import McpAddTool
-
+from hanzo_mcp.tools.common.context import create_tool_context
 
 ServerName = Annotated[
     str,
@@ -91,12 +90,12 @@ Use 'mcp_stats' to see all servers before removing.
 
         # Get current servers
         servers = McpAddTool.get_servers()
-        
+
         if name not in servers:
             return f"Error: Server '{name}' not found. Use 'mcp_stats' to see available servers."
 
         server = servers[name]
-        
+
         await tool_ctx.info(f"Removing MCP server '{name}'")
 
         # Check if server is running
@@ -116,10 +115,10 @@ Use 'mcp_stats' to see all servers before removing.
             f"  Type: {server.get('type', 'unknown')}",
             f"  Command: {' '.join(server.get('command', []))}",
         ]
-        
+
         if server.get("tools"):
             output.append(f"  Tools removed: {len(server['tools'])}")
-        
+
         return "\n".join(output)
 
     def register(self, mcp_server) -> None:

hanzo_mcp/tools/mcp/mcp_stats.py

@@ -1,18 +1,17 @@
 """MCP server statistics."""
 
-import json
-from typing import TypedDict, Unpack, final, override
-from datetime import datetime
+from typing import Unpack, TypedDict, final, override
 
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import BaseTool
-from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.mcp.mcp_add import McpAddTool
+from hanzo_mcp.tools.common.context import create_tool_context
 
 
 class McpStatsParams(TypedDict, total=False):
     """Parameters for MCP stats tool."""
+
     pass
 
 
@@ -67,66 +66,68 @@ Example:
 
         # Get all servers
         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 ===")
         output.append(f"Total Servers: {len(servers)}")
         output.append("")
-        
+
         # Count by type
         type_counts = {}
         status_counts = {}
         total_tools = 0
         total_resources = 0
-        
+
         for server in servers.values():
             # Count types
             server_type = server.get("type", "unknown")
             type_counts[server_type] = type_counts.get(server_type, 0) + 1
-            
+
             # Count status
             status = server.get("status", "unknown")
             status_counts[status] = status_counts.get(status, 0) + 1
-            
+
             # Count tools and resources
             total_tools += len(server.get("tools", []))
             total_resources += len(server.get("resources", []))
-        
+
         # Server types
         output.append("Server Types:")
         for stype, count in sorted(type_counts.items()):
             output.append(f"  {stype}: {count}")
         output.append("")
-        
+
         # Server status
         output.append("Server Status:")
         for status, count in sorted(status_counts.items()):
             output.append(f"  {status}: {count}")
         output.append("")
-        
+
         # Tools and resources
         output.append(f"Total Tools Available: {total_tools}")
         output.append(f"Total Resources Available: {total_resources}")
         output.append("")
-        
+
         # Individual server details
         output.append("=== Server Details ===")
-        
+
         for name, server in sorted(servers.items()):
             output.append(f"\n{name}:")
             output.append(f"  Type: {server.get('type', 'unknown')}")
             output.append(f"  Status: {server.get('status', 'unknown')}")
             output.append(f"  Command: {' '.join(server.get('command', []))}")
-            
+
             if server.get("process_id"):
                 output.append(f"  Process ID: {server['process_id']}")
-            
+
             if server.get("error"):
                 output.append(f"  Error: {server['error']}")
-            
+
             tools = server.get("tools", [])
             if tools:
                 output.append(f"  Tools ({len(tools)}):")
@@ -134,7 +135,7 @@ Example:
                     output.append(f"    - {tool}")
                 if len(tools) > 5:
                     output.append(f"    ... and {len(tools) - 5} more")
-            
+
             resources = server.get("resources", [])
             if resources:
                 output.append(f"  Resources ({len(resources)}):")
@@ -142,10 +143,10 @@ Example:
                     output.append(f"    - {resource}")
                 if len(resources) > 5:
                     output.append(f"    ... and {len(resources) - 5} more")
-            
+
             if server.get("env"):
                 output.append(f"  Environment vars: {list(server['env'].keys())}")
-        
+
         # Common MCP servers hint
         output.append("\n=== Available MCP Servers ===")
         output.append("Common servers you can add:")
@@ -157,7 +158,7 @@ Example:
         output.append("  - @modelcontextprotocol/server-iterm2")
         output.append("  - @modelcontextprotocol/server-linear")
         output.append("  - @modelcontextprotocol/server-slack")
-        
+
         return "\n".join(output)
 
     def register(self, mcp_server) -> None: