Add new file

rag/agents/coordinator.py

@@ -0,0 +1,76 @@
+# rag/agents/coordinator.py
+"""
+Координатор агентов: управляет специализированными агентами и маршрутизирует запросы.
+"""
+
+import logging
+from typing import Dict, List, Optional, Any
+
+from rag.agents.roles import BaseAgent
+from rag.services.giga_client import GigaClient
+from rag.config_models import AppConfig
+
+logger = logging.getLogger(__name__)
+
+
+class AgentCoordinator:
+    """
+    Координатор, который выбирает подходящего агента для запроса и управляет
+    диалогом между агентами (при необходимости).
+    """
+
+    def __init__(self, config: AppConfig, giga: GigaClient):
+        self.config = config
+        self.giga = giga
+        self.agents: Dict[str, BaseAgent] = {}
+
+    def register_agent(self, name: str, agent: BaseAgent) -> None:
+        """Регистрирует агента с указанным именем."""
+        self.agents[name] = agent
+        logger.info(f"Агент '{name}' зарегистрирован")
+
+    async def process(
+        self,
+        query: str,
+        context: str,
+        history: List[Dict[str, str]],
+        tools: Optional[List[Dict[str, Any]]] = None,
+    ) -> str:
+        """
+        Основной метод: выбирает агента и передаёт ему запрос.
+        Если агент возвращает действие, выполняет его (или передаёт другому агенту).
+        """
+        # Простая маршрутизация: определяем ключевые слова
+        query_lower = query.lower()
+        # Можно использовать классификатор намерений или просто по ключевым словам
+        if any(kw in query_lower for kw in ["методолог", "стандарт", "гост", "методика", "регламент"]):
+            agent_name = "methodologist"
+        elif any(kw in query_lower for kw in ["коробочное", "типовое", "решение", "шаблон", "создай документ"]):
+            agent_name = "box_solution"
+        elif any(kw in query_lower for kw in ["командировка", "бронирование", "билет", "отель", "календарь"]):
+            agent_name = "personal_assistant"
+        else:
+            # Если не определили, можно использовать первый зарегистрированный или общий
+            agent_name = "methodologist"  # по умолчанию
+
+        agent = self.agents.get(agent_name)
+        if not agent:
+            logger.warning(f"Агент '{agent_name}' не найден, используем первого доступного")
+            if self.agents:
+                agent = next(iter(self.agents.values()))
+            else:
+                return "Нет доступных агентов для обработки запроса."
+
+        # Обрабатываем запрос агентом
+        result = await agent.process(query, context, history, tools)
+
+        if result.get("type") == "answer":
+            return result.get("content", "")
+        elif result.get("type") == "action":
+            # Если агент вернул действие (например, вызов инструмента), можно выполнить его
+            # и передать результат другому агенту или тому же самому.
+            # Для простоты мы просто вернём сообщение о необходимости действия.
+            action = result.get("content", {})
+            return f"Агент запросил действие: {action}"
+        else:
+            return "Неизвестный тип ответа от агента."