claude-mpm 3.9.11__py3-none-any.whl → 4.0.4__py3-none-any.whl

claude_mpm/init.py

@@ -1,30 +1,32 @@
+from pathlib import Path
+
 """Initialization module for claude-mpm.
 
 Handles creation of necessary directories and configuration files.
 """
 
+import json
 import os
 import shutil
-from pathlib import Path
-from typing import Optional, Dict, Any
-import json
+from typing import Any, Dict, Optional
+
 import yaml
 
-from claude_mpm.core.logger import get_logger
 from claude_mpm.core.constants import NetworkConfig
+from claude_mpm.core.logger import get_logger
 
 
 class ProjectInitializer:
     """Handles initialization of claude-mpm directories and configuration."""
-    
+
     def __init__(self):
         self.logger = get_logger("initializer")
         self.user_dir = Path.home() / ".claude-mpm"
         self.project_dir = None
-        
+
     def initialize_user_directory(self) -> bool:
         """Initialize user-level .claude-mpm directory structure.
-        
+
         Creates:
         - ~/.claude-mpm/
           - agents/
@@ -37,7 +39,7 @@ class ProjectInitializer:
         try:
             # Create main user directory
             self.user_dir.mkdir(exist_ok=True)
-            
+
             # Create subdirectories
             directories = [
                 self.user_dir / "agents" / "user-defined",
@@ -46,34 +48,34 @@ class ProjectInitializer:
                 self.user_dir / "templates",
                 self.user_dir / "registry",
             ]
-            
+
             for directory in directories:
                 directory.mkdir(parents=True, exist_ok=True)
-            
+
             # Check for migration from old settings.json to new configuration.yaml
             old_config_file = self.user_dir / "config" / "settings.json"
             config_file = self.user_dir / "config" / "configuration.yaml"
-            
+
             # Migrate if old file exists but new doesn't
             if old_config_file.exists() and not config_file.exists():
                 self._migrate_json_to_yaml(old_config_file, config_file)
             elif not config_file.exists():
                 # Create default configuration if it doesn't exist
                 self._create_default_config(config_file)
-            
+
             # Copy agent templates if they don't exist
             self._copy_agent_templates()
-            
+
             self.logger.info(f"Initialized user directory at {self.user_dir}")
             return True
-            
+
         except Exception as e:
             self.logger.error(f"Failed to initialize user directory: {e}")
             return False
-    
+
     def initialize_project_directory(self, project_path: Optional[Path] = None) -> bool:
         """Initialize project-level .claude-mpm directory structure.
-        
+
         Creates:
         - .claude-mpm/
           - agents/     (for project agent JSON files)
@@ -89,94 +91,109 @@ class ProjectInitializer:
             else:
                 # Check for the user's original working directory from launch script
                 # The launch script sets CLAUDE_MPM_USER_PWD before changing to framework directory
-                user_pwd = os.environ.get('CLAUDE_MPM_USER_PWD')
-                
+                user_pwd = os.environ.get("CLAUDE_MPM_USER_PWD")
+
                 if user_pwd:
                     # Use the original user working directory
                     project_root = Path(user_pwd)
-                    self.logger.debug(f"Using user working directory from CLAUDE_MPM_USER_PWD: {project_root}")
+                    self.logger.debug(
+                        f"Using user working directory from CLAUDE_MPM_USER_PWD: {project_root}"
+                    )
                 else:
                     # Fallback to current working directory (backward compatibility)
                     project_root = Path.cwd()
-                    self.logger.debug(f"CLAUDE_MPM_USER_PWD not set, using cwd: {project_root}")
-                
+                    self.logger.debug(
+                        f"CLAUDE_MPM_USER_PWD not set, using cwd: {project_root}"
+                    )
+
                 self.project_dir = project_root / ".claude-mpm"
-            
+
             # Check if directory already exists
             directory_existed = self.project_dir.exists()
-            
+
             # Migrate existing agents from project-specific subdirectory if needed
             self._migrate_project_agents()
-            
+
             # Create project directory
             self.project_dir.mkdir(exist_ok=True)
-            
+
             # Create subdirectories
             directories = [
-                self.project_dir / "agents",  # Direct agents directory for project agents
+                self.project_dir
+                / "agents",  # Direct agents directory for project agents
                 self.project_dir / "config",
                 self.project_dir / "responses",
                 self.project_dir / "logs",
             ]
-            
+
             for directory in directories:
                 directory.mkdir(parents=True, exist_ok=True)
-            
+
             # Create project configuration
             config_file = self.project_dir / "config" / "project.json"
             if not config_file.exists():
                 self._create_project_config(config_file)
-            
+
             # Create .gitignore for project directory
             gitignore = self.project_dir / ".gitignore"
             if not gitignore.exists():
                 gitignore.write_text("logs/\n*.log\n*.pyc\n__pycache__/\n")
-            
+
             # Log successful creation with details
             self.logger.info(f"Initialized project directory at {self.project_dir}")
             self.logger.debug(f"Created directories: agents, config, responses, logs")
-            
+
             # Print appropriate message to console for visibility during startup
-            if directory_existed:
-                print(f"✓ Found existing .claude-mpm/ directory in {project_root}")
-            else:
-                print(f"✓ Initialized .claude-mpm/ in {project_root}")
-            
-            # Check if migration happened
-            agents_dir = self.project_dir / "agents"
-            if agents_dir.exists() and any(agents_dir.glob("*.json")):
-                agent_count = len(list(agents_dir.glob("*.json")))
-                print(f"✓ Found {agent_count} project agent(s) in .claude-mpm/agents/")
-            
+            # BUT: Don't print to stdout when running MCP server (interferes with JSON-RPC)
+            import sys
+
+            is_mcp_mode = "mcp" in sys.argv and "start" in sys.argv
+
+            if not is_mcp_mode:
+                if directory_existed:
+                    print(f"✓ Found existing .claude-mpm/ directory in {project_root}")
+                else:
+                    print(f"✓ Initialized .claude-mpm/ in {project_root}")
+
+                # Check if migration happened
+                agents_dir = self.project_dir / "agents"
+                if agents_dir.exists() and any(agents_dir.glob("*.json")):
+                    agent_count = len(list(agents_dir.glob("*.json")))
+                    print(
+                        f"✓ Found {agent_count} project agent(s) in .claude-mpm/agents/"
+                    )
+
             return True
-            
+
         except Exception as e:
             self.logger.error(f"Failed to initialize project directory: {e}")
             print(f"✗ Failed to create .claude-mpm/ directory: {e}")
             return False
-    
+
     def _migrate_project_agents(self):
         """Migrate agents from old subdirectory structure to direct agents directory.
-        
+
         WHY: We're simplifying the directory structure to match the deployment expectations.
-        The old structure had a subdirectory but the deployment now looks for agents 
+        The old structure had a subdirectory but the deployment now looks for agents
         directly in .claude-mpm/agents/.
         """
         if not self.project_dir:
             return
-        
+
         old_agents_dir = self.project_dir / "agents" / "project-specific"
         new_agents_dir = self.project_dir / "agents"
-        
+
         # Check if old directory exists with JSON files
         if old_agents_dir.exists() and old_agents_dir.is_dir():
             json_files = list(old_agents_dir.glob("*.json"))
             if json_files:
-                self.logger.info(f"Migrating {len(json_files)} agents from old subdirectory")
-                
+                self.logger.info(
+                    f"Migrating {len(json_files)} agents from old subdirectory"
+                )
+
                 # Ensure new agents directory exists
                 new_agents_dir.mkdir(parents=True, exist_ok=True)
-                
+
                 # Move each JSON file
                 migrated_count = 0
                 for json_file in json_files:
@@ -186,15 +203,26 @@ class ProjectInitializer:
                             # Move the file
                             shutil.move(str(json_file), str(target_file))
                             migrated_count += 1
-                            self.logger.debug(f"Migrated {json_file.name} to agents directory")
+                            self.logger.debug(
+                                f"Migrated {json_file.name} to agents directory"
+                            )
                         else:
-                            self.logger.debug(f"Skipping {json_file.name} - already exists in target")
+                            self.logger.debug(
+                                f"Skipping {json_file.name} - already exists in target"
+                            )
                     except Exception as e:
                         self.logger.error(f"Failed to migrate {json_file.name}: {e}")
-                
+
                 if migrated_count > 0:
-                    print(f"✓ Migrated {migrated_count} agent(s) from old location to agents/")
-                
+                    # Don't print to stdout when running MCP server
+                    import sys
+
+                    is_mcp_mode = "mcp" in sys.argv and "start" in sys.argv
+                    if not is_mcp_mode:
+                        print(
+                            f"✓ Migrated {migrated_count} agent(s) from old location to agents/"
+                        )
+
                 # Remove old directory if empty
                 try:
                     if not any(old_agents_dir.iterdir()):
@@ -202,7 +230,7 @@ class ProjectInitializer:
                         self.logger.debug("Removed empty old subdirectory")
                 except Exception as e:
                     self.logger.debug(f"Could not remove old directory: {e}")
-    
+
     def _find_project_root(self) -> Optional[Path]:
         """Find project root by looking for .git or other project markers."""
         current = Path.cwd()
@@ -215,149 +243,145 @@ class ProjectInitializer:
                 return current
             current = current.parent
         return None
-    
+
     def _migrate_json_to_yaml(self, old_file: Path, new_file: Path):
         """Migrate configuration from JSON to YAML format.
-        
+
         Args:
             old_file: Path to existing settings.json
             new_file: Path to new configuration.yaml
         """
         try:
             # Read existing JSON configuration
-            with open(old_file, 'r') as f:
+            with open(old_file, "r") as f:
                 config = json.load(f)
-            
+
             # Write as YAML
-            with open(new_file, 'w') as f:
+            with open(new_file, "w") as f:
                 yaml.dump(config, f, default_flow_style=False, sort_keys=False)
-            
-            self.logger.info(f"Migrated configuration from {old_file.name} to {new_file.name}")
-            
+
+            self.logger.info(
+                f"Migrated configuration from {old_file.name} to {new_file.name}"
+            )
+
             # Optionally rename old file to .backup
-            backup_file = old_file.with_suffix('.json.backup')
+            backup_file = old_file.with_suffix(".json.backup")
             old_file.rename(backup_file)
             self.logger.info(f"Renamed old configuration to {backup_file.name}")
-            
+
         except Exception as e:
             self.logger.error(f"Failed to migrate configuration: {e}")
             # Fall back to creating default config
             self._create_default_config(new_file)
-    
+
     def _create_default_config(self, config_file: Path):
         """Create default user configuration in YAML format."""
         default_config = {
             "version": "1.0",
             "hooks": {
                 "enabled": True,
-                "port_range": list(NetworkConfig.SOCKETIO_PORT_RANGE)
-            },
-            "logging": {
-                "level": "INFO",
-                "max_size_mb": 100,
-                "retention_days": 30
+                "port_range": list(NetworkConfig.SOCKETIO_PORT_RANGE),
             },
+            "logging": {"level": "INFO", "max_size_mb": 100, "retention_days": 30},
             "agents": {
                 "auto_discover": True,
-                "precedence": ["project", "user", "system"]
+                "precedence": ["project", "user", "system"],
             },
             "orchestration": {
                 "default_mode": "subprocess",
-                "enable_todo_hijacking": False
-            }
+                "enable_todo_hijacking": False,
+            },
         }
-        
-        with open(config_file, 'w') as f:
+
+        with open(config_file, "w") as f:
             yaml.dump(default_config, f, default_flow_style=False, sort_keys=False)
-    
+
     def _create_project_config(self, config_file: Path):
         """Create default project configuration."""
         project_config = {
             "version": "1.0",
             "project_name": Path.cwd().name,
-            "agents": {
-                "enabled": True
-            },
-            "tickets": {
-                "auto_create": True,
-                "prefix": "TSK"
-            }
+            "agents": {"enabled": True},
+            "tickets": {"auto_create": True, "prefix": "TSK"},
         }
-        
-        with open(config_file, 'w') as f:
+
+        with open(config_file, "w") as f:
             json.dump(project_config, f, indent=2)
-    
+
     def _copy_agent_templates(self):
         """Copy agent templates to user directory."""
         # Get the package directory
         package_dir = Path(__file__).parent
         templates_src = package_dir / "agents" / "templates"
         templates_dst = self.user_dir / "templates"
-        
+
         if templates_src.exists():
             for template_file in templates_src.glob("*.md"):
                 dst_file = templates_dst / template_file.name
                 if not dst_file.exists():
                     shutil.copy2(template_file, dst_file)
-    
+
     def validate_dependencies(self) -> Dict[str, bool]:
         """Validate that all required dependencies are available."""
         dependencies = {}
-        
+
         # Check Python version
         import sys
-        dependencies['python'] = sys.version_info >= (3, 8)
-        
+
+        dependencies["python"] = sys.version_info >= (3, 8)
+
         # Check Claude CLI
-        dependencies['claude_cli'] = shutil.which("claude") is not None
-        
+        dependencies["claude_cli"] = shutil.which("claude") is not None
+
         # Check required Python packages
         required_packages = [
-            'ai_trackdown_pytools',
-            'yaml',
-            'dotenv',
-            'click',
-            'pexpect',
-            'psutil',
-            'requests',
-            'flask',
-            'watchdog',
-            'tree_sitter'
+            "ai_trackdown_pytools",
+            "yaml",
+            "dotenv",
+            "click",
+            "pexpect",
+            "psutil",
+            "requests",
+            "flask",
+            "watchdog",
+            "tree_sitter",
         ]
-        
+
         for package in required_packages:
             try:
                 __import__(package)
                 dependencies[package] = True
             except ImportError:
                 dependencies[package] = False
-        
+
         return dependencies
-    
+
     def ensure_initialized(self) -> bool:
         """Ensure both user and project directories are initialized.
-        
+
         Shows clear information about where directories are being created.
         """
         # Determine actual working directory
-        user_pwd = os.environ.get('CLAUDE_MPM_USER_PWD')
+        user_pwd = os.environ.get("CLAUDE_MPM_USER_PWD")
         if user_pwd:
             actual_wd = Path(user_pwd)
-            self.logger.info(f"User working directory (from CLAUDE_MPM_USER_PWD): {actual_wd}")
+            self.logger.info(
+                f"User working directory (from CLAUDE_MPM_USER_PWD): {actual_wd}"
+            )
         else:
             actual_wd = Path.cwd()
             self.logger.info(f"Working directory: {actual_wd}")
-        
+
         framework_path = Path(__file__).parent.parent.parent
         self.logger.info(f"Framework path: {framework_path}")
-        
+
         # Initialize user directory (in home)
         user_ok = self.initialize_user_directory()
-        
+
         # Initialize project directory (in user's actual working directory)
         self.logger.info(f"Checking for .claude-mpm/ in {actual_wd}")
         project_ok = self.initialize_project_directory()
-        
+
         return user_ok and project_ok
 
 
@@ -371,9 +395,9 @@ def validate_installation():
     """Validate that claude-mpm is properly installed."""
     initializer = ProjectInitializer()
     deps = initializer.validate_dependencies()
-    
+
     all_ok = all(deps.values())
-    
+
     if not all_ok:
         print("❌ Missing dependencies:")
         for dep, status in deps.items():
@@ -381,5 +405,5 @@ def validate_installation():
                 print(f"  - {dep}")
     else:
         print("✅ All dependencies are installed")
-    
-    return all_ok
+
+    return all_ok

claude_mpm/models/__init__.py

@@ -8,17 +8,17 @@ providing a single source of truth for data structures across the system.
 from .agent_definition import (
     AgentDefinition,
     AgentMetadata,
-    AgentType,
+    AgentPermissions,
     AgentSection,
+    AgentType,
     AgentWorkflow,
-    AgentPermissions
 )
 
 __all__ = [
-    'AgentDefinition',
-    'AgentMetadata',
-    'AgentType',
-    'AgentSection',
-    'AgentWorkflow',
-    'AgentPermissions'
-]
+    "AgentDefinition",
+    "AgentMetadata",
+    "AgentType",
+    "AgentSection",
+    "AgentWorkflow",
+    "AgentPermissions",
+]