claude-mpm 3.9.9__py3-none-any.whl → 4.0.3__py3-none-any.whl

claude_mpm/services/agents/management/agent_capabilities_generator.py

@@ -4,8 +4,8 @@ This service generates markdown content for agent capabilities section
 from discovered deployed agents.
 """
 
-from typing import List, Dict, Any
 import logging
+from typing import Any, Dict, List
 
 from jinja2 import Template
 
@@ -14,123 +14,137 @@ logger = logging.getLogger(__name__)
 
 class AgentCapabilitiesGenerator:
     """Generates markdown content for agent capabilities section."""
-    
+
     def __init__(self):
         """Initialize the generator with default template."""
         self.template = self._load_template()
         logger.debug("Initialized AgentCapabilitiesGenerator")
-    
-    def generate_capabilities_section(self, deployed_agents: List[Dict[str, Any]]) -> str:
+
+    def generate_capabilities_section(
+        self, deployed_agents: List[Dict[str, Any]]
+    ) -> str:
         """Generate the complete agent capabilities markdown section.
-        
+
         Args:
             deployed_agents: List of agent information dictionaries
-            
+
         Returns:
             Generated markdown content for agent capabilities
         """
         try:
             # Group agents by source tier for organized display
             agents_by_tier = self._group_by_tier(deployed_agents)
-            
+
             # Generate core agent list
             core_agent_list = self._generate_core_agent_list(deployed_agents)
-            
+
             # Generate detailed capabilities
-            detailed_capabilities = self._generate_detailed_capabilities(deployed_agents)
-            
+            detailed_capabilities = self._generate_detailed_capabilities(
+                deployed_agents
+            )
+
             # Render template
             content = self.template.render(
                 core_agents=core_agent_list,
                 detailed_capabilities=detailed_capabilities,
                 agents_by_tier=agents_by_tier,
-                total_agents=len(deployed_agents)
+                total_agents=len(deployed_agents),
             )
-            
+
             return content
-            
+
         except Exception as e:
             logger.error(f"Failed to generate capabilities section: {e}")
             # Return fallback content on error
             return self._generate_fallback_content()
-    
-    def _group_by_tier(self, agents: List[Dict[str, Any]]) -> Dict[str, List[Dict[str, Any]]]:
+
+    def _group_by_tier(
+        self, agents: List[Dict[str, Any]]
+    ) -> Dict[str, List[Dict[str, Any]]]:
         """Group agents by their source tier.
-        
+
         Args:
             agents: List of agent information dictionaries
-            
+
         Returns:
             Dictionary mapping tiers to lists of agents
         """
-        tiers = {'system': [], 'user': [], 'project': []}
-        
+        tiers = {"system": [], "user": [], "project": []}
+
         for agent in agents:
-            tier = agent.get('source_tier', 'system')
+            tier = agent.get("source_tier", "system")
             if tier in tiers:
                 tiers[tier].append(agent)
             else:
                 # Handle unknown tiers gracefully
-                tiers['system'].append(agent)
-                logger.warning(f"Unknown source tier '{tier}' for agent {agent.get('id')}, defaulting to system")
-        
+                tiers["system"].append(agent)
+                logger.warning(
+                    f"Unknown source tier '{tier}' for agent {agent.get('id')}, defaulting to system"
+                )
+
         return tiers
-    
+
     def _generate_core_agent_list(self, agents: List[Dict[str, Any]]) -> str:
         """Generate comma-separated list of core agent IDs.
-        
+
         Args:
             agents: List of agent information dictionaries
-            
+
         Returns:
             Comma-separated string of agent IDs
         """
-        agent_ids = [agent['id'] for agent in agents]
-        return ', '.join(sorted(agent_ids))
-    
-    def _generate_detailed_capabilities(self, agents: List[Dict[str, Any]]) -> List[Dict[str, str]]:
+        agent_ids = [agent["id"] for agent in agents]
+        return ", ".join(sorted(agent_ids))
+
+    def _generate_detailed_capabilities(
+        self, agents: List[Dict[str, Any]]
+    ) -> List[Dict[str, str]]:
         """Generate detailed capability descriptions for each agent.
-        
+
         Args:
             agents: List of agent information dictionaries
-            
+
         Returns:
             List of capability dictionaries for template rendering
         """
         capabilities = []
-        
-        for agent in sorted(agents, key=lambda a: a['id']):
+
+        for agent in sorted(agents, key=lambda a: a["id"]):
             # Extract key capabilities
-            specializations = agent.get('specializations', [])
-            when_to_use = agent.get('capabilities', {}).get('when_to_use', [])
-            
+            specializations = agent.get("specializations", [])
+            when_to_use = agent.get("capabilities", {}).get("when_to_use", [])
+
             # Create capability summary
             if when_to_use:
-                capability_text = '; '.join(when_to_use[:2])  # First 2 items
+                capability_text = "; ".join(when_to_use[:2])  # First 2 items
             elif specializations:
-                capability_text = ', '.join(specializations[:3])  # First 3 specializations
+                capability_text = ", ".join(
+                    specializations[:3]
+                )  # First 3 specializations
             else:
-                capability_text = agent.get('description', 'General purpose agent')
-            
+                capability_text = agent.get("description", "General purpose agent")
+
             # Truncate long capability text
             if len(capability_text) > 100:
-                capability_text = capability_text[:97] + '...'
-            
+                capability_text = capability_text[:97] + "..."
+
             # Clean up the agent name for TodoWrite usage
-            clean_name = agent['name'].replace(' Agent', '').replace('-', ' ')
-            
-            capabilities.append({
-                'name': clean_name,  # Clean name for TodoWrite
-                'id': agent['id'],    # Agent ID for Task tool
-                'capability_text': capability_text,
-                'tools': ', '.join(agent.get('tools', [])[:5])  # First 5 tools
-            })
-        
+            clean_name = agent["name"].replace(" Agent", "").replace("-", " ")
+
+            capabilities.append(
+                {
+                    "name": clean_name,  # Clean name for TodoWrite
+                    "id": agent["id"],  # Agent ID for Task tool
+                    "capability_text": capability_text,
+                    "tools": ", ".join(agent.get("tools", [])[:5]),  # First 5 tools
+                }
+            )
+
         return capabilities
-    
+
     def _load_template(self) -> Template:
         """Load the Jinja2 template for agent capabilities.
-        
+
         Returns:
             Configured Jinja2 template
         """
@@ -163,12 +177,12 @@ You have the following specialized agents available for delegation:
 **Total Available Agents**: {{ total_agents }}
 Use the agent ID in parentheses when delegating tasks via the Task tool.
 """.strip()
-        
+
         return Template(template_content)
-    
+
     def _generate_fallback_content(self) -> str:
         """Generate fallback content when agent discovery fails.
-        
+
         Returns:
             Static fallback markdown content
         """
@@ -188,4 +202,4 @@ Use the agent ID in parentheses when delegating tasks via the Task tool.
 - **Data Engineer**: Data pipelines, ETL, database operations
 
 *Note: Unable to dynamically generate agent list. Using default agents.*
-""".strip()
+""".strip()