MyAIServ MCP Server

""" Базовый класс для промптов MCP. """ from abc import abstractmethod from typing import Any, Dict, List, Optional from app.core.base.component import MCPComponent class MCPPrompt(MCPComponent): """ Базовый класс для всех промптов MCP. Промпты представляют шаблоны для генерации сообщений, которые могут быть использованы для взаимодействия с LLM. Attributes: name: Имя промпта description: Описание промпта template: Шаблон промпта arguments: Аргументы промпта """ def __init__( self, name: str, description: Optional[str] = None, template: Optional[str] = None, arguments: Optional[List[Dict[str, Any]]] = None, ) -> None: """ Инициализирует промпт. Args: name: Имя промпта description: Описание промпта template: Шаблон промпта arguments: Аргументы промпта """ self.name = name self.description = description self.template = template self.arguments = arguments or [] async def initialize(self) -> bool: """ Инициализирует промпт. Returns: bool: True, если инициализация прошла успешно """ return True async def cleanup(self) -> bool: """ Освобождает ресурсы, используемые промптом. Returns: bool: True, если очистка прошла успешно """ return True @abstractmethod async def generate_messages( self, arguments: Dict[str, Any] ) -> List[Dict[str, Any]]: """ Генерирует сообщения на основе аргументов. Args: arguments: Аргументы для генерации сообщений Returns: List[Dict[str, Any]]: Сгенерированные сообщения """ pass