claude-mpm 4.1.0__py3-none-any.whl → 4.1.2__py3-none-any.whl

claude_mpm/cli/commands/mcp_pipx_config.py

@@ -12,7 +12,7 @@ import platform
 import subprocess
 import sys
 from pathlib import Path
-from typing import Optional, Dict, Any
+from typing import Any, Dict
 
 from claude_mpm.core.logger import get_logger
 
@@ -22,18 +22,30 @@ logger = get_logger(__name__)
 def find_claude_config_path() -> Path:
     """Find the Claude Code configuration file path."""
     system = platform.system()
-    
+
     if system == "Darwin":  # macOS
-        config_path = Path.home() / "Library" / "Application Support" / "Claude" / "claude_desktop_config.json"
+        config_path = (
+            Path.home()
+            / "Library"
+            / "Application Support"
+            / "Claude"
+            / "claude_desktop_config.json"
+        )
     elif system == "Windows":
         appdata = os.environ.get("APPDATA")
         if appdata:
             config_path = Path(appdata) / "Claude" / "claude_desktop_config.json"
         else:
-            config_path = Path.home() / "AppData" / "Roaming" / "Claude" / "claude_desktop_config.json"
+            config_path = (
+                Path.home()
+                / "AppData"
+                / "Roaming"
+                / "Claude"
+                / "claude_desktop_config.json"
+            )
     else:  # Linux and others
         config_path = Path.home() / ".config" / "Claude" / "claude_desktop_config.json"
-    
+
     return config_path
 
 
@@ -43,67 +55,62 @@ def check_pipx_installation() -> bool:
         # Check if running from pipx
         if "pipx" in sys.executable.lower():
             return True
-        
+
         # Check pipx list
         result = subprocess.run(
             ["pipx", "list", "--json"],
             capture_output=True,
             text=True,
-            timeout=5
+            timeout=5,
+            check=False,
         )
         if result.returncode == 0:
             pipx_data = json.loads(result.stdout)
             return "claude-mpm" in pipx_data.get("venvs", {})
     except Exception:
         pass
-    
+
     return False
 
 
 def create_mcp_config() -> Dict[str, Any]:
     """Create MCP configuration for pipx installation."""
-    return {
-        "mcpServers": {
-            "claude-mpm-gateway": {
-                "command": "claude-mpm-mcp"
-            }
-        }
-    }
+    return {"mcpServers": {"claude-mpm-gateway": {"command": "claude-mpm-mcp"}}}
 
 
 def configure_mcp_for_pipx(args) -> int:
     """
     Configure MCP for pipx installation.
-    
+
     Args:
         args: Command line arguments
-        
+
     Returns:
         Exit code (0 for success, 1 for failure)
     """
     print("Claude MPM - MCP Configuration for pipx")
     print("=" * 40)
-    
+
     # Check if this is a pipx installation
     if not check_pipx_installation():
         print("\n⚠️  This doesn't appear to be a pipx installation")
         print("This command is specifically for pipx users.")
         print("\nFor other installation methods, see:")
         print("  docs/MCP_SETUP.md")
-        
+
         if not args.force:
             return 1
         print("\n--force flag detected, continuing anyway...")
-    
+
     # Find Claude config
     config_path = find_claude_config_path()
     print(f"\n📁 Claude config path: {config_path}")
-    
+
     # Load existing config
     existing_config = {}
     if config_path.exists():
         try:
-            with open(config_path, 'r') as f:
+            with open(config_path) as f:
                 existing_config = json.load(f)
             print("✅ Existing config loaded")
         except json.JSONDecodeError:
@@ -114,28 +121,31 @@ def configure_mcp_for_pipx(args) -> int:
     else:
         print("📝 Config will be created")
         config_path.parent.mkdir(parents=True, exist_ok=True)
-    
+
     # Check for existing MCP config
-    if "mcpServers" in existing_config and "claude-mpm-gateway" in existing_config["mcpServers"]:
+    if (
+        "mcpServers" in existing_config
+        and "claude-mpm-gateway" in existing_config["mcpServers"]
+    ):
         print("\n⚠️  claude-mpm-gateway is already configured")
         if not args.force:
             print("Use --force to overwrite")
             return 0
         print("Overwriting existing configuration...")
-    
+
     # Create and merge config
     mcp_config = create_mcp_config()
     existing_config.update(mcp_config)
-    
+
     # Show what will be written
     if not args.quiet:
         print("\n📝 Configuration to write:")
         print(json.dumps(mcp_config, indent=2))
-    
+
     # Write config
     if not args.dry_run:
         try:
-            with open(config_path, 'w') as f:
+            with open(config_path, "w") as f:
                 json.dump(existing_config, f, indent=2)
             print(f"\n✅ Configuration written to: {config_path}")
         except Exception as e:
@@ -143,7 +153,7 @@ def configure_mcp_for_pipx(args) -> int:
             return 1
     else:
         print("\n--dry-run: Configuration not written")
-    
+
     # Test the command
     print("\n🧪 Testing claude-mpm-mcp command...")
     try:
@@ -151,7 +161,8 @@ def configure_mcp_for_pipx(args) -> int:
             ["which", "claude-mpm-mcp"],
             capture_output=True,
             text=True,
-            timeout=2
+            timeout=2,
+            check=False,
         )
         if result.returncode == 0:
             print(f"✅ Command found: {result.stdout.strip()}")
@@ -160,40 +171,38 @@ def configure_mcp_for_pipx(args) -> int:
             print("   Ensure pipx bin directory is in your PATH")
     except Exception as e:
         print(f"⚠️  Could not test command: {e}")
-    
+
     print("\n✨ Next steps:")
     print("1. Restart Claude Code")
     print("2. Look for the MCP icon in the interface")
     print("3. Try using @claude-mpm-gateway in a conversation")
     print("\nFor more help, see: docs/MCP_PIPX_SETUP.md")
-    
+
     return 0
 
 
 def add_parser(subparsers):
     """Add the mcp-pipx-config command parser."""
     parser = subparsers.add_parser(
-        'mcp-pipx-config',
-        help='Configure MCP for pipx installation',
-        description='Configure MCP Gateway for Claude Code when installed via pipx'
+        "mcp-pipx-config",
+        help="Configure MCP for pipx installation",
+        description="Configure MCP Gateway for Claude Code when installed via pipx",
     )
-    
+
     parser.add_argument(
-        '--force',
-        action='store_true',
-        help='Force configuration even if not pipx or already configured'
+        "--force",
+        action="store_true",
+        help="Force configuration even if not pipx or already configured",
     )
-    
+
     parser.add_argument(
-        '--dry-run',
-        action='store_true',
-        help='Show what would be done without making changes'
+        "--dry-run",
+        action="store_true",
+        help="Show what would be done without making changes",
     )
-    
+
     parser.add_argument(
-        '--quiet',
-        action='store_true',
-        help='Suppress non-essential output'
+        "--quiet", action="store_true", help="Suppress non-essential output"
     )
-    
-    parser.set_defaults(func=configure_mcp_for_pipx)
+
+    parser.set_defaults(func=configure_mcp_for_pipx)

claude_mpm/cli/commands/mcp_server_commands.py

@@ -4,9 +4,7 @@ This module provides MCP server management commands.
 Extracted from mcp.py to reduce complexity and improve maintainability.
 """
 
-import asyncio
 import os
-import shutil
 import sys
 from pathlib import Path
 
@@ -51,28 +49,31 @@ class MCPServerCommands:
             print("\n1. Run the configuration script:")
             print("   python scripts/configure_mcp_server.py")
             print("\n2. Or manually configure Claude Code:")
-            
+
             # Find project root for paths
             project_root = Path(__file__).parent.parent.parent.parent.parent
-            
+
             # Use the direct command, not the wrapper
             import shutil
+
             claude_mpm_path = shutil.which("claude-mpm")
             if not claude_mpm_path:
                 # Fallback to current executable
                 claude_mpm_path = sys.executable.replace("python", "claude-mpm")
-            
+
             print("\n   Add this to your Claude Code configuration:")
-            print("   (~/Library/Application Support/Claude/claude_desktop_config.json on macOS)")
+            print(
+                "   (~/Library/Application Support/Claude/claude_desktop_config.json on macOS)"
+            )
             print("\n   {")
             print('     "mcpServers": {')
             print('       "claude-mpm-gateway": {')
             print(f'         "command": "{claude_mpm_path}",')
-            print(f'         "args": ["mcp", "server"],')
+            print('         "args": ["mcp", "server"],')
             print(f'         "cwd": "{project_root}"')
-            print('       }')
-            print('     }')
-            print('   }')
+            print("       }")
+            print("     }")
+            print("   }")
             print("\n3. Restart Claude Code to load the MCP server")
             print("\nTo test the server directly:")
             print("   claude-mpm mcp server")
@@ -86,26 +87,30 @@ class MCPServerCommands:
         # Default behavior: Run the server directly in this process
         if test_mode:
             print("🧪 Starting MCP server in test mode...", file=sys.stderr)
-            print("   This will run the server with stdio communication.", file=sys.stderr)
+            print(
+                "   This will run the server with stdio communication.", file=sys.stderr
+            )
             print("   Press Ctrl+C to stop.\n", file=sys.stderr)
 
         try:
             # Import and run the server directly
-            from claude_mpm.services.mcp_gateway.server.stdio_server import SimpleMCPServer
-            
+            from claude_mpm.services.mcp_gateway.server.stdio_server import (
+                SimpleMCPServer,
+            )
+
             # Set environment variable if in test mode
             if test_mode:
                 os.environ["MCP_MODE"] = "test"
             else:
                 os.environ["MCP_MODE"] = "production"
-            
+
             # Create and run the server
             self.logger.info("Starting MCP Gateway Server directly...")
             server = SimpleMCPServer(name="claude-mpm-gateway", version="1.0.0")
-            
+
             # Run the server asynchronously
             await server.run()
-            
+
             return 0
 
         except ImportError as e:
@@ -113,7 +118,8 @@ class MCPServerCommands:
             # Don't print to stdout as it would interfere with JSON-RPC protocol
             # Log to stderr instead
             print(
-                f"❌ Error: Could not import MCP server components: {e}", file=sys.stderr
+                f"❌ Error: Could not import MCP server components: {e}",
+                file=sys.stderr,
             )
             print("\nMake sure the MCP package is installed:", file=sys.stderr)
             print("  pip install mcp", file=sys.stderr)