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

claude_mpm/cli/commands/agent_manager.py

@@ -9,84 +9,79 @@ This module provides CLI interface for:
 """
 
 import json
-import os
 import shutil
 import sys
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 
-from ...core.config import Config
 from ...core.logging_config import get_logger
 from ...services.agents.agent_builder import AgentBuilderService
 from ...services.agents.deployment.agent_deployment import AgentDeploymentService
-from ..shared import AgentCommand, CommandResult, add_output_arguments
+from ..shared import AgentCommand, CommandResult
 
 
 class AgentManagerCommand(AgentCommand):
     """Agent Manager command for comprehensive agent management."""
-    
+
     def __init__(self):
         super().__init__("agent-manager")
         self.builder_service = AgentBuilderService()
         self.deployment_service = None
         self.logger = get_logger(__name__)
-        
+
     @property
     def deployment(self):
         """Lazy load deployment service."""
         if self.deployment_service is None:
             self.deployment_service = AgentDeploymentService()
         return self.deployment_service
-        
+
     def run(self, args) -> CommandResult:
         """Execute agent manager command.
-        
+
         Args:
             args: Command arguments
-            
+
         Returns:
             CommandResult with operation status
         """
-        if not hasattr(args, 'agent_manager_command'):
+        if not hasattr(args, "agent_manager_command"):
             return self._show_help()
-            
+
         command_map = {
-            'list': self._list_agents,
-            'create': self._create_agent,
-            'variant': self._create_variant,
-            'deploy': self._deploy_agent,
-            'customize-pm': self._customize_pm,
-            'show': self._show_agent,
-            'test': self._test_agent,
-            'templates': self._list_templates
+            "list": self._list_agents,
+            "create": self._create_agent,
+            "variant": self._create_variant,
+            "deploy": self._deploy_agent,
+            "customize-pm": self._customize_pm,
+            "show": self._show_agent,
+            "test": self._test_agent,
+            "templates": self._list_templates,
         }
-        
+
         command = args.agent_manager_command
         if command in command_map:
             return command_map[command](args)
-        else:
-            return CommandResult.error_result(f"Unknown command: {command}")
-            
+        return CommandResult.error_result(f"Unknown command: {command}")
+
     def _list_agents(self, args) -> CommandResult:
         """List all agents across tiers with hierarchy.
-        
+
         Shows agents from:
         1. Project level (.claude/agents/)
         2. User level (~/.claude/agents/)
         3. System level (framework)
         """
-        agents = {
-            "project": [],
-            "user": [],
-            "system": []
-        }
-        
+        agents = {"project": [], "user": [], "system": []}
+
         # Check project level
         project_dir = Path.cwd() / ".claude" / "agents"
         if project_dir.exists():
             for agent_file in project_dir.glob("*.yaml"):
-                agents["project"].append(self._read_agent_summary(agent_file, "project"))
-                
+                agents["project"].append(
+                    self._read_agent_summary(agent_file, "project")
+                )
+
         # Check user level
         user_dir = Path.home() / ".claude" / "agents"
         if user_dir.exists():
@@ -95,263 +90,271 @@ class AgentManagerCommand(AgentCommand):
                 # Skip if overridden by project
                 if not any(a["id"] == agent_id for a in agents["project"]):
                     agents["user"].append(self._read_agent_summary(agent_file, "user"))
-                    
+
         # Get system agents
         templates = self.builder_service.list_available_templates()
         for template in templates:
             agent_id = template["id"]
             # Skip if overridden by project or user
             if not any(a["id"] == agent_id for a in agents["project"] + agents["user"]):
-                agents["system"].append({
-                    "id": agent_id,
-                    "name": template["name"],
-                    "tier": "system",
-                    "description": template["description"],
-                    "category": template.get("category", "custom")
-                })
-                
+                agents["system"].append(
+                    {
+                        "id": agent_id,
+                        "name": template["name"],
+                        "tier": "system",
+                        "description": template["description"],
+                        "category": template.get("category", "custom"),
+                    }
+                )
+
         # Format output
-        output_format = getattr(args, 'format', 'text')
+        output_format = getattr(args, "format", "text")
         if output_format == "json":
             return CommandResult.success_result("Agents listed", data=agents)
-        else:
-            output = self._format_agent_list(agents)
-            return CommandResult.success_result(output)
-            
+        output = self._format_agent_list(agents)
+        return CommandResult.success_result(output)
+
     def _create_agent(self, args) -> CommandResult:
         """Create a new agent interactively or from arguments."""
         try:
             # Interactive mode if no arguments
-            if not hasattr(args, 'agent_id'):
+            if not hasattr(args, "agent_id"):
                 return self._interactive_create()
-                
+
             # Create from arguments
             config, instructions = self.builder_service.create_agent(
                 agent_id=args.agent_id,
-                name=getattr(args, 'name', args.agent_id),
-                description=getattr(args, 'description', f"Custom agent {args.agent_id}"),
-                model=getattr(args, 'model', 'sonnet'),
-                tool_choice=getattr(args, 'tool_choice', 'auto'),
-                base_template=getattr(args, 'template', None)
+                name=getattr(args, "name", args.agent_id),
+                description=getattr(
+                    args, "description", f"Custom agent {args.agent_id}"
+                ),
+                model=getattr(args, "model", "sonnet"),
+                tool_choice=getattr(args, "tool_choice", "auto"),
+                base_template=getattr(args, "template", None),
             )
-            
+
             # Save agent files
             result = self._save_agent(config, instructions, args.agent_id)
-            
+
             if result:
-                return CommandResult.success_result(f"Agent '{args.agent_id}' created successfully")
-            else:
-                return CommandResult.error_result("Failed to save agent files")
-                
+                return CommandResult.success_result(
+                    f"Agent '{args.agent_id}' created successfully"
+                )
+            return CommandResult.error_result("Failed to save agent files")
+
         except Exception as e:
             return CommandResult.error_result(f"Failed to create agent: {e}")
-            
+
     def _create_variant(self, args) -> CommandResult:
         """Create an agent variant."""
         try:
-            if not hasattr(args, 'base_agent'):
-                return CommandResult.error_result("Base agent ID required for variant creation")
-                
+            if not hasattr(args, "base_agent"):
+                return CommandResult.error_result(
+                    "Base agent ID required for variant creation"
+                )
+
             modifications = {}
-            if hasattr(args, 'model'):
-                modifications['model'] = args.model
-            if hasattr(args, 'tool_choice'):
-                modifications['tool_choice'] = args.tool_choice
-                
+            if hasattr(args, "model"):
+                modifications["model"] = args.model
+            if hasattr(args, "tool_choice"):
+                modifications["tool_choice"] = args.tool_choice
+
             config, instructions = self.builder_service.create_variant(
                 base_agent_id=args.base_agent,
                 variant_id=args.variant_id,
-                variant_name=getattr(args, 'name', f"{args.base_agent}-variant"),
+                variant_name=getattr(args, "name", f"{args.base_agent}-variant"),
                 modifications=modifications,
-                instructions_append=getattr(args, 'instructions', None)
+                instructions_append=getattr(args, "instructions", None),
             )
-            
+
             # Save variant
             result = self._save_agent(config, instructions, args.variant_id)
-            
+
             if result:
-                return CommandResult.success_result(f"Variant '{args.variant_id}' created successfully")
-            else:
-                return CommandResult.error_result("Failed to save variant files")
-                
+                return CommandResult.success_result(
+                    f"Variant '{args.variant_id}' created successfully"
+                )
+            return CommandResult.error_result("Failed to save variant files")
+
         except Exception as e:
             return CommandResult.error_result(f"Failed to create variant: {e}")
-            
+
     def _deploy_agent(self, args) -> CommandResult:
         """Deploy an agent to specified tier."""
         try:
             agent_id = args.agent_id
-            tier = getattr(args, 'tier', 'project')  # Default to project (changed from 'user')
-            
+            tier = getattr(
+                args, "tier", "project"
+            )  # Default to project (changed from 'user')
+
             # Always deploy to project directory
             # Regardless of tier, all agents go to project .claude/agents
             deploy_path = Path.cwd() / ".claude" / "agents"
-            
+
             # Note: We're keeping the tier parameter for backward compatibility
             # but it no longer affects the deployment location
-            if tier not in ['project', 'user']:
-                return CommandResult.error_result("Invalid tier. Use 'project' or 'user'")
-                
+            if tier not in ["project", "user"]:
+                return CommandResult.error_result(
+                    "Invalid tier. Use 'project' or 'user'"
+                )
+
             # Create directory if needed
             deploy_path.mkdir(parents=True, exist_ok=True)
-            
+
             # Find agent files
             template_dir = Path(__file__).parent.parent.parent / "agents" / "templates"
             json_file = template_dir / f"{agent_id}.json"
-            md_file = template_dir / f"{agent_id}.md"
-            
+            template_dir / f"{agent_id}.md"
+
             if not json_file.exists():
                 return CommandResult.error_result(f"Agent '{agent_id}' not found")
-                
+
             # Deploy using deployment service
             # Pass Path object, not string
             self.deployment.deploy_agent(agent_id, deploy_path)
-            
+
             return CommandResult.success_result(
                 f"Agent '{agent_id}' deployed to {tier} level"
             )
-            
+
         except Exception as e:
             return CommandResult.error_result(f"Deployment failed: {e}")
-            
+
     def _customize_pm(self, args) -> CommandResult:
         """Customize PM instructions via .claude-mpm/INSTRUCTIONS.md."""
         try:
-            level = getattr(args, 'level', 'user')
-            
+            level = getattr(args, "level", "user")
+
             # Use .claude-mpm/INSTRUCTIONS.md for customization
-            if level == 'user':
+            if level == "user":
                 pm_file = Path.home() / ".claude-mpm" / "INSTRUCTIONS.md"
-            elif level == 'project':
+            elif level == "project":
                 pm_file = Path.cwd() / ".claude-mpm" / "INSTRUCTIONS.md"
             else:
-                return CommandResult.error_result("Invalid level. Use 'user' or 'project'")
-                
+                return CommandResult.error_result(
+                    "Invalid level. Use 'user' or 'project'"
+                )
+
             # Create backup if file exists
             if pm_file.exists():
-                backup_file = pm_file.with_suffix('.md.backup')
+                backup_file = pm_file.with_suffix(".md.backup")
                 shutil.copy(pm_file, backup_file)
                 self.logger.info(f"Backup created: {backup_file}")
-                
+
             # Generate or load instructions
-            if hasattr(args, 'template'):
+            if hasattr(args, "template"):
                 instructions = self._load_pm_template(args.template)
             else:
                 instructions = self.builder_service.generate_pm_instructions(
-                    delegation_patterns=getattr(args, 'patterns', None),
-                    workflow_overrides=getattr(args, 'workflows', None),
-                    custom_rules=getattr(args, 'rules', None)
+                    delegation_patterns=getattr(args, "patterns", None),
+                    workflow_overrides=getattr(args, "workflows", None),
+                    custom_rules=getattr(args, "rules", None),
                 )
-                
+
             # Save instructions to .claude-mpm directory
             pm_file.parent.mkdir(parents=True, exist_ok=True)
             pm_file.write_text(instructions)
-            
+
             return CommandResult.success_result(
                 f"PM instructions customized at {level} level: {pm_file}\n"
                 f"Note: These instructions will be loaded by the framework loader."
             )
-            
+
         except Exception as e:
             return CommandResult.error_result(f"Failed to customize PM: {e}")
-            
+
     def _show_agent(self, args) -> CommandResult:
         """Show detailed agent information."""
         try:
             agent_id = args.agent_id
-            
+
             # Find agent across tiers
             agent_info = self._find_agent(agent_id)
-            
+
             if not agent_info:
                 return CommandResult.error_result(f"Agent '{agent_id}' not found")
-                
-            output_format = getattr(args, 'format', 'text')
+
+            output_format = getattr(args, "format", "text")
             if output_format == "json":
                 return CommandResult.success_result("Agent details", data=agent_info)
-            else:
-                output = self._format_agent_details(agent_info)
-                return CommandResult.success_result(output)
-                
+            output = self._format_agent_details(agent_info)
+            return CommandResult.success_result(output)
+
         except Exception as e:
             return CommandResult.error_result(f"Failed to show agent: {e}")
-            
+
     def _test_agent(self, args) -> CommandResult:
         """Test agent configuration."""
         try:
             agent_id = args.agent_id
-            
+
             # Find agent configuration
             config = self._load_agent_config(agent_id)
-            
+
             if not config:
                 return CommandResult.error_result(f"Agent '{agent_id}' not found")
-                
+
             # Validate configuration
             errors = self.builder_service.validate_configuration(config)
-            
+
             if errors:
                 return CommandResult.error_result(
-                    f"Validation failed:\n" + "\n".join(f"  - {e}" for e in errors)
+                    "Validation failed:\n" + "\n".join(f"  - {e}" for e in errors)
                 )
-                
+
             # Check for conflicts
             conflicts = self._check_conflicts(agent_id)
-            
-            if conflicts:
-                warning = f"Warning: Agent overrides {conflicts}"
-            else:
-                warning = ""
-                
+
+            warning = f"Warning: Agent overrides {conflicts}" if conflicts else ""
+
             return CommandResult.success_result(
                 f"Agent '{agent_id}' configuration is valid. {warning}"
             )
-            
+
         except Exception as e:
             return CommandResult.error_result(f"Test failed: {e}")
-            
+
     def _list_templates(self, args) -> CommandResult:
         """List available agent templates."""
         templates = self.builder_service.list_available_templates()
-        
-        output_format = getattr(args, 'format', 'text')
+
+        output_format = getattr(args, "format", "text")
         if output_format == "json":
             return CommandResult.success_result("Templates listed", data=templates)
-        else:
-            output = "Available Agent Templates:\n\n"
-            for template in templates:
-                template_id = template.get('id', 'unknown')
-                template_name = template.get('name', 'Unnamed')
-                output += f"  {template_id:<20} - {template_name}\n"
-                if template.get('description'):
-                    output += f"    {template['description']}\n"
-            return CommandResult.success_result(output)
-            
+        output = "Available Agent Templates:\n\n"
+        for template in templates:
+            template_id = template.get("id", "unknown")
+            template_name = template.get("name", "Unnamed")
+            output += f"  {template_id:<20} - {template_name}\n"
+            if template.get("description"):
+                output += f"    {template['description']}\n"
+        return CommandResult.success_result(output)
+
     def _interactive_create(self) -> CommandResult:
         """Interactive agent creation wizard."""
         print("\n=== Agent Creation Wizard ===\n")
-        
+
         # Get agent ID
         agent_id = input("Agent ID (lowercase, hyphens only): ").strip()
         if not agent_id:
             return CommandResult.error_result("Agent ID is required")
-            
+
         # Get name
         name = input(f"Display name [{agent_id}]: ").strip() or agent_id
-        
+
         # Get description
         description = input("Description: ").strip()
         if not description:
             return CommandResult.error_result("Description is required")
-            
+
         # Get model
         print("\nAvailable models: sonnet, opus, haiku")
         model = input("Model [sonnet]: ").strip() or "sonnet"
-        
+
         # Get tool choice
         print("\nTool choices: auto, required, any, none")
         tool_choice = input("Tool choice [auto]: ").strip() or "auto"
-        
+
         # Create agent
         try:
             config, instructions = self.builder_service.create_agent(
@@ -359,49 +362,52 @@ class AgentManagerCommand(AgentCommand):
                 name=name,
                 description=description,
                 model=model,
-                tool_choice=tool_choice
+                tool_choice=tool_choice,
             )
-            
+
             # Save agent
             if self._save_agent(config, instructions, agent_id):
-                return CommandResult.success_result(f"\nAgent '{agent_id}' created successfully!")
-            else:
-                return CommandResult.error_result("Failed to save agent files")
-                
+                return CommandResult.success_result(
+                    f"\nAgent '{agent_id}' created successfully!"
+                )
+            return CommandResult.error_result("Failed to save agent files")
+
         except Exception as e:
             return CommandResult.error_result(f"Creation failed: {e}")
-            
-    def _save_agent(self, config: Dict[str, Any], instructions: str, agent_id: str) -> bool:
+
+    def _save_agent(
+        self, config: Dict[str, Any], instructions: str, agent_id: str
+    ) -> bool:
         """Save agent configuration and instructions.
-        
+
         Args:
             config: Agent configuration dictionary
             instructions: Agent instructions markdown
             agent_id: Agent identifier
-            
+
         Returns:
             True if saved successfully
         """
         try:
             template_dir = Path(__file__).parent.parent.parent / "agents" / "templates"
             template_dir.mkdir(parents=True, exist_ok=True)
-            
+
             # Save JSON configuration
             json_file = template_dir / f"{agent_id}.json"
-            with open(json_file, 'w') as f:
+            with open(json_file, "w") as f:
                 json.dump(config, f, indent=2)
-                
+
             # Save instructions
             md_file = template_dir / f"{agent_id}.md"
             md_file.write_text(instructions)
-            
+
             self.logger.info(f"Agent saved: {json_file} and {md_file}")
             return True
-            
+
         except Exception as e:
             self.logger.error(f"Failed to save agent: {e}")
             return False
-            
+
     def _read_agent_summary(self, agent_file: Path, tier: str) -> Dict[str, Any]:
         """Read agent summary from file."""
         try:
@@ -411,68 +417,68 @@ class AgentManagerCommand(AgentCommand):
                 "id": agent_id,
                 "name": agent_id.replace("-", " ").title(),
                 "tier": tier,
-                "file": str(agent_file)
+                "file": str(agent_file),
             }
         except Exception:
             return {}
-            
+
     def _format_agent_list(self, agents: Dict[str, List]) -> str:
         """Format agent list for display."""
         output = "=== Agent Hierarchy ===\n\n"
-        
+
         # Project agents
         if agents["project"]:
             output += "[P] PROJECT LEVEL (Highest Priority)\n"
             for agent in agents["project"]:
                 output += f"    {agent['id']:<20} - {agent.get('name', agent['id'])}\n"
             output += "\n"
-            
+
         # User agents
         if agents["user"]:
             output += "[U] USER LEVEL\n"
             for agent in agents["user"]:
                 output += f"    {agent['id']:<20} - {agent.get('name', agent['id'])}\n"
             output += "\n"
-            
+
         # System agents
         if agents["system"]:
             output += "[S] SYSTEM LEVEL (Framework Defaults)\n"
             for agent in agents["system"]:
                 output += f"    {agent['id']:<20} - {agent.get('name', agent['id'])}\n"
-                
+
         return output
-        
+
     def _find_agent(self, agent_id: str) -> Optional[Dict[str, Any]]:
         """Find agent across all tiers."""
         # Implementation would search across tiers
         # This is a simplified version
         return {"id": agent_id, "tier": "system"}
-        
+
     def _load_agent_config(self, agent_id: str) -> Optional[Dict[str, Any]]:
         """Load agent configuration."""
         try:
             return self.builder_service._load_template(agent_id)
         except:
             return None
-            
+
     def _check_conflicts(self, agent_id: str) -> Optional[str]:
         """Check for agent conflicts across tiers."""
         # Check if agent exists in multiple tiers
         # Return tier information if conflicts exist
         return None
-        
+
     def _format_agent_details(self, agent_info: Dict[str, Any]) -> str:
         """Format agent details for display."""
         output = f"=== Agent: {agent_info['id']} ===\n\n"
         for key, value in agent_info.items():
             output += f"{key}: {value}\n"
         return output
-        
+
     def _load_pm_template(self, template_name: str) -> str:
         """Load PM instruction template."""
         # Load predefined PM templates
         return "# PM Instructions Template\n"
-        
+
     def _show_help(self) -> CommandResult:
         """Show help for agent manager."""
         help_text = """
@@ -503,21 +509,20 @@ Note: PM customization writes to .claude-mpm/INSTRUCTIONS.md, not CLAUDE.md
 # Module-level function for CLI integration
 def manage_agent_manager(args) -> int:
     """Entry point for agent-manager command from CLI.
-    
+
     Args:
         args: Parsed command line arguments
-        
+
     Returns:
         Exit code (0 for success, non-zero for failure)
     """
     command = AgentManagerCommand()
     result = command.run(args)
-    
+
     if result.success:
         if result.message:
             print(result.message)
         return 0
-    else:
-        if result.message:
-            print(f"Error: {result.message}", file=sys.stderr)
-        return 1
+    if result.message:
+        print(f"Error: {result.message}", file=sys.stderr)
+    return 1