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

claude_mpm/services/agents/loading/agent_profile_loader.py

@@ -20,7 +20,6 @@ multi-file implementation for better maintainability.
 """
 
 import asyncio
-import hashlib
 import json
 import logging
 import os
@@ -28,7 +27,7 @@ from dataclasses import dataclass, field
 from datetime import datetime
 from enum import Enum
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Set, Tuple, Union
+from typing import Any, Dict, List, Optional, Tuple
 
 import yaml
 
@@ -37,7 +36,6 @@ from claude_mpm.core.config import Config
 from claude_mpm.core.unified_paths import get_path_manager
 from claude_mpm.services.agents.registry import AgentRegistry
 from claude_mpm.services.memory.cache.shared_prompt_cache import SharedPromptCache
-from claude_mpm.utils.path_operations import path_ops
 
 logger = logging.getLogger(__name__)
 
@@ -146,7 +144,7 @@ class AgentProfileLoader(BaseService):
         # Performance tracking
         self.load_metrics: Dict[str, float] = {}
 
-        logger.info(f"AgentProfileLoader initialized successfully")
+        logger.info("AgentProfileLoader initialized successfully")
         logger.info(f"  Working directory: {self.working_directory}")
         logger.info(f"  Framework path: {self.framework_path}")
         logger.info(f"  Tier paths: {list(self.tier_paths.keys())}")
@@ -329,7 +327,7 @@ class AgentProfileLoader(BaseService):
                     instructions = data.get("instructions", "")
 
             # Create profile
-            profile = AgentProfile(
+            return AgentProfile(
                 name=data.get("name", file_path.stem),
                 role=data.get("role", "agent"),
                 description=data.get("description", ""),
@@ -342,8 +340,6 @@ class AgentProfileLoader(BaseService):
                 status=ProfileStatus.LOADED,
             )
 
-            return profile
-
         except Exception as e:
             logger.error(f"Error parsing profile {file_path}: {e}")
             return AgentProfile(
@@ -421,9 +417,7 @@ class AgentProfileLoader(BaseService):
                 for file_path in tier_path.glob(pattern):
                     agent_name = file_path.stem
                     # Remove common suffixes
-                    if agent_name.endswith("_agent"):
-                        agent_name = agent_name[:-6]
-                    elif agent_name.endswith("-agent"):
+                    if agent_name.endswith(("_agent", "-agent")):
                         agent_name = agent_name[:-6]
 
                     if agent_name not in agents:
@@ -448,7 +442,7 @@ class AgentProfileLoader(BaseService):
         for agents in discovered.values():
             all_agents.update(agents)
 
-        return sorted(list(all_agents))
+        return sorted(all_agents)
 
     # ========================================================================
     # Improved Prompts Management
@@ -461,7 +455,7 @@ class AgentProfileLoader(BaseService):
 
         if prompt_file.exists():
             try:
-                with open(prompt_file, "r") as f:
+                with open(prompt_file) as f:
                     data = json.load(f)
                     for prompt_data in data:
                         prompt = ImprovedPrompt(

claude_mpm/services/agents/loading/base_agent_manager.py

@@ -67,7 +67,9 @@ class BaseAgentStructure:
 class BaseAgentManager(ConfigServiceBase):
     """Manages base_agent.md with structured updates and validation."""
 
-    def __init__(self, agents_dir: Optional[Path] = None, config: Optional[Dict[str, Any]] = None):
+    def __init__(
+        self, agents_dir: Optional[Path] = None, config: Optional[Dict[str, Any]] = None
+    ):
         """Initialize BaseAgentManager."""
         super().__init__("base_agent_manager", config=config)
 
@@ -75,7 +77,7 @@ class BaseAgentManager(ConfigServiceBase):
         self.agents_dir = self.get_config_value(
             "agents_dir",
             default=agents_dir or Path(__file__).parent.parent / "agents",
-            config_type=Path
+            config_type=Path,
         )
         self.base_agent_path = self.agents_dir / "BASE_AGENT_TEMPLATE.md"
         self.cache = SharedPromptCache.get_instance()
@@ -347,9 +349,7 @@ class BaseAgentManager(ConfigServiceBase):
         items = []
         for line in content.split("\n"):
             line = line.strip()
-            if line.startswith("- "):
-                items.append(line[2:].strip())
-            elif line.startswith("* "):
+            if line.startswith(("- ", "* ")):
                 items.append(line[2:].strip())
             elif line.startswith("**") and line.endswith("**"):
                 # Skip bold headers

claude_mpm/services/agents/loading/framework_agent_loader.py

@@ -12,14 +12,12 @@ This service integrates with the main agent_loader.py to provide
 markdown-based agent profiles alongside JSON-based templates.
 """
 
-import json
 import logging
 from pathlib import Path
 from typing import Any, Dict, Optional
 
 from claude_mpm.agents.agent_loader import (
     AgentTier,
-    get_agent_tier,
     list_agents_by_tier,
 )
 from claude_mpm.core.unified_paths import get_path_manager
@@ -97,7 +95,7 @@ class FrameworkAgentLoader:
         current = Path.cwd()
 
         # Check current directory and parents
-        for path in [current] + list(current.parents):
+        for path in [current, *list(current.parents)]:
             framework_instructions = path / "agents" / "INSTRUCTIONS.md"
             framework_claude = path / "agents" / "CLAUDE.md"  # Legacy
             if framework_instructions.exists() or framework_claude.exists():
@@ -110,7 +108,7 @@ class FrameworkAgentLoader:
         current = Path.cwd()
 
         # Check current directory and parents for .claude-mpm
-        for path in [current] + list(current.parents):
+        for path in [current, *list(current.parents)]:
             claude_pm_dir = path / get_path_manager().CONFIG_DIR
             if claude_pm_dir.exists():
                 return path

claude_mpm/services/agents/management/__init__.py

@@ -4,6 +4,6 @@ from .agent_capabilities_generator import AgentCapabilitiesGenerator
 from .agent_management_service import AgentManager
 
 __all__ = [
-    "AgentManager",
     "AgentCapabilitiesGenerator",
+    "AgentManager",
 ]

claude_mpm/services/agents/management/agent_capabilities_generator.py

@@ -44,15 +44,13 @@ class AgentCapabilitiesGenerator:
             )
 
             # Render template
-            content = self.template.render(
+            return self.template.render(
                 core_agents=core_agent_list,
                 detailed_capabilities=detailed_capabilities,
                 agents_by_tier=agents_by_tier,
                 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

claude_mpm/services/agents/management/agent_management_service.py

@@ -12,10 +12,9 @@ Uses python-frontmatter and mistune for markdown parsing as recommended.
 """
 
 import logging
-import os
 import re
 from datetime import datetime
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, Dict, List, Optional
 
 import frontmatter
 import mistune
@@ -371,7 +370,7 @@ class AgentManager:
         title = title_match.group(1) if title_match else name.replace("-", " ").title()
 
         # Build definition
-        definition = AgentDefinition(
+        return AgentDefinition(
             name=name,
             title=title,
             file_path=file_path,
@@ -403,8 +402,6 @@ class AgentManager:
             raw_sections=sections,
         )
 
-        return definition
-
     def _extract_sections(self, content: str) -> Dict[str, str]:
         """Extract sections from markdown content."""
         sections = {}
@@ -425,10 +422,9 @@ class AgentManager:
                 # Start new section
                 current_section = header_match.group(1).strip()
                 current_content = []
-            else:
-                # Add to current section
-                if current_section:
-                    current_content.append(line)
+            # Add to current section
+            elif current_section:
+                current_content.append(line)
 
         # Save last section
         if current_section:

claude_mpm/services/agents/memory/__init__.py

@@ -12,11 +12,11 @@ from .template_generator import MemoryTemplateGenerator
 
 __all__ = [
     "AgentMemoryManager",
-    "get_memory_manager",
     "AgentPersistenceService",
-    "PersistenceStrategy",
+    "MemoryContentManager",
+    "MemoryTemplateGenerator",
     "PersistenceOperation",
     "PersistenceRecord",
-    "MemoryTemplateGenerator",
-    "MemoryContentManager",
+    "PersistenceStrategy",
+    "get_memory_manager",
 ]