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

claude_mpm/core/tool_access_control.py

@@ -1,7 +1,7 @@
 """Tool access control system for managing which tools agents can use."""
 
 import logging
-from typing import Dict, List, Optional, Set
+from typing import Dict, List, Set
 
 logger = logging.getLogger(__name__)
 
@@ -76,7 +76,7 @@ class ToolAccessControl:
         if is_parent or agent_type == "pm":
             allowed = self.PM_TOOLS.copy()
             logger.debug(f"PM/Parent process allowed tools: {allowed}")
-            return sorted(list(allowed))
+            return sorted(allowed)
 
         # Check custom restrictions first
         if agent_type in self.custom_restrictions:
@@ -94,7 +94,7 @@ class ToolAccessControl:
         allowed = allowed - {"TodoWrite"}
 
         logger.debug(f"Agent '{agent_type}' allowed tools: {allowed}")
-        return sorted(list(allowed))
+        return sorted(allowed)
 
     def format_allowed_tools_arg(self, agent_type: str, is_parent: bool = False) -> str:
         """
@@ -166,8 +166,7 @@ class ToolAccessControl:
         """
         if agent_type.lower() == "pm":
             return """You have access to the TodoWrite tool for tracking tasks. Always prefix todos with [Agent] to indicate delegation target."""
-        else:
-            return """You do NOT have access to TodoWrite. Instead, when you identify tasks that need tracking:
+        return """You do NOT have access to TodoWrite. Instead, when you identify tasks that need tracking:
 1. Include them in your response with clear markers like "TODO:" or "TASK:"
 2. Format them as a structured list with priority and description
 3. The PM will extract and track these in the central todo list

claude_mpm/core/typing_utils.py

@@ -16,20 +16,15 @@ strict type safety.
 
 import logging
 from datetime import datetime
-from enum import Enum
 from typing import (
-    TYPE_CHECKING,
     Any,
-    AsyncIterator,
     Awaitable,
     Callable,
     Dict,
-    Iterator,
     List,
     Literal,
     Optional,
     Protocol,
-    Set,
     Tuple,
     TypeVar,
     Union,
@@ -37,10 +32,6 @@ from typing import (
 
 from typing_extensions import NotRequired, TypeAlias, TypedDict
 
-if TYPE_CHECKING:
-    from claude_mpm.core.claude_runner import ClaudeRunner
-    from claude_mpm.services.socketio_server import SocketIOClientProxy
-
 # Generic type variables
 T = TypeVar("T")
 TSession = TypeVar("TSession")  # Generic session type
@@ -488,74 +479,74 @@ ErrorResult = Tuple[bool, Optional[str]]
 SuccessResult = Tuple[bool, Any]
 
 __all__ = [
-    # Basic type aliases
-    "PathLike",
-    "JSONValue",
-    "JSONDict",
-    "Headers",
-    "ErrorCode",
-    "LogLevel",
-    # Session types
-    "SessionId",
-    "SessionStatus",
-    "LaunchMethod",
-    "SessionConfig",
-    "SessionResult",
-    "SessionEvent",
+    "AgentCapabilities",
+    "AgentDefinition",
     # Agent types
     "AgentId",
-    "AgentVersion",
-    "AgentTier",
-    "ModelName",
-    "ResourceTier",
-    "AgentCapabilities",
     "AgentMetadata",
-    "AgentDefinition",
-    # WebSocket types
-    "EventName",
-    "EventData",
-    "SocketId",
-    "WebSocketMessage",
+    "AgentServiceProtocol",
+    "AgentTier",
+    "AgentVersion",
     "ClaudeStatus",
+    "CommandArgs",
+    "CommandName",
+    "CommandResult",
+    # Common combinations
+    "CommonTypes",
+    "ConfigDict",
     "DelegationInfo",
+    "DeploymentResult",
+    "ErrorCode",
+    "ErrorResult",
+    "EventData",
+    # WebSocket types
+    "EventName",
+    "Headers",
+    "HookConfig",
+    "HookContext",
     # Hook types
     "HookName",
     "HookPriority",
     "HookResult",
-    "HookConfig",
-    "HookContext",
-    # Service types
-    "ServiceName",
-    "ServiceStatus",
-    "ServiceConfig",
-    "ServiceInfo",
-    # Memory types
-    "MemoryType",
-    "MemoryId",
+    "JSONDict",
+    "JSONValue",
+    "LaunchMethod",
+    "LogLevel",
+    "LoggerProtocol",
     "Memory",
+    "MemoryId",
     "MemorySearchResult",
+    "MemoryServiceProtocol",
+    # Memory types
+    "MemoryType",
+    "ModelName",
+    # Basic type aliases
+    "PathLike",
     # Other types
     "ProjectConfig",
-    "DeploymentResult",
+    "ResourceTier",
     "ResponseLogEntry",
-    "CommandName",
-    "CommandArgs",
-    "CommandResult",
+    "ServiceConfig",
+    "ServiceContainer",
+    "ServiceInfo",
+    # Service types
+    "ServiceName",
+    "ServiceStatus",
+    "SessionConfig",
+    "SessionEvent",
+    # Session types
+    "SessionId",
     # Protocols
     "SessionProtocol",
-    "LoggerProtocol",
-    "WebSocketServerProtocol",
-    "AgentServiceProtocol",
-    "MemoryServiceProtocol",
-    "ServiceContainer",
+    "SessionResult",
+    "SessionStatus",
+    "SocketId",
+    "SuccessResult",
     # Generic type variables
     "T",
-    "TSession",
     "TAgent",
     "TService",
-    # Common combinations
-    "CommonTypes",
-    "ConfigDict",
-    "ErrorResult",
-    "SuccessResult",
+    "TSession",
+    "WebSocketMessage",
+    "WebSocketServerProtocol",
 ]

claude_mpm/core/unified_agent_registry.py

@@ -25,6 +25,7 @@ Architecture:
 - Discovery engine with tier-based precedence
 """
 
+import contextlib
 import json
 import logging
 import time
@@ -322,13 +323,12 @@ class UnifiedAgentRegistry:
             or str(self.path_manager.get_project_agents_dir()) in path_str
         ):
             return AgentTier.PROJECT
-        elif (
+        if (
             "user" in path_str
             or str(self.path_manager.get_user_agents_dir()) in path_str
         ):
             return AgentTier.USER
-        else:
-            return AgentTier.SYSTEM
+        return AgentTier.SYSTEM
 
     def _determine_agent_type(self, file_path: Path, tier: AgentTier) -> AgentType:
         """Determine agent type based on file path and tier."""
@@ -388,8 +388,7 @@ class UnifiedAgentRegistry:
 
         # Look for frontmatter
         if lines and lines[0].strip() == "---":
-            in_frontmatter = True
-            for i, line in enumerate(lines[1:], 1):
+            for _i, line in enumerate(lines[1:], 1):
                 if line.strip() == "---":
                     break
                 if line.startswith("description:"):
@@ -410,8 +409,7 @@ class UnifiedAgentRegistry:
         # Look for frontmatter
         lines = content.split("\n")
         if lines and lines[0].strip() == "---":
-            in_frontmatter = True
-            for i, line in enumerate(lines[1:], 1):
+            for _i, line in enumerate(lines[1:], 1):
                 if line.strip() == "---":
                     break
                 if line.startswith("specializations:"):
@@ -419,10 +417,8 @@ class UnifiedAgentRegistry:
                     spec_content = line.split(":", 1)[1].strip()
                     if spec_content.startswith("[") and spec_content.endswith("]"):
                         # JSON-style list
-                        try:
+                        with contextlib.suppress(Exception):
                             specializations = json.loads(spec_content)
-                        except:
-                            pass
 
         return specializations
 
@@ -500,7 +496,6 @@ class UnifiedAgentRegistry:
         """Cache the current registry state."""
         # For now, we just store in memory
         # In a full implementation, this could write to disk
-        pass
 
     # ========================================================================
     # Public API Methods
@@ -607,7 +602,7 @@ class UnifiedAgentRegistry:
         """Import registry from JSON file."""
         input_path = Path(input_path)
 
-        with open(input_path, "r") as f:
+        with open(input_path) as f:
             data = json.load(f)
 
         # Clear current registry
@@ -700,22 +695,22 @@ def list_agents_all() -> List[AgentMetadata]:
 # ============================================================================
 
 __all__ = [
-    "UnifiedAgentRegistry",
+    "AgentFormat",
     "AgentMetadata",
     "AgentTier",
     "AgentType",
-    "AgentFormat",
-    "get_agent_registry",
+    "UnifiedAgentRegistry",
     "discover_agents",
-    "list_agents",
+    "discover_agents_sync",
     "get_agent",
+    "get_agent_names",
+    "get_agent_registry",
     "get_core_agents",
-    "get_specialized_agents",
     "get_project_agents",
-    "get_agent_names",
     "get_registry_stats",
+    "get_specialized_agents",
     # Legacy compatibility
     "listAgents",
-    "discover_agents_sync",
+    "list_agents",
     "list_agents_all",
 ]

claude_mpm/core/unified_config.py

@@ -13,7 +13,6 @@ Design Principles:
 6. Backward compatibility with existing config systems
 """
 
-import os
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Union
 
@@ -56,7 +55,7 @@ class LoggingConfig(BaseModel):
     )
 
     @validator("level")
-    def validate_log_level(cls, v):
+    def validate_log_level(self, v):
         valid_levels = ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
         if v.upper() not in valid_levels:
             raise ValueError(f"Invalid log level. Must be one of: {valid_levels}")
@@ -267,7 +266,7 @@ class UnifiedConfig(BaseSettings):
         extra = "allow"  # Allow extra fields for backward compatibility
 
     @validator("environment")
-    def validate_environment(cls, v):
+    def validate_environment(self, v):
         valid_envs = ["development", "testing", "production"]
         if v not in valid_envs:
             raise ValueError(f"Invalid environment. Must be one of: {valid_envs}")
@@ -437,7 +436,7 @@ class ConfigurationService:
                 if config_path.exists():
                     import yaml
 
-                    with open(config_path, "r") as f:
+                    with open(config_path) as f:
                         file_config = yaml.safe_load(f) or {}
                     config_data.update(file_config)
                     break
@@ -514,8 +513,7 @@ class ConfigurationService:
             # Pydantic validation happens automatically, but we can add custom validation here
             if self._config.network.socketio_port in range(1024, 65536):
                 return True
-            else:
-                raise ConfigurationError("Invalid SocketIO port range")
+            raise ConfigurationError("Invalid SocketIO port range")
         except Exception as e:
             raise ConfigurationError(f"Configuration validation failed: {e}")