claude-mpm 4.1.26__py3-none-any.whl → 5.0.9__py3-none-any.whl

claude_mpm/services/mcp_gateway/main.py

@@ -63,7 +63,7 @@ except ImportError:
 try:
     from claude_mpm.services.mcp_gateway.server.mcp_gateway import MCPGateway
 except ImportError as e:
-    raise ImportError(f"Critical: Cannot import MCPGateway server: {e}")
+    raise ImportError(f"Critical: Cannot import MCPGateway server: {e}") from e
 
 try:
     from claude_mpm.services.mcp_gateway.server.stdio_handler import StdioHandler
@@ -107,13 +107,16 @@ try:
 except ImportError:
     DocumentSummarizerTool = None
 
+# Ticket tools removed - using mcp-ticketer instead
+UnifiedTicketTool = None
+
 try:
-    from claude_mpm.services.mcp_gateway.tools.unified_ticket_tool import (
-        UnifiedTicketTool,
+    from claude_mpm.services.mcp_gateway.tools.external_mcp_services import (
+        ExternalMCPServiceManager,
     )
 except ImportError:
-    # Unified ticket tool is optional
-    UnifiedTicketTool = None
+    # External MCP services are optional
+    ExternalMCPServiceManager = None
 
 # Manager module removed - using simplified architecture
 
@@ -148,6 +151,7 @@ class MCPGatewayOrchestrator:
         self.registry: Optional[ToolRegistry] = None
         self.communication: Optional[StdioHandler] = None
         self.configuration: Optional[MCPConfiguration] = None
+        self.external_services: Optional[ExternalMCPServiceManager] = None
 
         # Shutdown handling
         self._shutdown_event = asyncio.Event()
@@ -175,10 +179,13 @@ class MCPGatewayOrchestrator:
 
             # Load configuration
             self.configuration = MCPConfiguration()
-            if self.config_path and self.config_path.exists():
-                if not self.configuration.load_config(self.config_path):
-                    self.logger.error("Failed to load configuration")
-                    return False
+            if (
+                self.config_path
+                and self.config_path.exists()
+                and not self.configuration.load_config(self.config_path)
+            ):
+                self.logger.error("Failed to load configuration")
+                return False
 
             # Initialize tool registry with error handling
             try:
@@ -199,6 +206,42 @@ class MCPGatewayOrchestrator:
                     self.logger.warning(f"Failed to register some tools: {e}")
                     # Continue - server can run with partial tools
 
+            # Initialize external MCP services if available
+            if ExternalMCPServiceManager is not None:
+                try:
+                    self.logger.info("Initializing external MCP services...")
+                    self.external_services = ExternalMCPServiceManager()
+                    external_services = (
+                        await self.external_services.initialize_services()
+                    )
+
+                    if external_services and self.registry:
+                        for service in external_services:
+                            try:
+                                if self.registry.register_tool(
+                                    service, category="external"
+                                ):
+                                    self.logger.info(
+                                        f"Registered external service: {service.service_name}"
+                                    )
+                                else:
+                                    self.logger.warning(
+                                        f"Failed to register external service: {service.service_name}"
+                                    )
+                            except Exception as e:
+                                self.logger.warning(
+                                    f"Error registering {service.service_name}: {e}"
+                                )
+
+                    self.logger.info(
+                        f"Initialized {len(external_services)} external MCP services"
+                    )
+                except Exception as e:
+                    self.logger.warning(
+                        f"Failed to initialize external MCP services: {e}"
+                    )
+                    self.external_services = None
+
             # Initialize communication handler with fallback
             try:
                 self.communication = StdioHandler()
@@ -271,12 +314,37 @@ class MCPGatewayOrchestrator:
             except Exception as e:
                 self.logger.warning(f"Could not load document summarizer: {e}")
 
-        # Optional: Unified ticket management tool
-        if UnifiedTicketTool is not None:
-            try:
-                tools.append(UnifiedTicketTool())
-            except Exception as e:
-                self.logger.warning(f"Could not load unified ticket tool: {e}")
+        # Kuzu-Memory Service (now a required dependency)
+        try:
+            from .tools.kuzu_memory_service import KuzuMemoryService
+
+            tools.append(KuzuMemoryService())
+            self.logger.info("KuzuMemoryService added to built-in tools")
+        except Exception as e:
+            self.logger.warning(f"Could not load KuzuMemoryService: {e}")
+
+        # MCP Vector Search Service (optional - will auto-install on first use)
+        try:
+            from .tools.external_mcp_services import MCPVectorSearchService
+
+            vector_search = MCPVectorSearchService()
+            # Try to initialize without interactive prompts during gateway startup
+            # This will only succeed if already installed
+            init_success = await vector_search.initialize(
+                auto_install=False, interactive=False
+            )
+
+            if init_success:
+                tools.append(vector_search)
+                self.logger.info("MCPVectorSearchService added to built-in tools")
+            else:
+                self.logger.debug(
+                    "mcp-vector-search not installed - will be available via auto-install on first use"
+                )
+        except Exception as e:
+            self.logger.debug(f"Could not load MCPVectorSearchService: {e}")
+
+        # Ticket tools removed - mcp-ticketer provides ticket functionality
 
         if not tools:
             self.logger.warning("No tools available to register")
@@ -371,6 +439,13 @@ class MCPGatewayOrchestrator:
                 except Exception as e:
                     self.logger.warning(f"Error during communication shutdown: {e}")
 
+            # Shutdown external services
+            if self.external_services:
+                try:
+                    await self.external_services.shutdown()
+                except Exception as e:
+                    self.logger.warning(f"Error during external services shutdown: {e}")
+
             self.logger.info("MCP Gateway shutdown complete")
 
         except Exception as e:

claude_mpm/services/mcp_gateway/registry/service_registry.py

@@ -224,7 +224,8 @@ class MCPServiceRegistry(ManagerBase):
                     try:
                         import asyncio
 
-                        asyncio.create_task(instance.stop())
+                        _task = asyncio.create_task(instance.stop())
+                        # Fire-and-forget shutdown during unregister
                     except Exception as e:
                         self.logger.warning(
                             f"Error stopping service {interface.__name__}: {e}"
@@ -335,7 +336,8 @@ class MCPServiceRegistry(ManagerBase):
                     try:
                         import asyncio
 
-                        asyncio.create_task(instance.stop())
+                        _task = asyncio.create_task(instance.stop())
+                        # Fire-and-forget shutdown during clear
                     except Exception as e:
                         self.logger.warning(f"Error stopping service: {e}")
 

claude_mpm/services/mcp_gateway/registry/tool_registry.py

@@ -11,7 +11,7 @@ Part of ISS-0035: MCP Server Implementation - Core Server and Tool Registry
 import asyncio
 import re
 import traceback
-from datetime import datetime
+from datetime import datetime, timezone
 from threading import RLock
 from typing import Any, Dict, List, Optional, Set
 
@@ -159,9 +159,9 @@ class ToolRegistry(BaseMCPService, IMCPToolRegistry):
 
                 # Update metrics
                 self._metrics["total_tools"] = len(self._adapters)
-                self._metrics["registration_time"][
-                    tool_name
-                ] = datetime.now().isoformat()
+                self._metrics["registration_time"][tool_name] = datetime.now(
+                    timezone.utc
+                ).isoformat()
                 self._metrics["invocations"][tool_name] = 0
                 self._metrics["errors"][tool_name] = 0
 
@@ -216,7 +216,8 @@ class ToolRegistry(BaseMCPService, IMCPToolRegistry):
 
             # Shutdown adapter (outside lock to avoid deadlock)
             try:
-                asyncio.create_task(adapter.shutdown())
+                _task = asyncio.create_task(adapter.shutdown())
+                # Fire-and-forget shutdown during tool unregister
             except Exception as e:
                 self.log_warning(f"Error shutting down tool adapter {tool_name}: {e}")
 
@@ -272,7 +273,7 @@ class ToolRegistry(BaseMCPService, IMCPToolRegistry):
         error handling, metrics tracking, and validation.
         """
         tool_name = invocation.tool_name
-        start_time = datetime.now()
+        start_time = datetime.now(timezone.utc)
 
         try:
             self.log_info(f"Invoking tool: {tool_name}")
@@ -300,7 +301,7 @@ class ToolRegistry(BaseMCPService, IMCPToolRegistry):
             result = await adapter.invoke(invocation)
 
             # Calculate execution time
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
             result.execution_time = execution_time
 
             # Update metrics
@@ -308,7 +309,9 @@ class ToolRegistry(BaseMCPService, IMCPToolRegistry):
                 self._metrics["invocations"][tool_name] = (
                     self._metrics["invocations"].get(tool_name, 0) + 1
                 )
-                self._metrics["last_invocation"][tool_name] = datetime.now().isoformat()
+                self._metrics["last_invocation"][tool_name] = datetime.now(
+                    timezone.utc
+                ).isoformat()
 
                 if not result.success:
                     self._metrics["errors"][tool_name] = (
@@ -331,7 +334,7 @@ class ToolRegistry(BaseMCPService, IMCPToolRegistry):
                     self._metrics["errors"].get(tool_name, 0) + 1
                 )
 
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
 
             return MCPToolResult(
                 success=False, error=error_msg, execution_time=execution_time

claude_mpm/services/mcp_gateway/server/mcp_gateway.py

@@ -16,17 +16,12 @@ import asyncio
 import contextlib
 import json
 import traceback
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import Any, Callable, Dict, List, Optional, Union
 
 # Import from the official MCP package
 from mcp.server import Server
-from mcp.types import (
-    EmbeddedResource,
-    ImageContent,
-    TextContent,
-    Tool,
-)
+from mcp.types import EmbeddedResource, ImageContent, TextContent, Tool
 
 from claude_mpm.services.mcp_gateway.core.base import BaseMCPService
 from claude_mpm.services.mcp_gateway.core.interfaces import (
@@ -148,7 +143,7 @@ class MCPGateway(BaseMCPService, IMCPGateway):
             invocation = MCPToolInvocation(
                 tool_name=name,
                 parameters=arguments,
-                request_id=f"req_{datetime.now().timestamp()}",
+                request_id=f"req_{datetime.now(timezone.utc).timestamp()}",
             )
 
             try:
@@ -212,7 +207,7 @@ class MCPGateway(BaseMCPService, IMCPGateway):
                 self.log_warning("No tool registry set - server will have no tools")
 
             # Initialize metrics
-            self._metrics["start_time"] = datetime.now().isoformat()
+            self._metrics["start_time"] = datetime.now(timezone.utc).isoformat()
 
             # Update capabilities based on registry
             if self._tool_registry:
@@ -287,7 +282,7 @@ class MCPGateway(BaseMCPService, IMCPGateway):
         try:
             # Update metrics
             self._metrics["requests_handled"] += 1
-            self._metrics["last_request_time"] = datetime.now().isoformat()
+            self._metrics["last_request_time"] = datetime.now(timezone.utc).isoformat()
 
             # Extract request details
             method = request.get("method", "")

claude_mpm/services/mcp_gateway/server/stdio_server.py

@@ -17,6 +17,8 @@ import logging
 import os
 import sys
 import time
+from datetime import timezone
+from pathlib import Path
 from typing import Any, Dict
 
 # Import MCP SDK components
@@ -28,15 +30,8 @@ from mcp.types import TextContent, Tool
 # Import pydantic for model patching
 from claude_mpm.core.logger import get_logger
 
-# Import unified ticket tool if available
-try:
-    from claude_mpm.services.mcp_gateway.tools.unified_ticket_tool import (
-        UnifiedTicketTool,
-    )
-
-    TICKET_TOOLS_AVAILABLE = True
-except ImportError:
-    TICKET_TOOLS_AVAILABLE = False
+# Ticket tools removed - using mcp-ticketer instead
+TICKET_TOOLS_AVAILABLE = False
 
 
 def apply_backward_compatibility_patches():
@@ -70,7 +65,6 @@ def apply_backward_compatibility_patches():
                         and hasattr(request, "params")
                         and request.params is not None
                     ):
-
                         # Convert params to dict to check for clientInfo
                         params_dict = request.params
                         if hasattr(params_dict, "model_dump"):
@@ -547,12 +541,9 @@ class SimpleMCPServer:
 
                         result = f"Python: {sys.version}"
                     elif info_type == "cwd":
-                        import os
-
-                        result = f"Working Directory: {os.getcwd()}"
+                        result = f"Working Directory: {Path.cwd()}"
                     elif info_type == "all":
                         import datetime
-                        import os
                         import platform
                         import sys
 
@@ -560,9 +551,9 @@ class SimpleMCPServer:
                             f"=== System Status ===\n"
                             f"Platform: {platform.system()} {platform.release()}\n"
                             f"Python: {sys.version.split()[0]}\n"
-                            f"Working Directory: {os.getcwd()}\n"
+                            f"Working Directory: {Path.cwd()}\n"
                             f"Server: {self.name} v{self.version}\n"
-                            f"Timestamp: {datetime.datetime.now().isoformat()}\n"
+                            f"Timestamp: {datetime.datetime.now(timezone.utc).isoformat()}\n"
                             f"Tools Available: status, document_summarizer{', ticket' if self.unified_ticket_tool else ''}"
                         )
                     else:
@@ -628,7 +619,8 @@ class SimpleMCPServer:
 
         try:
             self.logger.info("Initializing unified ticket tool...")
-            self.unified_ticket_tool = UnifiedTicketTool()
+            # Ticket tools removed - using mcp-ticketer instead
+            self.unified_ticket_tool = None
             # If the tool has an async init method, call it
             if hasattr(self.unified_ticket_tool, "initialize"):
                 await self.unified_ticket_tool.initialize()

claude_mpm/services/mcp_gateway/tools/__init__.py

@@ -12,13 +12,25 @@ from .base_adapter import (
     SystemInfoToolAdapter,
 )
 from .document_summarizer import DocumentSummarizerTool
-from .unified_ticket_tool import UnifiedTicketTool
+from .kuzu_memory_service import (
+    KuzuMemoryService,
+    get_context,
+    recall_memories,
+    search_memories,
+    store_memory,
+)
+
+# Ticket tools removed - using mcp-ticketer instead
 
 __all__ = [
     "BaseToolAdapter",
     "CalculatorToolAdapter",
     "DocumentSummarizerTool",
     "EchoToolAdapter",
+    "KuzuMemoryService",
     "SystemInfoToolAdapter",
-    "UnifiedTicketTool",
+    "get_context",
+    "recall_memories",
+    "search_memories",
+    "store_memory",
 ]

claude_mpm/services/mcp_gateway/tools/base_adapter.py

@@ -8,7 +8,7 @@ Part of ISS-0035: MCP Server Implementation - Core Server and Tool Registry
 """
 
 from abc import ABC
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import Any, Dict
 
 from claude_mpm.services.mcp_gateway.core.base import BaseMCPService
@@ -195,7 +195,7 @@ class BaseToolAdapter(BaseMCPService, IMCPToolAdapter, ABC):
         self._metrics["average_execution_time"] = (
             self._metrics["total_execution_time"] / self._metrics["invocations"]
         )
-        self._metrics["last_invocation"] = datetime.now().isoformat()
+        self._metrics["last_invocation"] = datetime.now(timezone.utc).isoformat()
 
 
 class EchoToolAdapter(BaseToolAdapter):
@@ -235,7 +235,7 @@ class EchoToolAdapter(BaseToolAdapter):
         Returns:
             Tool execution result with echoed message
         """
-        start_time = datetime.now()
+        start_time = datetime.now(timezone.utc)
 
         try:
             # Get parameters
@@ -246,7 +246,7 @@ class EchoToolAdapter(BaseToolAdapter):
             result = message.upper() if uppercase else message
 
             # Calculate execution time
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
 
             # Update metrics
             self._update_metrics(True, execution_time)
@@ -259,7 +259,7 @@ class EchoToolAdapter(BaseToolAdapter):
             )
 
         except Exception as e:
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
             self._update_metrics(False, execution_time)
             self._metrics["last_error"] = str(e)
 
@@ -321,7 +321,7 @@ class CalculatorToolAdapter(BaseToolAdapter):
         Returns:
             Tool execution result with calculation
         """
-        start_time = datetime.now()
+        start_time = datetime.now(timezone.utc)
 
         try:
             # Get parameters
@@ -348,7 +348,7 @@ class CalculatorToolAdapter(BaseToolAdapter):
                 raise ValueError(f"Unknown operation: {operation}")
 
             # Calculate execution time
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
 
             # Update metrics
             self._update_metrics(True, execution_time)
@@ -361,7 +361,7 @@ class CalculatorToolAdapter(BaseToolAdapter):
             )
 
         except Exception as e:
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
             self._update_metrics(False, execution_time)
             self._metrics["last_error"] = str(e)
 
@@ -408,7 +408,7 @@ class SystemInfoToolAdapter(BaseToolAdapter):
         Returns:
             Tool execution result with system information
         """
-        start_time = datetime.now()
+        start_time = datetime.now(timezone.utc)
 
         try:
             import platform
@@ -443,15 +443,15 @@ class SystemInfoToolAdapter(BaseToolAdapter):
                 }
             elif info_type == "time":
                 result = {
-                    "current": datetime.now().isoformat(),
-                    "timestamp": datetime.now().timestamp(),
-                    "timezone": str(datetime.now().astimezone().tzinfo),
+                    "current": datetime.now(timezone.utc).isoformat(),
+                    "timestamp": datetime.now(timezone.utc).timestamp(),
+                    "timezone": str(datetime.now(timezone.utc).astimezone().tzinfo),
                 }
             else:
                 raise ValueError(f"Unknown info type: {info_type}")
 
             # Calculate execution time
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
 
             # Update metrics
             self._update_metrics(True, execution_time)
@@ -465,7 +465,7 @@ class SystemInfoToolAdapter(BaseToolAdapter):
 
         except ImportError as e:
             # Handle missing psutil dependency gracefully
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
             self._update_metrics(False, execution_time)
 
             return MCPToolResult(
@@ -474,7 +474,7 @@ class SystemInfoToolAdapter(BaseToolAdapter):
                 execution_time=execution_time,
             )
         except Exception as e:
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
             self._update_metrics(False, execution_time)
             self._metrics["last_error"] = str(e)
 

claude_mpm/services/mcp_gateway/tools/document_summarizer.py

@@ -10,10 +10,9 @@ Part of ISS-0037: Document Summarizer Tool - Intelligent Document Processing
 
 import hashlib
 import mimetypes
-import os
 import re
 from collections import OrderedDict
-from datetime import datetime
+from datetime import datetime, timezone
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Tuple
 
@@ -74,7 +73,7 @@ class LRUCache:
             if not self.cache:
                 break
             # Remove least recently used item
-            removed_key, removed_value = self.cache.popitem(last=False)
+            _removed_key, removed_value = self.cache.popitem(last=False)
             self.current_memory -= removed_value.get("size_bytes", 0)
 
         # Add new item
@@ -240,7 +239,7 @@ class DocumentSummarizerTool(BaseToolAdapter):
 
     def _get_file_hash(self, file_path: str) -> str:
         """Generate hash for file caching."""
-        stat = os.stat(file_path)
+        stat = Path(file_path).stat()
         hash_input = f"{file_path}:{stat.st_size}:{stat.st_mtime}"
         return hashlib.md5(hash_input.encode()).hexdigest()
 
@@ -306,7 +305,7 @@ class DocumentSummarizerTool(BaseToolAdapter):
                 continue
 
         # If all fail, read as binary and decode with errors='ignore'
-        with open(file_path, "rb") as f:
+        with file_path.open("rb") as f:
             content = f.read()
             return content.decode("utf-8", errors="ignore")
 
@@ -605,7 +604,7 @@ class DocumentSummarizerTool(BaseToolAdapter):
         Returns:
             Tool execution result with summary
         """
-        start_time = datetime.now()
+        start_time = datetime.now(timezone.utc)
 
         try:
             # Get parameters
@@ -628,7 +627,9 @@ class DocumentSummarizerTool(BaseToolAdapter):
                 cached_result = self._cache.get(cache_key)
                 if cached_result:
                     cache_hit = True
-                    execution_time = (datetime.now() - start_time).total_seconds()
+                    execution_time = (
+                        datetime.now(timezone.utc) - start_time
+                    ).total_seconds()
                     self._update_metrics(True, execution_time)
 
                     return MCPToolResult(
@@ -709,7 +710,7 @@ class DocumentSummarizerTool(BaseToolAdapter):
                 self._cache.put(cache_key, result.copy(), summary_size)
 
             # Calculate execution time
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
 
             # Update metrics
             self._update_metrics(True, execution_time)
@@ -730,7 +731,7 @@ class DocumentSummarizerTool(BaseToolAdapter):
             )
 
         except Exception as e:
-            execution_time = (datetime.now() - start_time).total_seconds()
+            execution_time = (datetime.now(timezone.utc) - start_time).total_seconds()
             self._update_metrics(False, execution_time)
             self._metrics["last_error"] = str(e)