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

claude_mpm/cli/startup_logging.py

@@ -22,9 +22,58 @@ from datetime import datetime
 from pathlib import Path
 from typing import Any, Dict, Optional
 
+try:
+    import psutil
+
+    PSUTIL_AVAILABLE = True
+except ImportError:
+    PSUTIL_AVAILABLE = False
+
 from ..core.logger import get_logger
 
 
+def log_memory_stats(logger=None, prefix="Memory Usage"):
+    """
+    Log current memory statistics.
+
+    Args:
+        logger: Logger to use (defaults to 'cli' logger)
+        prefix: Prefix for the log message
+
+    Returns:
+        Dict with memory stats or None if psutil unavailable
+    """
+    if not PSUTIL_AVAILABLE:
+        return None
+
+    if logger is None:
+        logger = get_logger("cli")
+
+    try:
+        process = psutil.Process()
+        memory_info = process.memory_info()
+
+        # Convert to MB for readability
+        rss_mb = memory_info.rss / (1024 * 1024)
+        vms_mb = memory_info.vms / (1024 * 1024)
+
+        # Get percentage of system memory if available
+        try:
+            memory_percent = process.memory_percent()
+            logger.info(
+                f"{prefix}: RSS={rss_mb:.1f}MB, VMS={vms_mb:.1f}MB, "
+                f"System={memory_percent:.1f}%"
+            )
+            return {"rss_mb": rss_mb, "vms_mb": vms_mb, "percent": memory_percent}
+        except:
+            logger.info(f"{prefix}: RSS={rss_mb:.1f}MB, VMS={vms_mb:.1f}MB")
+            return {"rss_mb": rss_mb, "vms_mb": vms_mb, "percent": None}
+
+    except Exception as e:
+        logger.debug(f"Failed to get memory info: {e}")
+        return None
+
+
 class StartupStatusLogger:
     """Logs MCP server and monitor setup status during startup."""
 
@@ -90,6 +139,19 @@ class StartupStatusLogger:
         except Exception as e:
             self.logger.warning(f"MCP Server: Status check failed - {e}")
 
+    def log_memory_status(self) -> None:
+        """
+        Log current process memory usage.
+
+        Logs both RSS (Resident Set Size) and VMS (Virtual Memory Size)
+        to help track memory consumption and potential leaks.
+        """
+        stats = log_memory_stats(self.logger, "Memory Usage")
+
+        # Log warning if memory usage is high
+        if stats and stats.get("rss_mb", 0) > 500:  # Warn if using more than 500MB
+            self.logger.warning(f"High memory usage detected: {stats['rss_mb']:.1f}MB")
+
     def log_monitor_setup_status(
         self, monitor_mode: bool = False, websocket_port: int = 8765
     ) -> None:
@@ -447,6 +509,17 @@ def setup_startup_logging(project_root: Optional[Path] = None) -> Path:
     logger.info(f"CWD: {Path.cwd()}")
     logger.info(f"Project root: {project_root}")
 
+    # Log initial memory usage
+    if PSUTIL_AVAILABLE:
+        try:
+            process = psutil.Process()
+            memory_info = process.memory_info()
+            rss_mb = memory_info.rss / (1024 * 1024)
+            vms_mb = memory_info.vms / (1024 * 1024)
+            logger.info(f"Initial Memory: RSS={rss_mb:.1f}MB, VMS={vms_mb:.1f}MB")
+        except Exception as e:
+            logger.debug(f"Failed to get initial memory info: {e}")
+
     return log_file
 
 
@@ -558,6 +631,9 @@ def log_startup_status(monitor_mode: bool = False, websocket_port: int = 8765) -
     try:
         status_logger = StartupStatusLogger("cli")
 
+        # Log memory status at startup
+        status_logger.log_memory_status()
+
         # Log MCP server status
         status_logger.log_mcp_server_status()
 

claude_mpm/core/agent_registry.py

@@ -19,22 +19,18 @@ from pathlib import Path
 from typing import Any, Dict, List, Optional, Set
 
 # Import from the unified agent registry system
-from .unified_agent_registry import AgentMetadata as UnifiedAgentMetadata
 from .unified_agent_registry import (
+    AgentMetadata as UnifiedAgentMetadata,
     AgentTier,
     AgentType,
-)
-from .unified_agent_registry import discover_agents as unified_discover_agents
-from .unified_agent_registry import get_agent as unified_get_agent
-from .unified_agent_registry import (
+    discover_agents as unified_discover_agents,
+    get_agent as unified_get_agent,
     get_agent_registry,
-)
-from .unified_agent_registry import get_core_agents as unified_get_core_agents
-from .unified_agent_registry import get_registry_stats as unified_get_registry_stats
-from .unified_agent_registry import (
+    get_core_agents as unified_get_core_agents,
+    get_registry_stats as unified_get_registry_stats,
     get_specialized_agents as unified_get_specialized_agents,
+    list_agents as unified_list_agents,
 )
-from .unified_agent_registry import list_agents as unified_list_agents
 
 try:
     from ..core.logger import get_logger