claude-mpm 4.3.12__py3-none-any.whl → 4.3.13__py3-none-any.whl

claude_mpm/services/mcp_gateway/server/stdio_server.py

@@ -17,6 +17,7 @@ import logging
 import os
 import sys
 import time
+from datetime import timezone
 from typing import Any, Dict
 
 # Import MCP SDK components
@@ -172,7 +173,7 @@ class SimpleMCPServer:
 
     async def _summarize_content(
         self, content: str, style: str, max_length: int
-    ) -> str:
+    ) -> str:  # noqa: PLR0911
         """
         Summarize text content based on style and length constraints.
 
@@ -214,7 +215,7 @@ class SimpleMCPServer:
         # Default to brief
         return self._create_brief_summary(sentences, max_length)
 
-    def _create_brief_summary(self, sentences: list[str], max_length: int) -> str:
+    def _create_brief_summary(self, sentences: list[str], max_length: int) -> str:  # noqa: PLR0911
         """Create a brief summary by selecting most important sentences."""
         if not sentences:
             return ""
@@ -562,7 +563,7 @@ class SimpleMCPServer:
                             f"Python: {sys.version.split()[0]}\n"
                             f"Working Directory: {os.getcwd()}\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:

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

@@ -13,7 +13,7 @@ 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
 
@@ -605,7 +605,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 +628,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 +711,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 +732,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)
 

claude_mpm/services/mcp_gateway/tools/external_mcp_services.py

@@ -10,14 +10,12 @@ Note: As of the latest architecture, external services are registered as separat
 MCP servers in Claude Desktop configuration, not as tools within the gateway.
 """
 
-import asyncio
 import json
 import subprocess
 import sys
 from pathlib import Path
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List
 
-from claude_mpm.services.mcp_gateway.core.base import BaseMCPService
 from claude_mpm.services.mcp_gateway.tools.base_adapter import BaseMCPToolAdapter
 
 
@@ -45,7 +43,9 @@ class ExternalMCPService(BaseMCPToolAdapter):
             self._is_installed = await self._check_installation()
 
             if not self._is_installed:
-                self.logger.warning(f"{self.package_name} not installed, attempting installation...")
+                self.logger.warning(
+                    f"{self.package_name} not installed, attempting installation..."
+                )
                 await self._install_package()
                 self._is_installed = await self._check_installation()
 
@@ -67,10 +67,15 @@ class ExternalMCPService(BaseMCPToolAdapter):
                 [sys.executable, "-m", self.package_name.replace("-", "_"), "--help"],
                 capture_output=True,
                 text=True,
-                timeout=5
+                timeout=5,
+                check=False,
             )
             return result.returncode == 0
-        except (subprocess.TimeoutExpired, FileNotFoundError, subprocess.CalledProcessError):
+        except (
+            subprocess.TimeoutExpired,
+            FileNotFoundError,
+            subprocess.CalledProcessError,
+        ):
             return False
 
     async def _install_package(self) -> bool:
@@ -81,15 +86,15 @@ class ExternalMCPService(BaseMCPToolAdapter):
                 [sys.executable, "-m", "pip", "install", self.package_name],
                 capture_output=True,
                 text=True,
-                timeout=30
+                timeout=30,
+                check=False,
             )
 
             if result.returncode == 0:
                 self.logger.info(f"Successfully installed {self.package_name}")
                 return True
-            else:
-                self.logger.error(f"Failed to install {self.package_name}: {result.stderr}")
-                return False
+            self.logger.error(f"Failed to install {self.package_name}: {result.stderr}")
+            return False
 
         except Exception as e:
             self.logger.error(f"Error installing {self.package_name}: {e}")
@@ -102,7 +107,7 @@ class ExternalMCPService(BaseMCPToolAdapter):
             "description": f"External MCP service: {self.package_name}",
             "type": "external_service",
             "package": self.package_name,
-            "installed": self._is_installed
+            "installed": self._is_installed,
         }
 
 
@@ -116,77 +121,103 @@ class MCPVectorSearchService(ExternalMCPService):
     def get_definition(self) -> Dict[str, Any]:
         """Get tool definition for MCP registration."""
         base_def = super().get_definition()
-        base_def.update({
-            "description": "Semantic code search powered by vector embeddings",
-            "tools": [
-                {
-                    "name": "mcp__mcp-vector-search__search_code",
-                    "description": "Search for code using semantic similarity",
-                    "inputSchema": {
-                        "type": "object",
-                        "properties": {
-                            "query": {"type": "string", "description": "The search query"},
-                            "limit": {"type": "integer", "default": 10},
-                            "similarity_threshold": {"type": "number", "default": 0.3},
-                            "language": {"type": "string"},
-                            "file_extensions": {"type": "array", "items": {"type": "string"}},
-                            "files": {"type": "string"},
-                            "class_name": {"type": "string"},
-                            "function_name": {"type": "string"}
+        base_def.update(
+            {
+                "description": "Semantic code search powered by vector embeddings",
+                "tools": [
+                    {
+                        "name": "mcp__mcp-vector-search__search_code",
+                        "description": "Search for code using semantic similarity",
+                        "inputSchema": {
+                            "type": "object",
+                            "properties": {
+                                "query": {
+                                    "type": "string",
+                                    "description": "The search query",
+                                },
+                                "limit": {"type": "integer", "default": 10},
+                                "similarity_threshold": {
+                                    "type": "number",
+                                    "default": 0.3,
+                                },
+                                "language": {"type": "string"},
+                                "file_extensions": {
+                                    "type": "array",
+                                    "items": {"type": "string"},
+                                },
+                                "files": {"type": "string"},
+                                "class_name": {"type": "string"},
+                                "function_name": {"type": "string"},
+                            },
+                            "required": ["query"],
                         },
-                        "required": ["query"]
-                    }
-                },
-                {
-                    "name": "mcp__mcp-vector-search__search_similar",
-                    "description": "Find code similar to a specific file or function",
-                    "inputSchema": {
-                        "type": "object",
-                        "properties": {
-                            "file_path": {"type": "string", "description": "Path to the file"},
-                            "function_name": {"type": "string"},
-                            "limit": {"type": "integer", "default": 10},
-                            "similarity_threshold": {"type": "number", "default": 0.3}
+                    },
+                    {
+                        "name": "mcp__mcp-vector-search__search_similar",
+                        "description": "Find code similar to a specific file or function",
+                        "inputSchema": {
+                            "type": "object",
+                            "properties": {
+                                "file_path": {
+                                    "type": "string",
+                                    "description": "Path to the file",
+                                },
+                                "function_name": {"type": "string"},
+                                "limit": {"type": "integer", "default": 10},
+                                "similarity_threshold": {
+                                    "type": "number",
+                                    "default": 0.3,
+                                },
+                            },
+                            "required": ["file_path"],
                         },
-                        "required": ["file_path"]
-                    }
-                },
-                {
-                    "name": "mcp__mcp-vector-search__search_context",
-                    "description": "Search for code based on contextual description",
-                    "inputSchema": {
-                        "type": "object",
-                        "properties": {
-                            "description": {"type": "string", "description": "Contextual description"},
-                            "focus_areas": {"type": "array", "items": {"type": "string"}},
-                            "limit": {"type": "integer", "default": 10}
+                    },
+                    {
+                        "name": "mcp__mcp-vector-search__search_context",
+                        "description": "Search for code based on contextual description",
+                        "inputSchema": {
+                            "type": "object",
+                            "properties": {
+                                "description": {
+                                    "type": "string",
+                                    "description": "Contextual description",
+                                },
+                                "focus_areas": {
+                                    "type": "array",
+                                    "items": {"type": "string"},
+                                },
+                                "limit": {"type": "integer", "default": 10},
+                            },
+                            "required": ["description"],
                         },
-                        "required": ["description"]
-                    }
-                },
-                {
-                    "name": "mcp__mcp-vector-search__get_project_status",
-                    "description": "Get project indexing status and statistics",
-                    "inputSchema": {
-                        "type": "object",
-                        "properties": {},
-                        "required": []
-                    }
-                },
-                {
-                    "name": "mcp__mcp-vector-search__index_project",
-                    "description": "Index or reindex the project codebase",
-                    "inputSchema": {
-                        "type": "object",
-                        "properties": {
-                            "force": {"type": "boolean", "default": False},
-                            "file_extensions": {"type": "array", "items": {"type": "string"}}
+                    },
+                    {
+                        "name": "mcp__mcp-vector-search__get_project_status",
+                        "description": "Get project indexing status and statistics",
+                        "inputSchema": {
+                            "type": "object",
+                            "properties": {},
+                            "required": [],
                         },
-                        "required": []
-                    }
-                }
-            ]
-        })
+                    },
+                    {
+                        "name": "mcp__mcp-vector-search__index_project",
+                        "description": "Index or reindex the project codebase",
+                        "inputSchema": {
+                            "type": "object",
+                            "properties": {
+                                "force": {"type": "boolean", "default": False},
+                                "file_extensions": {
+                                    "type": "array",
+                                    "items": {"type": "string"},
+                                },
+                            },
+                            "required": [],
+                        },
+                    },
+                ],
+            }
+        )
         return base_def
 
     async def invoke(self, tool_name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
@@ -197,9 +228,13 @@ class MCPVectorSearchService(ExternalMCPService):
 
             # Prepare the command
             cmd = [
-                sys.executable, "-m", "mcp_vector_search",
-                "--tool", actual_tool,
-                "--args", json.dumps(arguments)
+                sys.executable,
+                "-m",
+                "mcp_vector_search",
+                "--tool",
+                actual_tool,
+                "--args",
+                json.dumps(arguments),
             ]
 
             # Run the command
@@ -208,7 +243,8 @@ class MCPVectorSearchService(ExternalMCPService):
                 capture_output=True,
                 text=True,
                 timeout=30,
-                cwd=Path.cwd()  # Use current working directory for project context
+                cwd=Path.cwd(),
+                check=False,  # Use current working directory for project context
             )
 
             if result.returncode == 0:
@@ -235,47 +271,61 @@ class MCPBrowserService(ExternalMCPService):
     def get_definition(self) -> Dict[str, Any]:
         """Get tool definition for MCP registration."""
         base_def = super().get_definition()
-        base_def.update({
-            "description": "Web browsing and content extraction capabilities",
-            "tools": [
-                {
-                    "name": "mcp__mcp-browser__browse",
-                    "description": "Browse a webpage and extract content",
-                    "inputSchema": {
-                        "type": "object",
-                        "properties": {
-                            "url": {"type": "string", "description": "URL to browse"},
-                            "extract": {"type": "string", "description": "What to extract"}
+        base_def.update(
+            {
+                "description": "Web browsing and content extraction capabilities",
+                "tools": [
+                    {
+                        "name": "mcp__mcp-browser__browse",
+                        "description": "Browse a webpage and extract content",
+                        "inputSchema": {
+                            "type": "object",
+                            "properties": {
+                                "url": {
+                                    "type": "string",
+                                    "description": "URL to browse",
+                                },
+                                "extract": {
+                                    "type": "string",
+                                    "description": "What to extract",
+                                },
+                            },
+                            "required": ["url"],
                         },
-                        "required": ["url"]
-                    }
-                },
-                {
-                    "name": "mcp__mcp-browser__search",
-                    "description": "Search the web",
-                    "inputSchema": {
-                        "type": "object",
-                        "properties": {
-                            "query": {"type": "string", "description": "Search query"},
-                            "num_results": {"type": "integer", "default": 10}
+                    },
+                    {
+                        "name": "mcp__mcp-browser__search",
+                        "description": "Search the web",
+                        "inputSchema": {
+                            "type": "object",
+                            "properties": {
+                                "query": {
+                                    "type": "string",
+                                    "description": "Search query",
+                                },
+                                "num_results": {"type": "integer", "default": 10},
+                            },
+                            "required": ["query"],
                         },
-                        "required": ["query"]
-                    }
-                },
-                {
-                    "name": "mcp__mcp-browser__screenshot",
-                    "description": "Take a screenshot of a webpage",
-                    "inputSchema": {
-                        "type": "object",
-                        "properties": {
-                            "url": {"type": "string", "description": "URL to screenshot"},
-                            "full_page": {"type": "boolean", "default": False}
+                    },
+                    {
+                        "name": "mcp__mcp-browser__screenshot",
+                        "description": "Take a screenshot of a webpage",
+                        "inputSchema": {
+                            "type": "object",
+                            "properties": {
+                                "url": {
+                                    "type": "string",
+                                    "description": "URL to screenshot",
+                                },
+                                "full_page": {"type": "boolean", "default": False},
+                            },
+                            "required": ["url"],
                         },
-                        "required": ["url"]
-                    }
-                }
-            ]
-        })
+                    },
+                ],
+            }
+        )
         return base_def
 
     async def invoke(self, tool_name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
@@ -286,17 +336,18 @@ class MCPBrowserService(ExternalMCPService):
 
             # Prepare the command
             cmd = [
-                sys.executable, "-m", "mcp_browser",
-                "--tool", actual_tool,
-                "--args", json.dumps(arguments)
+                sys.executable,
+                "-m",
+                "mcp_browser",
+                "--tool",
+                actual_tool,
+                "--args",
+                json.dumps(arguments),
             ]
 
             # Run the command
             result = subprocess.run(
-                cmd,
-                capture_output=True,
-                text=True,
-                timeout=30
+                cmd, capture_output=True, text=True, timeout=30, check=False
             )
 
             if result.returncode == 0:
@@ -338,10 +389,7 @@ class ExternalMCPServiceManager:
         them as tools in the gateway - they run as separate MCP servers.
         """
         # Create service instances
-        services = [
-            MCPVectorSearchService(),
-            MCPBrowserService()
-        ]
+        services = [MCPVectorSearchService(), MCPBrowserService()]
 
         # Initialize each service
         initialized_services = []
@@ -350,10 +398,11 @@ class ExternalMCPServiceManager:
                 if await service.initialize():
                     initialized_services.append(service)
                     if self.logger:
-                        self.logger.info(f"Initialized external service: {service.service_name}")
-                else:
-                    if self.logger:
-                        self.logger.warning(f"Failed to initialize: {service.service_name}")
+                        self.logger.info(
+                            f"Initialized external service: {service.service_name}"
+                        )
+                elif self.logger:
+                    self.logger.warning(f"Failed to initialize: {service.service_name}")
             except Exception as e:
                 if self.logger:
                     self.logger.error(f"Error initializing {service.service_name}: {e}")
@@ -370,7 +419,9 @@ class ExternalMCPServiceManager:
                 all_tools.extend(service_def["tools"])
         return all_tools
 
-    async def invoke_tool(self, tool_name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
+    async def invoke_tool(
+        self, tool_name: str, arguments: Dict[str, Any]
+    ) -> Dict[str, Any]:
         """Invoke a tool from any registered external service."""
         # Find the service that handles this tool
         for service in self.services:
@@ -387,4 +438,6 @@ class ExternalMCPServiceManager:
                 await service.shutdown()
             except Exception as e:
                 if self.logger:
-                    self.logger.warning(f"Error shutting down {service.service_name}: {e}")
+                    self.logger.warning(
+                        f"Error shutting down {service.service_name}: {e}"
+                    )