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

claude_mpm/agents/agent_loader_integration.py

@@ -8,55 +8,52 @@ Provides backward compatibility while enabling advanced features.
 """
 
 import logging
-from typing import Optional, Dict, Any
+from typing import Any, Dict, Optional
 
-from .agent_loader import (
-    load_agent_prompt_from_md,
-    get_agent_prompt
-)
-from ..services import AgentManager
 from ..models.agent_definition import AgentDefinition
+from ..services import AgentManager
+from .agent_loader import get_agent_prompt, load_agent_prompt_from_md
 
 logger = logging.getLogger(__name__)
 
 
 class EnhancedAgentLoader:
     """Enhanced agent loader with management capabilities."""
-    
+
     def __init__(self):
         """Initialize enhanced loader."""
         self.manager = AgentManager()
-        
+
     def get_agent_definition(self, agent_name: str) -> Optional[AgentDefinition]:
         """
         Get full agent definition with structured data.
-        
+
         Args:
             agent_name: Agent name (e.g., 'documentation', 'ticketing')
-            
+
         Returns:
             AgentDefinition or None
         """
         # Map from old naming to new naming if needed
         # Since AGENT_MAPPINGS no longer exists, use direct naming
         agent_file_name = agent_name
-        
+
         return self.manager.read_agent(agent_file_name)
-    
+
     def get_agent_metadata(self, agent_name: str) -> Optional[Dict[str, Any]]:
         """
         Get agent metadata only.
-        
+
         Args:
             agent_name: Agent name
-            
+
         Returns:
             Metadata dict or None
         """
         agent_def = self.get_agent_definition(agent_name)
         if not agent_def:
             return None
-            
+
         return {
             "name": agent_def.name,
             "title": agent_def.title,
@@ -64,80 +61,83 @@ class EnhancedAgentLoader:
             "type": agent_def.metadata.type.value,
             "model_preference": agent_def.metadata.model_preference,
             "specializations": agent_def.metadata.specializations,
-            "last_updated": agent_def.metadata.last_updated.isoformat() 
-                if agent_def.metadata.last_updated else None
+            "last_updated": (
+                agent_def.metadata.last_updated.isoformat()
+                if agent_def.metadata.last_updated
+                else None
+            ),
         }
-    
+
     def get_agent_capabilities(self, agent_name: str) -> Optional[Dict[str, Any]]:
         """
         Get agent capabilities and permissions.
-        
+
         Args:
             agent_name: Agent name
-            
+
         Returns:
             Capabilities dict or None
         """
         agent_def = self.get_agent_definition(agent_name)
         if not agent_def:
             return None
-            
+
         return {
             "capabilities": agent_def.capabilities,
             "authority": {
                 "write_access": agent_def.authority.exclusive_write_access,
                 "forbidden": agent_def.authority.forbidden_operations,
-                "read_access": agent_def.authority.read_access
+                "read_access": agent_def.authority.read_access,
             },
             "workflows": [w.name for w in agent_def.workflows],
-            "tools_available": bool(agent_def.tools_commands)
+            "tools_available": bool(agent_def.tools_commands),
         }
-    
+
     def get_agent_selection_criteria(self, agent_name: str) -> Optional[Dict[str, Any]]:
         """
         Get agent selection criteria for routing.
-        
+
         Args:
             agent_name: Agent name
-            
+
         Returns:
             Selection criteria dict or None
         """
         agent_def = self.get_agent_definition(agent_name)
         if not agent_def:
             return None
-            
+
         return {
             "when_to_select": agent_def.when_to_use.get("select", []),
             "when_not_to_select": agent_def.when_to_use.get("do_not_select", []),
             "specializations": agent_def.metadata.specializations,
             "dependencies": agent_def.dependencies,
-            "escalation_triggers": agent_def.escalation_triggers
+            "escalation_triggers": agent_def.escalation_triggers,
         }
-    
+
     def update_agent_version(self, agent_name: str) -> bool:
         """
         Increment agent version (serial).
-        
+
         Args:
             agent_name: Agent name
-            
+
         Returns:
             True if updated, False otherwise
         """
         # Map name - since AGENT_MAPPINGS no longer exists, use direct naming
         agent_file_name = agent_name
-        
+
         result = self.manager.update_agent(agent_file_name, {}, increment_version=True)
         return result is not None
-    
+
     def refresh_agent_prompt(self, agent_name: str) -> str:
         """
         Force refresh agent prompt from file.
-        
+
         Args:
             agent_name: Agent name
-            
+
         Returns:
             Agent prompt with base instructions
         """
@@ -174,7 +174,7 @@ def check_agent_capability(agent_name: str, capability_keyword: str) -> bool:
     caps = get_enhanced_loader().get_agent_capabilities(agent_name)
     if not caps:
         return False
-        
+
     # Check in capabilities list
     for cap in caps["capabilities"]:
         if capability_keyword.lower() in cap.lower():
@@ -185,42 +185,42 @@ def check_agent_capability(agent_name: str, capability_keyword: str) -> bool:
 def should_select_agent(agent_name: str, task_description: str) -> Dict[str, Any]:
     """
     Determine if an agent should be selected for a task.
-    
+
     Args:
         agent_name: Agent name
         task_description: Task description
-        
+
     Returns:
         Dict with selection recommendation and reasons
     """
     criteria = get_enhanced_loader().get_agent_selection_criteria(agent_name)
     if not criteria:
         return {"should_select": False, "reason": "Agent not found"}
-    
+
     task_lower = task_description.lower()
-    
+
     # Check positive matches
     positive_matches = []
     for criterion in criteria["when_to_select"]:
         # Extract keywords from criterion
         if "Keywords:" in criterion:
             keywords = criterion.split("Keywords:")[1].strip()
-            for keyword in keywords.split(','):
+            for keyword in keywords.split(","):
                 keyword = keyword.strip().strip('"').strip("'")
                 if keyword.lower() in task_lower:
                     positive_matches.append(f"Keyword match: {keyword}")
         elif any(word in task_lower for word in criterion.lower().split()):
             positive_matches.append(f"Criterion match: {criterion[:50]}...")
-    
+
     # Check negative matches
     negative_matches = []
     for criterion in criteria["when_not_to_select"]:
         if any(word in task_lower for word in criterion.lower().split()[:5]):
             negative_matches.append(f"Exclusion: {criterion[:50]}...")
-    
+
     return {
         "should_select": len(positive_matches) > 0 and len(negative_matches) == 0,
         "positive_matches": positive_matches,
         "negative_matches": negative_matches,
-        "specializations": criteria["specializations"]
-    }
+        "specializations": criteria["specializations"],
+    }

claude_mpm/agents/agents_metadata.py

@@ -19,17 +19,16 @@ DOCUMENTATION_CONFIG = {
         "api_documentation",
         "version_documentation",
         "operational_docs",
-        "quality_assurance"
+        "quality_assurance",
     ],
     "primary_interface": "documentation_management",
     "performance_targets": {
         "changelog_generation": "5m",
         "documentation_update": "24h",
-        "coverage_target": "90%"
-    }
+        "coverage_target": "90%",
+    },
 }
 
-
 # Version Control Agent Metadata
 VERSION_CONTROL_CONFIG = {
     "name": "version_control_agent",
@@ -42,15 +41,15 @@ VERSION_CONTROL_CONFIG = {
         "semantic_versioning",
         "tag_management",
         "release_coordination",
-        "version_file_updates"
+        "version_file_updates",
     ],
     "primary_interface": "git_cli",
     "performance_targets": {
         "branch_creation": "5s",
         "merge_operation": "30s",
         "version_bump": "10s",
-        "conflict_resolution": "5m"
-    }
+        "conflict_resolution": "5m",
+    },
 }
 
 # QA Agent Metadata
@@ -65,15 +64,15 @@ QA_CONFIG = {
         "performance_testing",
         "security_testing",
         "regression_testing",
-        "test_automation"
+        "test_automation",
     ],
     "primary_interface": "testing_framework",
     "performance_targets": {
         "unit_test_suite": "5m",
         "integration_tests": "15m",
         "full_test_suite": "30m",
-        "coverage_target": "80%"
-    }
+        "coverage_target": "80%",
+    },
 }
 
 # Research Agent Metadata
@@ -88,14 +87,14 @@ RESEARCH_CONFIG = {
         "performance_analysis",
         "security_research",
         "market_analysis",
-        "feasibility_studies"
+        "feasibility_studies",
     ],
     "primary_interface": "research_tools",
     "performance_targets": {
         "quick_research": "15m",
         "deep_analysis": "2h",
-        "comprehensive_report": "24h"
-    }
+        "comprehensive_report": "24h",
+    },
 }
 
 # Ops Agent Metadata
@@ -110,15 +109,15 @@ OPS_CONFIG = {
         "ci_cd_pipeline",
         "containerization",
         "cloud_services",
-        "performance_optimization"
+        "performance_optimization",
     ],
     "primary_interface": "deployment_tools",
     "performance_targets": {
         "deployment": "10m",
         "rollback": "5m",
         "infrastructure_update": "30m",
-        "monitoring_setup": "1h"
-    }
+        "monitoring_setup": "1h",
+    },
 }
 
 # Security Agent Metadata
@@ -133,15 +132,15 @@ SECURITY_CONFIG = {
         "code_security_review",
         "dependency_scanning",
         "security_patching",
-        "compliance_checking"
+        "compliance_checking",
     ],
     "primary_interface": "security_tools",
     "performance_targets": {
         "quick_scan": "10m",
         "full_audit": "2h",
         "penetration_test": "4h",
-        "dependency_scan": "30m"
-    }
+        "dependency_scan": "30m",
+    },
 }
 
 # Engineer Agent Metadata
@@ -156,15 +155,15 @@ ENGINEER_CONFIG = {
         "code_refactoring",
         "performance_optimization",
         "api_development",
-        "database_design"
+        "database_design",
     ],
     "primary_interface": "development_tools",
     "performance_targets": {
         "feature_implementation": "4h",
         "bug_fix": "1h",
         "code_review": "30m",
-        "refactoring": "2h"
-    }
+        "refactoring": "2h",
+    },
 }
 
 # Data Engineer Agent Metadata
@@ -180,15 +179,15 @@ DATA_ENGINEER_CONFIG = {
         "data_migration",
         "api_key_management",
         "data_analytics",
-        "schema_design"
+        "schema_design",
     ],
     "primary_interface": "data_management_tools",
     "performance_targets": {
         "pipeline_setup": "2h",
         "data_migration": "4h",
         "api_integration": "1h",
-        "schema_update": "30m"
-    }
+        "schema_update": "30m",
+    },
 }
 
 # Project Organizer Agent Metadata
@@ -203,15 +202,15 @@ PROJECT_ORGANIZER_CONFIG = {
         "convention_enforcement",
         "batch_reorganization",
         "framework_recognition",
-        "documentation_maintenance"
+        "documentation_maintenance",
     ],
     "primary_interface": "organization_tools",
     "performance_targets": {
         "structure_analysis": "2m",
         "file_placement": "10s",
         "validation_scan": "5m",
-        "batch_reorganization": "15m"
-    }
+        "batch_reorganization": "15m",
+    },
 }
 
 # Aggregate all configs for easy access
@@ -224,5 +223,5 @@ ALL_AGENT_CONFIGS = {
     "security": SECURITY_CONFIG,
     "engineer": ENGINEER_CONFIG,
     "data_engineer": DATA_ENGINEER_CONFIG,
-    "project_organizer": PROJECT_ORGANIZER_CONFIG
-}
+    "project_organizer": PROJECT_ORGANIZER_CONFIG,
+}