claude-mpm 4.7.4__py3-none-any.whl → 4.18.2__py3-none-any.whl

claude_mpm/skills/skill_manager.py

@@ -0,0 +1,310 @@
+"""Skills manager - integrates skills with agents."""
+
+import json
+from pathlib import Path
+from typing import Dict, List, Optional
+
+from claude_mpm.core.logging_utils import get_logger
+
+from .registry import Skill, get_registry
+
+logger = get_logger(__name__)
+
+
+class SkillManager:
+    """Manages skills and their integration with agents."""
+
+    def __init__(self):
+        """Initialize the skill manager."""
+        self.registry = get_registry()
+        self.agent_skill_mapping: Dict[str, List[str]] = {}
+        self._load_agent_mappings()
+
+    def _load_agent_mappings(self):
+        """Load skill mappings from agent templates."""
+        # Load mappings from agent JSON templates that have 'skills' field
+        agent_templates_dir = Path(__file__).parent.parent / "agents" / "templates"
+
+        if not agent_templates_dir.exists():
+            logger.warning(
+                f"Agent templates directory not found: {agent_templates_dir}"
+            )
+            return
+
+        mapping_count = 0
+        for template_file in agent_templates_dir.glob("*.json"):
+            try:
+                with open(template_file, encoding="utf-8") as f:
+                    agent_data = json.load(f)
+
+                agent_id = agent_data.get("agent_id") or agent_data.get("agent_type")
+                if not agent_id:
+                    continue
+
+                # Extract skills list if present
+                skills = agent_data.get("skills", [])
+                if skills:
+                    self.agent_skill_mapping[agent_id] = skills
+                    mapping_count += 1
+                    logger.debug(
+                        f"Agent '{agent_id}' mapped to {len(skills)} skills: {', '.join(skills)}"
+                    )
+
+            except Exception as e:
+                logger.error(f"Error loading agent mapping from {template_file}: {e}")
+
+        if mapping_count > 0:
+            logger.info(f"Loaded skill mappings for {mapping_count} agents")
+
+    def get_agent_skills(self, agent_type: str) -> List[Skill]:
+        """
+        Get all skills for an agent (bundled + discovered).
+
+        Args:
+            agent_type: Agent type/ID (e.g., 'engineer', 'python_engineer')
+
+        Returns:
+            List of Skill objects for this agent
+        """
+        skill_names = self.agent_skill_mapping.get(agent_type, [])
+
+        # Get skills from registry
+        skills = []
+        for name in skill_names:
+            skill = self.registry.get_skill(name)
+            if skill:
+                skills.append(skill)
+            else:
+                logger.warning(
+                    f"Skill '{name}' referenced by agent '{agent_type}' not found"
+                )
+
+        # Also include skills that have no agent restriction
+        # or explicitly list this agent type
+        additional_skills = self.registry.get_skills_for_agent(agent_type)
+        for skill in additional_skills:
+            if skill not in skills:
+                skills.append(skill)
+
+        return skills
+
+    def enhance_agent_prompt(
+        self, agent_type: str, base_prompt: str, include_all: bool = False
+    ) -> str:
+        """
+        Enhance agent prompt with available skills.
+
+        Args:
+            agent_type: Agent type/ID
+            base_prompt: Original agent prompt
+            include_all: If True, include all available skills regardless of mapping
+
+        Returns:
+            Enhanced prompt with skills section appended
+        """
+        if include_all:
+            skills = self.registry.list_skills()
+        else:
+            skills = self.get_agent_skills(agent_type)
+
+        if not skills:
+            return base_prompt
+
+        # Build skills section
+        skills_section = "\n\n" + "=" * 80 + "\n"
+        skills_section += "## 🎯 Available Skills\n\n"
+        skills_section += f"You have access to {len(skills)} specialized skills:\n\n"
+
+        for skill in skills:
+            skills_section += f"### 📚 {skill.name.replace('-', ' ').title()}\n\n"
+            skills_section += f"**Source:** {skill.source}\n"
+            if skill.description:
+                skills_section += f"**Description:** {skill.description}\n"
+            skills_section += "\n```\n"
+            skills_section += skill.content
+            skills_section += "\n```\n\n"
+
+        skills_section += "=" * 80 + "\n"
+
+        return base_prompt + skills_section
+
+    def list_agent_skill_mappings(self) -> Dict[str, List[str]]:
+        """
+        Get all agent-to-skill mappings.
+
+        Returns:
+            Dictionary mapping agent IDs to lists of skill names
+        """
+        return self.agent_skill_mapping.copy()
+
+    def add_skill_to_agent(self, agent_type: str, skill_name: str) -> bool:
+        """
+        Add a skill to an agent's mapping.
+
+        Args:
+            agent_type: Agent type/ID
+            skill_name: Name of the skill to add
+
+        Returns:
+            True if successful, False if skill not found
+        """
+        skill = self.registry.get_skill(skill_name)
+        if not skill:
+            logger.error(f"Cannot add skill '{skill_name}': skill not found")
+            return False
+
+        if agent_type not in self.agent_skill_mapping:
+            self.agent_skill_mapping[agent_type] = []
+
+        if skill_name not in self.agent_skill_mapping[agent_type]:
+            self.agent_skill_mapping[agent_type].append(skill_name)
+            logger.info(f"Added skill '{skill_name}' to agent '{agent_type}'")
+
+        return True
+
+    def remove_skill_from_agent(self, agent_type: str, skill_name: str) -> bool:
+        """
+        Remove a skill from an agent's mapping.
+
+        Args:
+            agent_type: Agent type/ID
+            skill_name: Name of the skill to remove
+
+        Returns:
+            True if successful, False if not found
+        """
+        if agent_type not in self.agent_skill_mapping:
+            return False
+
+        if skill_name in self.agent_skill_mapping[agent_type]:
+            self.agent_skill_mapping[agent_type].remove(skill_name)
+            logger.info(f"Removed skill '{skill_name}' from agent '{agent_type}'")
+            return True
+
+        return False
+
+    def reload(self):
+        """Reload skills and agent mappings."""
+        logger.info("Reloading skill manager...")
+        self.registry.reload()
+        self.agent_skill_mapping.clear()
+        self._load_agent_mappings()
+        logger.info("Skill manager reloaded")
+
+    def infer_agents_for_skill(self, skill) -> List[str]:
+        """Infer which agents should have this skill based on tags/name.
+
+        Args:
+            skill: Skill object to analyze
+
+        Returns:
+            List of agent IDs that should have this skill
+        """
+        agents = []
+        content_lower = skill.content.lower()
+        name_lower = skill.name.lower()
+
+        # Python-related
+        if any(
+            tag in content_lower or tag in name_lower
+            for tag in ["python", "django", "flask", "fastapi"]
+        ):
+            agents.append("python-engineer")
+
+        # TypeScript/JavaScript-related
+        if any(
+            tag in content_lower or tag in name_lower
+            for tag in ["typescript", "javascript", "react", "next", "vue", "node"]
+        ):
+            agents.extend(["typescript-engineer", "react-engineer", "nextjs-engineer"])
+
+        # Go-related
+        if any(tag in content_lower or tag in name_lower for tag in ["golang", "go "]):
+            agents.append("golang-engineer")
+
+        # Ops-related
+        if any(
+            tag in content_lower or tag in name_lower
+            for tag in ["docker", "kubernetes", "deploy", "devops", "ops"]
+        ):
+            agents.extend(["ops", "devops", "local-ops"])
+
+        # Testing/QA-related
+        if any(
+            tag in content_lower or tag in name_lower
+            for tag in ["test", "qa", "quality", "assert"]
+        ):
+            agents.extend(["qa", "web-qa", "api-qa"])
+
+        # Documentation-related
+        if any(
+            tag in content_lower or tag in name_lower
+            for tag in ["documentation", "docs", "api doc", "openapi"]
+        ):
+            agents.extend(["docs", "documentation", "technical-writer"])
+
+        # Remove duplicates
+        return list(set(agents))
+
+    def save_mappings_to_config(self, config_path: Optional[Path] = None):
+        """Save current agent-skill mappings to configuration file.
+
+        Args:
+            config_path: Path to configuration file. If None, uses default.
+        """
+        import json
+
+        if config_path is None:
+            config_path = Path.cwd() / ".claude-mpm" / "skills_config.json"
+
+        # Ensure directory exists
+        config_path.parent.mkdir(parents=True, exist_ok=True)
+
+        # Save mappings
+        with open(config_path, "w", encoding="utf-8") as f:
+            json.dump(self.agent_skill_mapping, f, indent=2)
+
+        logger.info(f"Saved skill mappings to {config_path}")
+
+    def load_mappings_from_config(self, config_path: Optional[Path] = None):
+        """Load agent-skill mappings from configuration file.
+
+        Args:
+            config_path: Path to configuration file. If None, uses default.
+        """
+        import json
+
+        if config_path is None:
+            config_path = Path.cwd() / ".claude-mpm" / "skills_config.json"
+
+        if not config_path.exists():
+            logger.debug(f"No skill mappings config found at {config_path}")
+            return
+
+        try:
+            with open(config_path, encoding="utf-8") as f:
+                loaded_mappings = json.load(f)
+
+            # Merge with existing mappings
+            for agent_id, skills in loaded_mappings.items():
+                if agent_id not in self.agent_skill_mapping:
+                    self.agent_skill_mapping[agent_id] = []
+                for skill in skills:
+                    if skill not in self.agent_skill_mapping[agent_id]:
+                        self.agent_skill_mapping[agent_id].append(skill)
+
+            logger.info(f"Loaded skill mappings from {config_path}")
+        except Exception as e:
+            logger.error(f"Error loading skill mappings from {config_path}: {e}")
+
+
+# Global manager instance (singleton pattern)
+_manager: Optional[SkillManager] = None
+
+
+def get_manager() -> SkillManager:
+    """Get the global skill manager (singleton)."""
+    global _manager
+    if _manager is None:
+        _manager = SkillManager()
+    return _manager

claude_mpm/storage/state_storage.py

@@ -86,7 +86,7 @@ class StateStorage:
             return True
 
         except Exception as e:
-            self.logger.error(f"Failed to write JSON to {file_path}: {e}")
+            logger.error(f"Failed to write JSON to {file_path}: {e}")
             self.error_count += 1
             return False
 
@@ -106,7 +106,7 @@ class StateStorage:
             file_path = Path(file_path)
 
             if not file_path.exists():
-                self.logger.debug(f"File not found: {file_path}")
+                logger.debug(f"File not found: {file_path}")
                 return None
 
             # Auto-detect compression
@@ -125,7 +125,7 @@ class StateStorage:
             return data
 
         except Exception as e:
-            self.logger.error(f"Failed to read JSON from {file_path}: {e}")
+            logger.error(f"Failed to read JSON from {file_path}: {e}")
             self.error_count += 1
             return None
 
@@ -166,7 +166,7 @@ class StateStorage:
             return True
 
         except Exception as e:
-            self.logger.error(f"Failed to write pickle to {file_path}: {e}")
+            logger.error(f"Failed to write pickle to {file_path}: {e}")
             self.error_count += 1
             return False
 
@@ -186,7 +186,7 @@ class StateStorage:
             file_path = Path(file_path)
 
             if not file_path.exists():
-                self.logger.debug(f"File not found: {file_path}")
+                logger.debug(f"File not found: {file_path}")
                 return None
 
             # Auto-detect compression
@@ -205,7 +205,7 @@ class StateStorage:
             return data
 
         except Exception as e:
-            self.logger.error(f"Failed to read pickle from {file_path}: {e}")
+            logger.error(f"Failed to read pickle from {file_path}: {e}")
             self.error_count += 1
             return None
 
@@ -262,7 +262,7 @@ class StateStorage:
                 Path(temp_path).replace(file_path)
 
                 self.write_count += 1
-                self.logger.debug(f"Atomic write successful: {file_path}")
+                logger.debug(f"Atomic write successful: {file_path}")
                 return True
 
             finally:
@@ -271,7 +271,7 @@ class StateStorage:
                     Path(temp_path).unlink()
 
         except Exception as e:
-            self.logger.error(f"Atomic write failed for {file_path}: {e}")
+            logger.error(f"Atomic write failed for {file_path}: {e}")
             self.error_count += 1
             return False
 
@@ -339,7 +339,7 @@ class StateStorage:
                 f.write(checksum)
 
         except Exception as e:
-            self.logger.warning(f"Could not add checksum: {e}")
+            logger.warning(f"Could not add checksum: {e}")
 
     def verify_checksum(self, file_path: Union[str, Path]) -> bool:
         """Verify file checksum for integrity.
@@ -370,13 +370,13 @@ class StateStorage:
             actual = hasher.hexdigest()
 
             if actual != expected:
-                self.logger.error(f"Checksum mismatch for {file_path}")
+                logger.error(f"Checksum mismatch for {file_path}")
                 return False
 
             return True
 
         except Exception as e:
-            self.logger.warning(f"Could not verify checksum: {e}")
+            logger.warning(f"Could not verify checksum: {e}")
             return True  # Assume valid if can't verify
 
     def cleanup_temp_files(self) -> int:
@@ -396,7 +396,7 @@ class StateStorage:
                     if age > 3600:
                         temp_file.unlink()
                         cleaned += 1
-                        self.logger.debug(f"Cleaned up temp file: {temp_file}")
+                        logger.debug(f"Cleaned up temp file: {temp_file}")
                 except Exception:
                     pass
 
@@ -408,12 +408,12 @@ class StateStorage:
                     cleaned += 1
 
             if cleaned > 0:
-                self.logger.info(f"Cleaned up {cleaned} temporary files")
+                logger.info(f"Cleaned up {cleaned} temporary files")
 
             return cleaned
 
         except Exception as e:
-            self.logger.error(f"Error cleaning up temp files: {e}")
+            logger.error(f"Error cleaning up temp files: {e}")
             return 0
 
     def get_storage_info(self) -> Dict[str, Any]:
@@ -447,7 +447,7 @@ class StateStorage:
             }
 
         except Exception as e:
-            self.logger.error(f"Error getting storage info: {e}")
+            logger.error(f"Error getting storage info: {e}")
             return {"storage_directory": str(self.storage_dir), "error": str(e)}