mcp-proxy-adapter 2.0.0__py3-none-any.whl → 2.0.2__py3-none-any.whl

mcp_proxy_adapter/schema.py

@@ -0,0 +1,257 @@
+"""
+Module for optimizing OpenAPI schema for MCP Proxy.
+"""
+from typing import Dict, Any, List, Optional
+
+class SchemaOptimizer:
+    """
+    OpenAPI schema optimizer for use with MCP Proxy.
+    
+    This class transforms a standard OpenAPI schema into a format
+    more suitable for use with MCP Proxy and AI models.
+    """
+    
+    def optimize(
+        self, 
+        schema: Dict[str, Any], 
+        cmd_endpoint: str,
+        commands_info: Dict[str, Dict[str, Any]]
+    ) -> Dict[str, Any]:
+        """
+        Optimizes OpenAPI schema for MCP Proxy.
+        
+        Args:
+            schema: Original OpenAPI schema
+            cmd_endpoint: Path for universal JSON-RPC endpoint
+            commands_info: Information about registered commands
+            
+        Returns:
+            Dict[str, Any]: Optimized schema
+        """
+        # Create a new schema in a format compatible with the proxy
+        optimized = {
+            "openapi": "3.0.2",
+            "info": {
+                "title": "Command Registry API",
+                "description": "API for executing commands through MCPProxy",
+                "version": "1.0.0"
+            },
+            "paths": {
+                cmd_endpoint: {
+                    "post": {
+                        "summary": "Execute command",
+                        "description": "Universal endpoint for executing various commands",
+                        "operationId": "execute_command",
+                        "requestBody": {
+                            "content": {
+                                "application/json": {
+                                    "schema": {
+                                        "$ref": "#/components/schemas/CommandRequest"
+                                    },
+                                    "examples": {}
+                                }
+                            },
+                            "required": True
+                        },
+                        "responses": {
+                            "200": {
+                                "description": "Command executed successfully",
+                                "content": {
+                                    "application/json": {
+                                        "schema": {
+                                            "$ref": "#/components/schemas/CommandResponse"
+                                        }
+                                    }
+                                }
+                            },
+                            "400": {
+                                "description": "Error in request or during command execution",
+                                "content": {
+                                    "application/json": {
+                                        "schema": {
+                                            "type": "object",
+                                            "properties": {
+                                                "detail": {
+                                                    "type": "string",
+                                                    "description": "Error description"
+                                                }
+                                            }
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                }
+            },
+            "components": {
+                "schemas": {
+                    "CommandRequest": {
+                        "title": "CommandRequest",
+                        "description": "Command execution request",
+                        "type": "object",
+                        "required": ["command"],
+                        "properties": {
+                            "command": {
+                                "title": "Command",
+                                "description": "Command to execute",
+                                "type": "string",
+                                "enum": list(commands_info.keys())
+                            },
+                            "params": {
+                                "title": "Parameters",
+                                "description": "Command parameters, depend on command type",
+                                "oneOf": []
+                            }
+                        }
+                    },
+                    "CommandResponse": {
+                        "title": "CommandResponse",
+                        "description": "Command execution response",
+                        "type": "object",
+                        "required": ["result"],
+                        "properties": {
+                            "result": {
+                                "title": "Result",
+                                "description": "Command execution result"
+                            }
+                        }
+                    }
+                },
+                "examples": {}
+            }
+        }
+        
+        # Add parameter schemas and examples for each command
+        for cmd_name, cmd_info in commands_info.items():
+            param_schema_name = f"{cmd_name.capitalize()}Params"
+            params = cmd_info.get("params", {})
+            has_params = bool(params)
+
+            # Define parameter schema (даже если params пустой, схема будет пустым объектом)
+            param_schema = {
+                "title": param_schema_name,
+                "description": f"Parameters for command {cmd_name}",
+                "type": "object",
+                "properties": {},
+            }
+            required_params = []
+            example_params = {}
+
+            for param_name, param_info in params.items():
+                param_property = {
+                    "title": param_name.capitalize(),
+                    "type": param_info.get("type", "string"),
+                    "description": param_info.get("description", "")
+                }
+                if "default" in param_info:
+                    param_property["default"] = param_info["default"]
+                if "enum" in param_info:
+                    param_property["enum"] = param_info["enum"]
+                param_schema["properties"][param_name] = param_property
+                if param_info.get("required", False):
+                    required_params.append(param_name)
+                if "example" in param_info:
+                    example_params[param_name] = param_info["example"]
+                elif "default" in param_info:
+                    example_params[param_name] = param_info["default"]
+                elif param_info.get("type") == "string":
+                    example_params[param_name] = "example_value"
+                elif param_info.get("type") == "integer":
+                    example_params[param_name] = 1
+                elif param_info.get("type") == "boolean":
+                    example_params[param_name] = False
+
+            if required_params:
+                param_schema["required"] = required_params
+
+            # Добавляем схему параметров всегда, даже если она пустая
+            optimized["components"]["schemas"][param_schema_name] = param_schema
+
+            # Добавляем $ref на схему параметров в oneOf всегда
+            optimized["components"]["schemas"]["CommandRequest"]["properties"]["params"]["oneOf"].append({
+                "$ref": f"#/components/schemas/{param_schema_name}"
+            })
+
+            # Пример использования команды
+            example_id = f"{cmd_name}_example"
+            example = {
+                "summary": f"Example of using command {cmd_name}",
+                "value": {
+                    "command": cmd_name
+                }
+            }
+            if has_params:
+                example["value"]["params"] = example_params
+            optimized["components"]["examples"][example_id] = example
+            optimized["paths"][cmd_endpoint]["post"]["requestBody"]["content"]["application/json"]["examples"][example_id] = {
+                "$ref": f"#/components/examples/{example_id}"
+            }
+
+        # Для команд без параметров добавляем type: null в oneOf
+        optimized_oneof = optimized["components"]["schemas"]["CommandRequest"]["properties"]["params"]["oneOf"]
+        optimized_oneof.append({"type": "null"})
+
+        # Add tool descriptions to schema for AI models
+        self._add_tool_descriptions(optimized, commands_info)
+        return optimized
+    
+    def _add_tool_descriptions(
+        self, 
+        schema: Dict[str, Any],
+        commands_info: Dict[str, Dict[str, Any]]
+    ) -> None:
+        """
+        Adds AI tool descriptions to the schema.
+        
+        This method enhances the OpenAPI schema with special descriptions
+        for better integration with AI models and MCPProxy.
+        
+        Args:
+            schema: OpenAPI schema to enhance
+            commands_info: Information about registered commands
+        """
+        # Add AI tool descriptions to x-mcp-tools
+        schema["x-mcp-tools"] = []
+        
+        for cmd_name, cmd_info in commands_info.items():
+            # Create tool description
+            tool = {
+                "name": f"mcp_{cmd_name}",  # Add mcp_ prefix to command name
+                "description": cmd_info.get("description", "") or cmd_info.get("summary", ""),
+                "parameters": {
+                    "type": "object",
+                    "properties": {},
+                    "required": []
+                }
+            }
+            
+            # Add parameters
+            for param_name, param_info in cmd_info.get("params", {}).items():
+                # Convert parameter to JSON Schema format
+                param_schema = {}
+                
+                # Parameter type
+                param_schema["type"] = param_info.get("type", "string")
+                
+                # Description
+                if "description" in param_info:
+                    param_schema["description"] = param_info["description"]
+                
+                # Default value
+                if "default" in param_info:
+                    param_schema["default"] = param_info["default"]
+                
+                # Possible values
+                if "enum" in param_info:
+                    param_schema["enum"] = param_info["enum"]
+                
+                # Add parameter to schema
+                tool["parameters"]["properties"][param_name] = param_schema
+                
+                # If parameter is required, add to required list
+                if param_info.get("required", False):
+                    tool["parameters"]["required"].append(param_name)
+            
+            # Add tool to list
+            schema["x-mcp-tools"].append(tool) 

{mcp_proxy_adapter-2.0.0.dist-info → mcp_proxy_adapter-2.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-proxy-adapter
-Version: 2.0.0
+Version: 2.0.2
 Summary: Adapter for exposing Command Registry commands as tools for AI models via MCP Proxy.
 Home-page: https://github.com/vasilyvz/mcp-proxy-adapter
 Author: Vasiliy VZ

mcp_proxy_adapter-2.0.2.dist-info/RECORD

@@ -0,0 +1,33 @@
+mcp_proxy_adapter/__init__.py,sha256=9qzzMqbq2lbVI4wGAsFSLqTKMGDfnfaCXb5YPKqFt3s,463
+mcp_proxy_adapter/adapter.py,sha256=76dkVeDuqLsJ5AhuftzLlwy2M6yr_PfNbmNfo9dXVhc,28844
+mcp_proxy_adapter/models.py,sha256=8zVWU6ly18pWozOnKQ2gsGpmTgL37-fFE_Fr1SDW-Nk,2530
+mcp_proxy_adapter/registry.py,sha256=jgC4TKaPbMbAsoxvGp2ToaOE4drD-VfZug7WJbm4IW4,15853
+mcp_proxy_adapter/schema.py,sha256=HZM0TTQTSi8ha1TEeVevdCyGZOUPoT1soB7Nex0hV50,10947
+mcp_proxy_adapter/adapters/__init__.py,sha256=7QraK1j5y29KFSRF4mVxTiWC2Y3IYZLd6GhxUCtjyhg,790
+mcp_proxy_adapter/analyzers/__init__.py,sha256=2rcYZDP-bXq078MQpxP32lAwYYyRhOwAQGBcefBfBzY,368
+mcp_proxy_adapter/analyzers/docstring_analyzer.py,sha256=T3FLJEo_uChShfiEKRl8GpVoHvh5HiudZkxnj4KixfA,7541
+mcp_proxy_adapter/analyzers/type_analyzer.py,sha256=6Wac7osKwF03waFSwQ8ZM0Wqn_zAP2D-I4WMEpR0hQM,5230
+mcp_proxy_adapter/cli/__init__.py,sha256=vJ0VTT7D4KRIOi9nQSgBLub7xywq68i2R1zYQkdA0Uk,416
+mcp_proxy_adapter/cli/__main__.py,sha256=qtjEOB4vkeMASjTppyRFok4IFihRrVOqjk6vLx8OW1g,2400
+mcp_proxy_adapter/cli/command_runner.py,sha256=4d0BZKRG9k4CLk3ZE26jfSxwwrls2e-KsflRoVimfyE,8328
+mcp_proxy_adapter/dispatchers/__init__.py,sha256=FWgimgInGphIjCEnvA3-ZExiapUzYAVis2H9C5IWivU,365
+mcp_proxy_adapter/dispatchers/base_dispatcher.py,sha256=S5_Xri058jAmOWeit1tedB_GMZQ9RLcNcYabA83ZF6k,2288
+mcp_proxy_adapter/dispatchers/json_rpc_dispatcher.py,sha256=ffu1M32E1AdC7IB44mlbV2L56eJQMsp-7fYi_r4rmHc,6331
+mcp_proxy_adapter/generators/__init__.py,sha256=KC8p2HcIBqdyCY1wHnN_c1EfbnPM39YhZLNBDcuv3vA,538
+mcp_proxy_adapter/generators/endpoint_generator.py,sha256=fhEInY3JPwdc5EENesN9VrrONwVZMpsledVpubouJLA,6513
+mcp_proxy_adapter/generators/openapi_generator.py,sha256=0Iqsn0fRr10sTWUK9-MU5MXOXKDHccxKYnmaiQPkKFw,9773
+mcp_proxy_adapter/generators/rest_api_generator.py,sha256=Zg9HVeIuspowXd6aBlyho4UXz-te_Hd4egGzjjIVF48,9159
+mcp_proxy_adapter/openapi_schema/__init__.py,sha256=S_lEg-VtlgDdhFxB_u9ZUM1dWKbzW_L_FM-6OCT4EXM,1334
+mcp_proxy_adapter/openapi_schema/command_registry.py,sha256=VqqdsVCcd5neqZ-bsfeKTHeLBeputwUZHxvNxVuvqZs,12075
+mcp_proxy_adapter/openapi_schema/rest_schema.py,sha256=nfF9axtmEA-B-Rw8TCnpmRf6Br3MfQ3TJVx-QHUHoy8,24103
+mcp_proxy_adapter/openapi_schema/rpc_generator.py,sha256=5FGMqjMXcaW0Ej6oey5mgiIxtEKTtA87dc-_kWLWqb0,13640
+mcp_proxy_adapter/openapi_schema/rpc_schema.py,sha256=pebSNt9rKyP2dyHpbjSllak6pQe5AnllReKHhR3UIaI,20162
+mcp_proxy_adapter/validators/__init__.py,sha256=s6zd5oZ3V2pXtzONoH3CiZYLzSVtCoLXpETMFUQfwWY,403
+mcp_proxy_adapter/validators/base_validator.py,sha256=qTEcXnfGqsgKKtc-lnttLiPTuq6vOHfDpZrQ5oP5Fi0,602
+mcp_proxy_adapter/validators/docstring_validator.py,sha256=Onpq2iNJ1qF4ejkJJIlBkLROuSNIVALHVmXIgkCpaFI,2934
+mcp_proxy_adapter/validators/metadata_validator.py,sha256=uCrn38-VYYn89l6f5CC_GoTAHAweaOW2Z6Esro1rtGw,3155
+mcp_proxy_adapter-2.0.2.dist-info/licenses/LICENSE,sha256=OkApFEwdgMCt_mbvUI-eIwKMSTe38K3XnU2DT5ub-wI,1072
+mcp_proxy_adapter-2.0.2.dist-info/METADATA,sha256=3-LXLlAgg1swHpgFqA015-zEi_kQ9jAfNoe6e-ExKPk,6076
+mcp_proxy_adapter-2.0.2.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
+mcp_proxy_adapter-2.0.2.dist-info/top_level.txt,sha256=JZT7vPLBYrtroX-ij68JBhJYbjDdghcV-DFySRy-Nnw,18
+mcp_proxy_adapter-2.0.2.dist-info/RECORD,,

mcp_proxy_adapter-2.0.2.dist-info/top_level.txt

@@ -0,0 +1 @@
+mcp_proxy_adapter

mcp_proxy_adapter-2.0.0.dist-info/RECORD

@@ -1,28 +0,0 @@
-adapters/__init__.py,sha256=7QraK1j5y29KFSRF4mVxTiWC2Y3IYZLd6GhxUCtjyhg,790
-analyzers/__init__.py,sha256=2rcYZDP-bXq078MQpxP32lAwYYyRhOwAQGBcefBfBzY,368
-analyzers/docstring_analyzer.py,sha256=T3FLJEo_uChShfiEKRl8GpVoHvh5HiudZkxnj4KixfA,7541
-analyzers/type_analyzer.py,sha256=6Wac7osKwF03waFSwQ8ZM0Wqn_zAP2D-I4WMEpR0hQM,5230
-cli/__init__.py,sha256=vJ0VTT7D4KRIOi9nQSgBLub7xywq68i2R1zYQkdA0Uk,416
-cli/__main__.py,sha256=qtjEOB4vkeMASjTppyRFok4IFihRrVOqjk6vLx8OW1g,2400
-cli/command_runner.py,sha256=4d0BZKRG9k4CLk3ZE26jfSxwwrls2e-KsflRoVimfyE,8328
-dispatchers/__init__.py,sha256=FWgimgInGphIjCEnvA3-ZExiapUzYAVis2H9C5IWivU,365
-dispatchers/base_dispatcher.py,sha256=S5_Xri058jAmOWeit1tedB_GMZQ9RLcNcYabA83ZF6k,2288
-dispatchers/json_rpc_dispatcher.py,sha256=ffu1M32E1AdC7IB44mlbV2L56eJQMsp-7fYi_r4rmHc,6331
-generators/__init__.py,sha256=KC8p2HcIBqdyCY1wHnN_c1EfbnPM39YhZLNBDcuv3vA,538
-generators/endpoint_generator.py,sha256=fhEInY3JPwdc5EENesN9VrrONwVZMpsledVpubouJLA,6513
-generators/openapi_generator.py,sha256=0Iqsn0fRr10sTWUK9-MU5MXOXKDHccxKYnmaiQPkKFw,9773
-generators/rest_api_generator.py,sha256=Zg9HVeIuspowXd6aBlyho4UXz-te_Hd4egGzjjIVF48,9159
-mcp_proxy_adapter-2.0.0.dist-info/licenses/LICENSE,sha256=OkApFEwdgMCt_mbvUI-eIwKMSTe38K3XnU2DT5ub-wI,1072
-openapi_schema/__init__.py,sha256=S_lEg-VtlgDdhFxB_u9ZUM1dWKbzW_L_FM-6OCT4EXM,1334
-openapi_schema/command_registry.py,sha256=VqqdsVCcd5neqZ-bsfeKTHeLBeputwUZHxvNxVuvqZs,12075
-openapi_schema/rest_schema.py,sha256=nfF9axtmEA-B-Rw8TCnpmRf6Br3MfQ3TJVx-QHUHoy8,24103
-openapi_schema/rpc_generator.py,sha256=5FGMqjMXcaW0Ej6oey5mgiIxtEKTtA87dc-_kWLWqb0,13640
-openapi_schema/rpc_schema.py,sha256=pebSNt9rKyP2dyHpbjSllak6pQe5AnllReKHhR3UIaI,20162
-validators/__init__.py,sha256=s6zd5oZ3V2pXtzONoH3CiZYLzSVtCoLXpETMFUQfwWY,403
-validators/base_validator.py,sha256=qTEcXnfGqsgKKtc-lnttLiPTuq6vOHfDpZrQ5oP5Fi0,602
-validators/docstring_validator.py,sha256=Onpq2iNJ1qF4ejkJJIlBkLROuSNIVALHVmXIgkCpaFI,2934
-validators/metadata_validator.py,sha256=uCrn38-VYYn89l6f5CC_GoTAHAweaOW2Z6Esro1rtGw,3155
-mcp_proxy_adapter-2.0.0.dist-info/METADATA,sha256=AhX6kieFNWrG7fY7tRv9Dk1L31zIW2knezWJK9HnyFQ,6076
-mcp_proxy_adapter-2.0.0.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
-mcp_proxy_adapter-2.0.0.dist-info/top_level.txt,sha256=SlVzE4ry_0xTgDmO2uiaI7ihb9MpR62izN4MH_t1BTU,72
-mcp_proxy_adapter-2.0.0.dist-info/RECORD,,

mcp_proxy_adapter-2.0.0.dist-info/top_level.txt

@@ -1,7 +0,0 @@
-adapters
-analyzers
-cli
-dispatchers
-generators
-openapi_schema
-validators