mcp-proxy-adapter 6.3.4__py3-none-any.whl → 6.3.6__py3-none-any.whl

mcp_proxy_adapter/api/tools.py

@@ -19,46 +19,48 @@ logger = logging.getLogger(__name__)
 class TSTCommandExecutor:
     """
     Инструмент для выполнения команд через JSON-RPC протокол на сервере проекта.
-    
+
     Этот класс предоставляет функциональность для выполнения команд микросервиса
     через внешний интерфейс TST (Tool-System-Transport).
     """
-    
+
     name = "tst_execute_command"
     description = "Выполняет команду через JSON-RPC протокол."
-    
+
     @classmethod
-    async def execute(cls, command: str, params: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
+    async def execute(
+        cls, command: str, params: Optional[Dict[str, Any]] = None
+    ) -> Dict[str, Any]:
         """
         Выполняет команду с указанными параметрами.
-        
+
         Args:
             command: Имя команды для выполнения
             params: Параметры команды (опционально)
-            
+
         Returns:
             Результат выполнения команды
-            
+
         Raises:
             NotFoundError: Если команда не найдена
             InvalidParamsError: Если переданы некорректные параметры
         """
         if not params:
             params = {}
-            
+
         logger.info(f"Executing command via TST: {command}, params: {params}")
-        
+
         try:
             # Проверяем существование команды
             if not registry.command_exists_with_priority(command):
                 raise NotFoundError(f"Команда '{command}' не найдена")
-            
+
             # Получаем класс команды
             command_class = registry.get_command_with_priority(command)
-            
+
             # Выполняем команду
             result = await command_class.execute(**params)
-            
+
             # Возвращаем результат
             return result.to_dict()
         except NotFoundError as e:
@@ -67,79 +69,87 @@ class TSTCommandExecutor:
         except Exception as e:
             logger.exception(f"Error executing command {command}: {e}")
             raise
-    
+
     @classmethod
     def get_schema(cls) -> Dict[str, Any]:
         """
         Возвращает схему инструмента в формате OpenAPI.
-        
+
         Returns:
             Словарь со схемой инструмента
         """
         return ToolIntegration.generate_tool_schema(cls.name, registry, cls.description)
-    
+
     @classmethod
     def get_description(cls, format: str = "json") -> Union[Dict[str, Any], str]:
         """
         Возвращает полное описание инструмента в указанном формате.
-        
+
         Args:
             format: Формат описания (json, markdown, html)
-            
+
         Returns:
             Описание инструмента в указанном формате
         """
         if format.lower() == "json":
             # Получаем базовое описание инструмента
-            base_description = ToolIntegration.generate_tool_schema(cls.name, registry, cls.description)
-            
+            base_description = ToolIntegration.generate_tool_schema(
+                cls.name, registry, cls.description
+            )
+
             # Добавляем дополнительную информацию
             base_description["examples"] = cls._generate_examples()
             base_description["error_codes"] = cls._generate_error_codes()
-            
+
             return base_description
         elif format.lower() in ["markdown", "text", "html"]:
-            return ToolIntegration.generate_tool_documentation(cls.name, registry, format)
+            return ToolIntegration.generate_tool_documentation(
+                cls.name, registry, format
+            )
         else:
             # По умолчанию возвращаем JSON формат
-            return ToolIntegration.generate_tool_schema(cls.name, registry, cls.description)
-    
+            return ToolIntegration.generate_tool_schema(
+                cls.name, registry, cls.description
+            )
+
     @classmethod
     def _generate_examples(cls) -> List[Dict[str, Any]]:
         """
         Генерирует примеры использования инструмента.
-        
+
         Returns:
             Список примеров использования
         """
         examples = []
-        
+
         # Получаем метаданные всех команд
         all_metadata = registry.get_all_metadata()
-        
+
         # Добавляем по одному примеру для каждой команды
         for cmd_name, metadata in all_metadata.items():
             if metadata.get("examples"):
                 # Берем первый пример из метаданных
                 example = metadata["examples"][0]
-                
+
                 # Формируем пример использования инструмента
-                examples.append({
-                    "command": cls.name,
-                    "params": {
-                        "command": example.get("command", cmd_name),
-                        "params": example.get("params", {})
-                    },
-                    "description": f"Выполнение команды {cmd_name}"
-                })
-        
+                examples.append(
+                    {
+                        "command": cls.name,
+                        "params": {
+                            "command": example.get("command", cmd_name),
+                            "params": example.get("params", {}),
+                        },
+                        "description": f"Выполнение команды {cmd_name}",
+                    }
+                )
+
         return examples
-    
+
     @classmethod
     def _generate_error_codes(cls) -> Dict[str, str]:
         """
         Генерирует словарь возможных кодов ошибок.
-        
+
         Returns:
             Словарь с кодами ошибок и их описаниями
         """
@@ -148,27 +158,27 @@ class TSTCommandExecutor:
             "-32601": "Команда не найдена",
             "-32602": "Некорректные параметры",
             "-32603": "Внутренняя ошибка",
-            "-32000": "Ошибка выполнения команды"
+            "-32000": "Ошибка выполнения команды",
         }
 
 
 # Экспортируем доступные инструменты API
-available_tools = {
-    TSTCommandExecutor.name: TSTCommandExecutor
-}
+available_tools = {TSTCommandExecutor.name: TSTCommandExecutor}
 
 
-def get_tool_description(tool_name: str, format: str = "json") -> Union[Dict[str, Any], str]:
+def get_tool_description(
+    tool_name: str, format: str = "json"
+) -> Union[Dict[str, Any], str]:
     """
     Получает описание инструмента API по имени.
-    
+
     Args:
         tool_name: Имя инструмента API
         format: Формат описания (json, markdown, html)
-        
+
     Returns:
         Описание инструмента в указанном формате
-        
+
     Raises:
         NotFoundError: Если инструмент не найден
     """
@@ -181,18 +191,18 @@ def get_tool_description(tool_name: str, format: str = "json") -> Union[Dict[str
 async def execute_tool(tool_name: str, **params) -> Dict[str, Any]:
     """
     Выполняет инструмент API с указанными параметрами.
-    
+
     Args:
         tool_name: Имя инструмента API
         **params: Параметры инструмента
-        
+
     Returns:
         Результат выполнения инструмента
-        
+
     Raises:
         NotFoundError: Если инструмент не найден
     """
     if tool_name in available_tools:
         return await available_tools[tool_name].execute(**params)
     else:
-        raise NotFoundError(f"Инструмент '{tool_name}' не найден") 
+        raise NotFoundError(f"Инструмент '{tool_name}' не найден")

mcp_proxy_adapter/commands/__init__.py

@@ -4,22 +4,31 @@ Commands module initialization file.
 
 from mcp_proxy_adapter.commands.base import Command
 from mcp_proxy_adapter.commands.command_registry import registry, CommandRegistry
-from mcp_proxy_adapter.commands.dependency_container import container, DependencyContainer
+from mcp_proxy_adapter.commands.dependency_container import (
+    container,
+    DependencyContainer,
+)
 from mcp_proxy_adapter.commands.result import CommandResult, SuccessResult, ErrorResult
 from mcp_proxy_adapter.commands.auth_validation_command import AuthValidationCommand
 from mcp_proxy_adapter.commands.ssl_setup_command import SSLSetupCommand
-from mcp_proxy_adapter.commands.certificate_management_command import CertificateManagementCommand
+from mcp_proxy_adapter.commands.certificate_management_command import (
+    CertificateManagementCommand,
+)
 from mcp_proxy_adapter.commands.key_management_command import KeyManagementCommand
 from mcp_proxy_adapter.commands.cert_monitor_command import CertMonitorCommand
-from mcp_proxy_adapter.commands.transport_management_command import TransportManagementCommand
+from mcp_proxy_adapter.commands.transport_management_command import (
+    TransportManagementCommand,
+)
 from mcp_proxy_adapter.commands.role_test_command import RoleTestCommand
 from mcp_proxy_adapter.commands.echo_command import EchoCommand
-from mcp_proxy_adapter.commands.proxy_registration_command import ProxyRegistrationCommand
+from mcp_proxy_adapter.commands.proxy_registration_command import (
+    ProxyRegistrationCommand,
+)
 
 __all__ = [
     "Command",
     "CommandResult",
-    "SuccessResult", 
+    "SuccessResult",
     "ErrorResult",
     "registry",
     "CommandRegistry",
@@ -33,5 +42,5 @@ __all__ = [
     "TransportManagementCommand",
     "RoleTestCommand",
     "EchoCommand",
-    "ProxyRegistrationCommand"
+    "ProxyRegistrationCommand",
 ]