claude-mpm 3.9.11__py3-none-any.whl → 4.0.3__py3-none-any.whl

claude_mpm/services/mcp_gateway/tools/__init__.py

@@ -7,11 +7,12 @@ Tool adapters and implementations for the MCP Gateway service.
 
 from .base_adapter import (
     BaseToolAdapter,
-    EchoToolAdapter,
     CalculatorToolAdapter,
+    EchoToolAdapter,
     SystemInfoToolAdapter,
 )
 from .document_summarizer import DocumentSummarizerTool
+from .unified_ticket_tool import UnifiedTicketTool
 
 __all__ = [
     "BaseToolAdapter",
@@ -19,4 +20,5 @@ __all__ = [
     "CalculatorToolAdapter",
     "SystemInfoToolAdapter",
     "DocumentSummarizerTool",
-]
+    "UnifiedTicketTool",
+]

claude_mpm/services/mcp_gateway/tools/base_adapter.py

@@ -7,47 +7,46 @@ Base class for MCP tool adapters and example implementations.
 Part of ISS-0035: MCP Server Implementation - Core Server and Tool Registry
 """
 
-from abc import ABC
-from typing import Any, Dict, Optional
-import json
 import traceback
+from abc import ABC
 from datetime import datetime
+from typing import Any, Dict, Optional
 
+from claude_mpm.services.mcp_gateway.core.base import BaseMCPService
 from claude_mpm.services.mcp_gateway.core.interfaces import (
     IMCPToolAdapter,
     MCPToolDefinition,
     MCPToolInvocation,
     MCPToolResult,
 )
-from claude_mpm.services.mcp_gateway.core.base import BaseMCPService
 
 
 class BaseToolAdapter(BaseMCPService, IMCPToolAdapter, ABC):
     """
     Base class for MCP tool adapters.
-    
+
     WHY: This base class provides common functionality for all tool adapters,
     including parameter validation, error handling, and metrics tracking.
     Concrete tool implementations should extend this class.
-    
+
     DESIGN DECISIONS:
     - Provide default parameter validation using JSON Schema
     - Include standard error handling and logging
     - Track execution metrics for monitoring
     - Support both sync and async tool implementations
     """
-    
+
     def __init__(self, tool_definition: MCPToolDefinition):
         """
         Initialize the base tool adapter.
-        
+
         Args:
             tool_definition: Tool definition with schema and metadata
         """
         super().__init__(f"Tool-{tool_definition.name}")
         self._definition = tool_definition
         self._initialized = False
-        
+
         # Metrics
         self._metrics = {
             "invocations": 0,
@@ -56,28 +55,28 @@ class BaseToolAdapter(BaseMCPService, IMCPToolAdapter, ABC):
             "total_execution_time": 0.0,
             "average_execution_time": 0.0,
             "last_invocation": None,
-            "last_error": None
+            "last_error": None,
         }
-    
+
     def get_definition(self) -> MCPToolDefinition:
         """
         Get the tool definition.
-        
+
         Returns:
             Tool definition with schema and metadata
         """
         return self._definition
-    
+
     def validate_parameters(self, parameters: Dict[str, Any]) -> bool:
         """
         Validate tool parameters against schema.
-        
+
         Default implementation performs basic JSON Schema validation.
         Override for custom validation logic.
-        
+
         Args:
             parameters: Parameters to validate
-            
+
         Returns:
             True if parameters are valid
         """
@@ -86,36 +85,38 @@ class BaseToolAdapter(BaseMCPService, IMCPToolAdapter, ABC):
             schema = self._definition.input_schema
             required = schema.get("required", [])
             properties = schema.get("properties", {})
-            
+
             # Check required parameters
             for param in required:
                 if param not in parameters:
                     self.log_error(f"Missing required parameter: {param}")
                     return False
-            
+
             # Check parameter types (basic validation)
             for param_name, param_value in parameters.items():
                 if param_name in properties:
                     expected_type = properties[param_name].get("type")
                     if expected_type:
                         if not self._validate_type(param_value, expected_type):
-                            self.log_error(f"Invalid type for parameter {param_name}: expected {expected_type}")
+                            self.log_error(
+                                f"Invalid type for parameter {param_name}: expected {expected_type}"
+                            )
                             return False
-            
+
             return True
-            
+
         except Exception as e:
             self.log_error(f"Error validating parameters: {e}")
             return False
-    
+
     def _validate_type(self, value: Any, expected_type: str) -> bool:
         """
         Validate a value against an expected JSON Schema type.
-        
+
         Args:
             value: Value to validate
             expected_type: Expected type (string, number, boolean, array, object)
-            
+
         Returns:
             True if type matches
         """
@@ -125,22 +126,22 @@ class BaseToolAdapter(BaseMCPService, IMCPToolAdapter, ABC):
             "boolean": bool,
             "array": list,
             "object": dict,
-            "null": type(None)
+            "null": type(None),
         }
-        
+
         expected_python_type = type_map.get(expected_type)
         if expected_python_type:
             return isinstance(value, expected_python_type)
-        
+
         return True  # Unknown type, allow it
-    
+
     async def initialize(self) -> bool:
         """
         Initialize the tool adapter.
-        
+
         Default implementation marks as initialized.
         Override for custom initialization logic.
-        
+
         Returns:
             True if initialization successful
         """
@@ -148,49 +149,49 @@ class BaseToolAdapter(BaseMCPService, IMCPToolAdapter, ABC):
             self.log_info(f"Initializing tool: {self._definition.name}")
             self._initialized = True
             return True
-            
+
         except Exception as e:
             self.log_error(f"Failed to initialize tool: {e}")
             return False
-    
+
     async def shutdown(self) -> None:
         """
         Shutdown the tool adapter and clean up resources.
-        
+
         Default implementation marks as not initialized.
         Override for custom cleanup logic.
         """
         try:
             self.log_info(f"Shutting down tool: {self._definition.name}")
             self._initialized = False
-            
+
         except Exception as e:
             self.log_error(f"Error during tool shutdown: {e}")
-    
+
     def get_metrics(self) -> Dict[str, Any]:
         """
         Get tool metrics.
-        
+
         Returns:
             Metrics dictionary
         """
         return self._metrics.copy()
-    
+
     def _update_metrics(self, success: bool, execution_time: float) -> None:
         """
         Update tool metrics after invocation.
-        
+
         Args:
             success: Whether invocation was successful
             execution_time: Execution time in seconds
         """
         self._metrics["invocations"] += 1
-        
+
         if success:
             self._metrics["successes"] += 1
         else:
             self._metrics["failures"] += 1
-        
+
         self._metrics["total_execution_time"] += execution_time
         self._metrics["average_execution_time"] = (
             self._metrics["total_execution_time"] / self._metrics["invocations"]
@@ -201,10 +202,10 @@ class BaseToolAdapter(BaseMCPService, IMCPToolAdapter, ABC):
 class EchoToolAdapter(BaseToolAdapter):
     """
     Example tool adapter that echoes input back.
-    
+
     This is a simple example showing how to implement a concrete tool adapter.
     """
-    
+
     def __init__(self):
         """Initialize the echo tool."""
         definition = MCPToolDefinition(
@@ -213,73 +214,70 @@ class EchoToolAdapter(BaseToolAdapter):
             input_schema={
                 "type": "object",
                 "properties": {
-                    "message": {
-                        "type": "string",
-                        "description": "The message to echo"
-                    },
+                    "message": {"type": "string", "description": "The message to echo"},
                     "uppercase": {
                         "type": "boolean",
                         "description": "Whether to convert to uppercase",
-                        "default": False
-                    }
+                        "default": False,
+                    },
                 },
-                "required": ["message"]
-            }
+                "required": ["message"],
+            },
         )
         super().__init__(definition)
-    
+
     async def invoke(self, invocation: MCPToolInvocation) -> MCPToolResult:
         """
         Invoke the echo tool.
-        
+
         Args:
             invocation: Tool invocation request
-            
+
         Returns:
             Tool execution result with echoed message
         """
         start_time = datetime.now()
-        
+
         try:
             # Get parameters
             message = invocation.parameters.get("message", "")
             uppercase = invocation.parameters.get("uppercase", False)
-            
+
             # Process message
             result = message.upper() if uppercase else message
-            
+
             # Calculate execution time
             execution_time = (datetime.now() - start_time).total_seconds()
-            
+
             # Update metrics
             self._update_metrics(True, execution_time)
-            
+
             return MCPToolResult(
                 success=True,
                 data=result,
                 execution_time=execution_time,
-                metadata={"tool": "echo", "length": len(result)}
+                metadata={"tool": "echo", "length": len(result)},
             )
-            
+
         except Exception as e:
             execution_time = (datetime.now() - start_time).total_seconds()
             self._update_metrics(False, execution_time)
             self._metrics["last_error"] = str(e)
-            
+
             return MCPToolResult(
                 success=False,
                 error=f"Echo tool failed: {str(e)}",
-                execution_time=execution_time
+                execution_time=execution_time,
             )
 
 
 class CalculatorToolAdapter(BaseToolAdapter):
     """
     Example calculator tool adapter.
-    
+
     Demonstrates a more complex tool with multiple operations.
     """
-    
+
     def __init__(self):
         """Initialize the calculator tool."""
         definition = MCPToolDefinition(
@@ -291,53 +289,47 @@ class CalculatorToolAdapter(BaseToolAdapter):
                     "operation": {
                         "type": "string",
                         "enum": ["add", "subtract", "multiply", "divide"],
-                        "description": "The mathematical operation to perform"
+                        "description": "The mathematical operation to perform",
                     },
-                    "a": {
-                        "type": "number",
-                        "description": "First operand"
-                    },
-                    "b": {
-                        "type": "number",
-                        "description": "Second operand"
-                    }
+                    "a": {"type": "number", "description": "First operand"},
+                    "b": {"type": "number", "description": "Second operand"},
                 },
-                "required": ["operation", "a", "b"]
+                "required": ["operation", "a", "b"],
             },
             output_schema={
                 "type": "object",
                 "properties": {
                     "result": {
                         "type": "number",
-                        "description": "The calculation result"
+                        "description": "The calculation result",
                     },
                     "expression": {
                         "type": "string",
-                        "description": "The mathematical expression"
-                    }
-                }
-            }
+                        "description": "The mathematical expression",
+                    },
+                },
+            },
         )
         super().__init__(definition)
-    
+
     async def invoke(self, invocation: MCPToolInvocation) -> MCPToolResult:
         """
         Invoke the calculator tool.
-        
+
         Args:
             invocation: Tool invocation request
-            
+
         Returns:
             Tool execution result with calculation
         """
         start_time = datetime.now()
-        
+
         try:
             # Get parameters
             operation = invocation.parameters["operation"]
             a = invocation.parameters["a"]
             b = invocation.parameters["b"]
-            
+
             # Perform calculation
             if operation == "add":
                 result = a + b
@@ -355,42 +347,39 @@ class CalculatorToolAdapter(BaseToolAdapter):
                 expression = f"{a} / {b} = {result}"
             else:
                 raise ValueError(f"Unknown operation: {operation}")
-            
+
             # Calculate execution time
             execution_time = (datetime.now() - start_time).total_seconds()
-            
+
             # Update metrics
             self._update_metrics(True, execution_time)
-            
+
             return MCPToolResult(
                 success=True,
-                data={
-                    "result": result,
-                    "expression": expression
-                },
+                data={"result": result, "expression": expression},
                 execution_time=execution_time,
-                metadata={"tool": "calculator", "operation": operation}
+                metadata={"tool": "calculator", "operation": operation},
             )
-            
+
         except Exception as e:
             execution_time = (datetime.now() - start_time).total_seconds()
             self._update_metrics(False, execution_time)
             self._metrics["last_error"] = str(e)
-            
+
             return MCPToolResult(
                 success=False,
                 error=f"Calculator tool failed: {str(e)}",
-                execution_time=execution_time
+                execution_time=execution_time,
             )
 
 
 class SystemInfoToolAdapter(BaseToolAdapter):
     """
     Example system information tool adapter.
-    
+
     Demonstrates async operations and system interaction.
     """
-    
+
     def __init__(self):
         """Initialize the system info tool."""
         definition = MCPToolDefinition(
@@ -402,32 +391,33 @@ class SystemInfoToolAdapter(BaseToolAdapter):
                     "info_type": {
                         "type": "string",
                         "enum": ["platform", "memory", "cpu", "time"],
-                        "description": "Type of system information to retrieve"
+                        "description": "Type of system information to retrieve",
                     }
                 },
-                "required": ["info_type"]
-            }
+                "required": ["info_type"],
+            },
         )
         super().__init__(definition)
-    
+
     async def invoke(self, invocation: MCPToolInvocation) -> MCPToolResult:
         """
         Invoke the system info tool.
-        
+
         Args:
             invocation: Tool invocation request
-            
+
         Returns:
             Tool execution result with system information
         """
         start_time = datetime.now()
-        
+
         try:
             import platform
+
             import psutil
 
             info_type = invocation.parameters["info_type"]
-            
+
             if info_type == "platform":
                 result = {
                     "system": platform.system(),
@@ -435,7 +425,7 @@ class SystemInfoToolAdapter(BaseToolAdapter):
                     "version": platform.version(),
                     "machine": platform.machine(),
                     "processor": platform.processor(),
-                    "python_version": platform.python_version()
+                    "python_version": platform.python_version(),
                 }
             elif info_type == "memory":
                 mem = psutil.virtual_memory()
@@ -444,53 +434,53 @@ class SystemInfoToolAdapter(BaseToolAdapter):
                     "available": mem.available,
                     "percent": mem.percent,
                     "used": mem.used,
-                    "free": mem.free
+                    "free": mem.free,
                 }
             elif info_type == "cpu":
                 result = {
                     "count": psutil.cpu_count(),
                     "percent": psutil.cpu_percent(interval=0.1),
-                    "freq": psutil.cpu_freq()._asdict() if psutil.cpu_freq() else None
+                    "freq": psutil.cpu_freq()._asdict() if psutil.cpu_freq() else None,
                 }
             elif info_type == "time":
                 result = {
                     "current": datetime.now().isoformat(),
                     "timestamp": datetime.now().timestamp(),
-                    "timezone": str(datetime.now().astimezone().tzinfo)
+                    "timezone": str(datetime.now().astimezone().tzinfo),
                 }
             else:
                 raise ValueError(f"Unknown info type: {info_type}")
-            
+
             # Calculate execution time
             execution_time = (datetime.now() - start_time).total_seconds()
-            
+
             # Update metrics
             self._update_metrics(True, execution_time)
-            
+
             return MCPToolResult(
                 success=True,
                 data=result,
                 execution_time=execution_time,
-                metadata={"tool": "system_info", "info_type": info_type}
+                metadata={"tool": "system_info", "info_type": info_type},
             )
-            
+
         except ImportError as e:
             # Handle missing psutil dependency gracefully
             execution_time = (datetime.now() - start_time).total_seconds()
             self._update_metrics(False, execution_time)
-            
+
             return MCPToolResult(
                 success=False,
                 error=f"System info tool requires psutil: {str(e)}",
-                execution_time=execution_time
+                execution_time=execution_time,
             )
         except Exception as e:
             execution_time = (datetime.now() - start_time).total_seconds()
             self._update_metrics(False, execution_time)
             self._metrics["last_error"] = str(e)
-            
+
             return MCPToolResult(
                 success=False,
                 error=f"System info tool failed: {str(e)}",
-                execution_time=execution_time
-            )
+                execution_time=execution_time,
+            )