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

claude_mpm/services/shared/service_factory.py

@@ -8,137 +8,141 @@ from ...core.config import Config
 from ...core.logger import get_logger
 from ...core.shared import ConfigLoader, PathResolver, SingletonManager
 
-T = TypeVar('T')
+T = TypeVar("T")
 
 
 class ServiceFactory:
     """
     Factory for creating services with common patterns.
-    
+
     Reduces duplication by providing standard service creation patterns:
     - Configuration injection
     - Dependency resolution
     - Singleton management
     - Standard initialization
     """
-    
+
     def __init__(self):
         """Initialize service factory."""
         self.logger = get_logger("service_factory")
         self._config_loader = ConfigLoader()
         self._path_resolver = PathResolver()
         self._registered_services: Dict[str, Type] = {}
-    
+
     def register_service(self, service_name: str, service_class: Type[T]) -> None:
         """
         Register a service class.
-        
+
         Args:
             service_name: Name to register service under
             service_class: Service class to register
         """
         self._registered_services[service_name] = service_class
-        self.logger.debug(f"Registered service: {service_name} -> {service_class.__name__}")
-    
-    def create_service(self, 
-                      service_class: Type[T],
-                      service_name: str = None,
-                      config: Optional[Dict[str, Any]] = None,
-                      singleton: bool = False,
-                      **kwargs) -> T:
+        self.logger.debug(
+            f"Registered service: {service_name} -> {service_class.__name__}"
+        )
+
+    def create_service(
+        self,
+        service_class: Type[T],
+        service_name: Optional[str] = None,
+        config: Optional[Dict[str, Any]] = None,
+        singleton: bool = False,
+        **kwargs,
+    ) -> T:
         """
         Create a service instance with standard patterns.
-        
+
         Args:
             service_class: Service class to instantiate
             service_name: Optional service name (defaults to class name)
             config: Optional configuration override
             singleton: Whether to use singleton pattern
             **kwargs: Additional constructor arguments
-            
+
         Returns:
             Service instance
         """
         if service_name is None:
             service_name = service_class.__name__
-        
+
         self.logger.debug(f"Creating service: {service_name}")
-        
+
         # Load configuration if not provided
         if config is None:
             config = self._load_service_config(service_name)
-        
+
         # Prepare constructor arguments
-        constructor_args = {
-            "service_name": service_name,
-            "config": config,
-            **kwargs
-        }
-        
+        constructor_args = {"service_name": service_name, "config": config, **kwargs}
+
         # Filter arguments based on constructor signature
         filtered_args = self._filter_constructor_args(service_class, constructor_args)
-        
+
         # Create instance
         if singleton:
             instance = SingletonManager.get_instance(service_class, **filtered_args)
         else:
             instance = service_class(**filtered_args)
-        
+
         self.logger.info(f"Created service: {service_name} ({service_class.__name__})")
         return instance
-    
-    def create_registered_service(self, 
-                                 service_name: str,
-                                 config: Optional[Dict[str, Any]] = None,
-                                 singleton: bool = False,
-                                 **kwargs) -> Any:
+
+    def create_registered_service(
+        self,
+        service_name: str,
+        config: Optional[Dict[str, Any]] = None,
+        singleton: bool = False,
+        **kwargs,
+    ) -> Any:
         """
         Create a service by registered name.
-        
+
         Args:
             service_name: Name of registered service
             config: Optional configuration override
             singleton: Whether to use singleton pattern
             **kwargs: Additional constructor arguments
-            
+
         Returns:
             Service instance
-            
+
         Raises:
             ValueError: If service not registered
         """
         if service_name not in self._registered_services:
             raise ValueError(f"Service not registered: {service_name}")
-        
+
         service_class = self._registered_services[service_name]
         return self.create_service(
             service_class=service_class,
             service_name=service_name,
             config=config,
             singleton=singleton,
-            **kwargs
+            **kwargs,
         )
-    
-    def create_agent_service(self, 
-                           service_class: Type[T],
-                           agent_name: str,
-                           agent_dir: str = None,
-                           **kwargs) -> T:
+
+    def create_agent_service(
+        self,
+        service_class: Type[T],
+        agent_name: str,
+        agent_dir: Optional[str] = None,
+        **kwargs,
+    ) -> T:
         """
         Create an agent-related service.
-        
+
         Args:
             service_class: Service class to instantiate
             agent_name: Name of the agent
             agent_dir: Optional agent directory
             **kwargs: Additional constructor arguments
-            
+
         Returns:
             Service instance
         """
         # Load agent-specific configuration
         config = self._config_loader.load_agent_config(agent_dir)
-        
+
         # Add agent-specific paths
         if agent_dir:
             agent_path = self._path_resolver.resolve_relative_path(agent_dir)
@@ -146,97 +150,94 @@ class ServiceFactory:
             agent_path = self._path_resolver.find_agent_file(agent_name)
             if agent_path:
                 agent_path = agent_path.parent
-        
+
         constructor_args = {
             "agent_name": agent_name,
             "agent_dir": agent_path,
             "config": config,
-            **kwargs
+            **kwargs,
         }
-        
+
         return self.create_service(
             service_class=service_class,
             service_name=f"agent_{agent_name}",
             config=config,
-            **constructor_args
+            **constructor_args,
         )
-    
-    def create_memory_service(self, 
-                            service_class: Type[T],
-                            agent_name: str = None,
-                            memory_dir: str = None,
-                            **kwargs) -> T:
+
+    def create_memory_service(
+        self,
+        service_class: Type[T],
+        agent_name: Optional[str] = None,
+        memory_dir: Optional[str] = None,
+        **kwargs,
+    ) -> T:
         """
         Create a memory-related service.
-        
+
         Args:
             service_class: Service class to instantiate
             agent_name: Optional agent name
             memory_dir: Optional memory directory
             **kwargs: Additional constructor arguments
-            
+
         Returns:
             Service instance
         """
         # Load memory-specific configuration
         config = self._config_loader.load_memory_config(memory_dir)
-        
+
         # Resolve memory directory
         if memory_dir:
             memory_path = self._path_resolver.resolve_relative_path(memory_dir)
         else:
             memory_path = self._path_resolver.resolve_memories_dir(create=True)
-        
-        constructor_args = {
-            "memory_dir": memory_path,
-            "config": config,
-            **kwargs
-        }
-        
+
+        constructor_args = {"memory_dir": memory_path, "config": config, **kwargs}
+
         if agent_name:
             constructor_args["agent_name"] = agent_name
-        
+
         service_name = f"memory_{agent_name}" if agent_name else "memory"
-        
+
         return self.create_service(
             service_class=service_class,
             service_name=service_name,
             config=config,
-            **constructor_args
+            **constructor_args,
         )
-    
-    def create_config_service(self, 
-                            service_class: Type[T],
-                            config_section: str = None,
-                            **kwargs) -> T:
+
+    def create_config_service(
+        self, service_class: Type[T], config_section: Optional[str] = None, **kwargs
+    ) -> T:
         """
         Create a configuration-heavy service.
-        
+
         Args:
             service_class: Service class to instantiate
             config_section: Configuration section name
             **kwargs: Additional constructor arguments
-            
+
         Returns:
             Service instance
         """
         # Load service-specific configuration
         service_name = config_section or service_class.__name__.lower()
         config = self._config_loader.load_service_config(service_name)
-        
+
         constructor_args = {
             "config": config,
             "config_section": config_section,
-            **kwargs
+            **kwargs,
         }
-        
+
         return self.create_service(
             service_class=service_class,
             service_name=service_name,
             config=config,
-            **constructor_args
+            **constructor_args,
         )
-    
+
     def _load_service_config(self, service_name: str) -> Config:
         """Load configuration for a service."""
         try:
@@ -245,44 +246,41 @@ class ServiceFactory:
             self.logger.warning(f"Failed to load config for {service_name}: {e}")
             # Return default config using ConfigLoader
             return self._config_loader.load_main_config()
-    
-    def _filter_constructor_args(self, 
-                                service_class: Type,
-                                args: Dict[str, Any]) -> Dict[str, Any]:
+
+    def _filter_constructor_args(
+        self, service_class: Type, args: Dict[str, Any]
+    ) -> Dict[str, Any]:
         """Filter constructor arguments based on class signature."""
         import inspect
-        
+
         try:
             # Get constructor signature
             sig = inspect.signature(service_class.__init__)
-            param_names = set(sig.parameters.keys()) - {'self'}
-            
+            param_names = set(sig.parameters.keys()) - {"self"}
+
             # Filter arguments
             filtered = {k: v for k, v in args.items() if k in param_names}
-            
+
             # Log filtered arguments
             if len(filtered) != len(args):
                 filtered_out = set(args.keys()) - set(filtered.keys())
                 self.logger.debug(f"Filtered out arguments: {filtered_out}")
-            
+
             return filtered
-            
+
         except Exception as e:
             self.logger.warning(f"Failed to filter constructor args: {e}")
             return args
-    
+
     def get_registered_services(self) -> Dict[str, str]:
         """
         Get list of registered services.
-        
+
         Returns:
             Dictionary mapping service names to class names
         """
-        return {
-            name: cls.__name__ 
-            for name, cls in self._registered_services.items()
-        }
-    
+        return {name: cls.__name__ for name, cls in self._registered_services.items()}
+
     def clear_registrations(self) -> None:
         """Clear all service registrations."""
         self._registered_services.clear()

claude_mpm/services/socketio/__init__.py

@@ -17,9 +17,9 @@ from .handlers import (
 __all__ = [
     "BaseEventHandler",
     "ConnectionEventHandler",
-    "ProjectEventHandler",
-    "MemoryEventHandler",
+    "EventHandlerRegistry",
     "FileEventHandler",
     "GitEventHandler",
-    "EventHandlerRegistry",
+    "MemoryEventHandler",
+    "ProjectEventHandler",
 ]

claude_mpm/services/socketio/client_proxy.py

@@ -14,7 +14,7 @@ import asyncio
 import threading
 import time
 from datetime import datetime
-from typing import Any, Dict, List, Optional, Set
+from typing import Any, Dict, List, Optional
 
 try:
     import socketio
@@ -55,7 +55,7 @@ class SocketIOClientProxy:
 
     def stop_sync(self):
         """Stop the Socket.IO client connection."""
-        self.logger.debug(f"SocketIOClientProxy: Disconnecting from server")
+        self.logger.debug("SocketIOClientProxy: Disconnecting from server")
         if self._sio_client:
             self._sio_client.disconnect()
 
@@ -90,7 +90,7 @@ class SocketIOClientProxy:
 
             @self._sio_client.event
             async def disconnect():
-                self.logger.info(f"SocketIOClientProxy: Disconnected from server")
+                self.logger.info("SocketIOClientProxy: Disconnected from server")
 
             # Connect to the server
             await self._sio_client.connect(f"http://127.0.0.1:{self.port}")
@@ -128,7 +128,7 @@ class SocketIOClientProxy:
                     and not self._client_loop.is_closed()
                 ):
                     try:
-                        future = asyncio.run_coroutine_threadsafe(
+                        asyncio.run_coroutine_threadsafe(
                             self._sio_client.emit("claude_event", event),
                             self._client_loop,
                         )
@@ -160,7 +160,7 @@ class SocketIOClientProxy:
         self.logger.debug(f"SocketIOClientProxy: Session started {session_id}")
 
     def session_ended(self):
-        self.logger.debug(f"SocketIOClientProxy: Session ended")
+        self.logger.debug("SocketIOClientProxy: Session ended")
 
     def claude_status_changed(
         self, status: str, pid: Optional[int] = None, message: str = ""