Added code samples for AI-Agents

agents/chat_agent.py

@@ -0,0 +1,115 @@
+"""ИИ-агент для чата 'Планета Земля'."""
+from typing import List, Optional
+from uuid import UUID
+
+from models.gigachat_types import GigaChatMessage
+from prompts.persona import EARTH_PERSONA
+
+from agents.gigachat_client import GigaChatClient
+from services.cache_service import CacheService
+
+
+class ChatAgent:
+    """ИИ-агент для общения с детьми и родителями."""
+
+    def __init__(self, gigachat: GigaChatClient, cache: CacheService):
+        self.gigachat = gigachat
+        self.cache = cache
+
+    async def chat(
+        self,
+        user_id: UUID,
+        message: str,
+        conversation_id: Optional[str] = None,
+        model: str = "GigaChat-2-Lite",
+    ) -> tuple[str, int]:
+        """
+        Отправить сообщение и получить ответ.
+
+        Args:
+            user_id: ID пользователя
+            message: Текст сообщения
+            conversation_id: ID разговора (для контекста)
+            model: Модель GigaChat
+
+        Returns:
+            (ответ, количество использованных токенов)
+        """
+        # Загружаем контекст из кэша
+        context_messages = []
+        if conversation_id:
+            cached_context = await self.cache.get_context(str(conversation_id))
+            context_messages = [
+                GigaChatMessage(role=msg["role"], content=msg["content"])
+                for msg in cached_context
+            ]
+
+        # Добавляем системный промпт в начало
+        system_message = GigaChatMessage(role="system", content=EARTH_PERSONA)
+        if not context_messages or context_messages[0].role != "system":
+            context_messages.insert(0, system_message)
+
+        # Добавляем текущее сообщение пользователя
+        context_messages.append(GigaChatMessage(role="user", content=message))
+
+        # Отправляем запрос
+        response = await self.gigachat.chat_with_response(
+            message=message,
+            context=context_messages,
+            model=model,
+            temperature=0.7,
+            max_tokens=1500,
+        )
+
+        assistant_message = response.choices[0].message.content
+        tokens_used = response.usage.total_tokens
+
+        # Сохраняем в контекст
+        if conversation_id:
+            await self.cache.add_message(str(conversation_id), "user", message)
+            await self.cache.add_message(str(conversation_id), "assistant", assistant_message)
+
+        return assistant_message, tokens_used
+
+    async def chat_with_context(
+        self,
+        user_id: UUID,
+        message: str,
+        context: Optional[List[dict]] = None,
+        model: str = "GigaChat-2-Lite",
+    ) -> tuple[str, int]:
+        """
+        Отправить сообщение с явным контекстом.
+
+        Args:
+            user_id: ID пользователя
+            message: Текст сообщения
+            context: Явный контекст разговора
+            model: Модель GigaChat
+
+        Returns:
+            (ответ, количество использованных токенов)
+        """
+        context_messages = [GigaChatMessage(role="system", content=EARTH_PERSONA)]
+
+        if context:
+            for msg in context:
+                context_messages.append(
+                    GigaChatMessage(role=msg["role"], content=msg["content"])
+                )
+
+        context_messages.append(GigaChatMessage(role="user", content=message))
+
+        response = await self.gigachat.chat_with_response(
+            message=message,
+            context=context_messages,
+            model=model,
+            temperature=0.7,
+            max_tokens=1500,
+        )
+
+        assistant_message = response.choices[0].message.content
+        tokens_used = response.usage.total_tokens
+
+        return assistant_message, tokens_used
+