Added code samples for AI-Agents

tests/test_gigachat_client.py

@@ -0,0 +1,92 @@
+"""Тесты для GigaChat клиента."""
+import pytest
+from unittest.mock import AsyncMock, MagicMock, patch
+
+from agents.gigachat_client import GigaChatClient
+from models.gigachat_types import GigaChatMessage, GigaChatResponse, GigaChatUsage, GigaChatChoice
+from services.token_manager import TokenManager
+
+
+@pytest.fixture
+def token_manager():
+    """Фикстура для TokenManager."""
+    manager = TokenManager(
+        client_id="test_id",
+        client_secret="test_secret",
+    )
+    manager._access_token = "test_token"
+    manager._expires_at = 9999999999
+    return manager
+
+
+@pytest.fixture
+def gigachat_client(token_manager):
+    """Фикстура для GigaChatClient."""
+    return GigaChatClient(token_manager=token_manager)
+
+
+@pytest.mark.asyncio
+async def test_chat_success(gigachat_client):
+    """Тест успешного запроса к GigaChat."""
+    mock_response = GigaChatResponse(
+        id="test_id",
+        object="chat.completion",
+        created=1234567890,
+        model="GigaChat-2",
+        choices=[
+            GigaChatChoice(
+                message=GigaChatMessage(role="assistant", content="Тестовый ответ"),
+                index=0,
+                finish_reason="stop",
+            )
+        ],
+        usage=GigaChatUsage(
+            prompt_tokens=10,
+            completion_tokens=5,
+            total_tokens=15,
+        ),
+    )
+
+    with patch("aiohttp.ClientSession.post") as mock_post:
+        mock_response_obj = AsyncMock()
+        mock_response_obj.status = 200
+        mock_response_obj.json = AsyncMock(return_value=mock_response.model_dump())
+        mock_post.return_value.__aenter__.return_value = mock_response_obj
+
+        response = await gigachat_client.chat("Привет!")
+
+        assert response == "Тестовый ответ"
+
+
+@pytest.mark.asyncio
+async def test_chat_with_context(gigachat_client):
+    """Тест запроса с контекстом."""
+    context = [
+        GigaChatMessage(role="system", content="Ты помощник"),
+        GigaChatMessage(role="user", content="Привет"),
+    ]
+
+    mock_response = GigaChatResponse(
+        id="test_id",
+        object="chat.completion",
+        created=1234567890,
+        model="GigaChat-2",
+        choices=[
+            GigaChatChoice(
+                message=GigaChatMessage(role="assistant", content="Ответ с контекстом"),
+                index=0,
+            )
+        ],
+        usage=GigaChatUsage(prompt_tokens=20, completion_tokens=10, total_tokens=30),
+    )
+
+    with patch("aiohttp.ClientSession.post") as mock_post:
+        mock_response_obj = AsyncMock()
+        mock_response_obj.status = 200
+        mock_response_obj.json = AsyncMock(return_value=mock_response.model_dump())
+        mock_post.return_value.__aenter__.return_value = mock_response_obj
+
+        response = await gigachat_client.chat("Как дела?", context=context)
+
+        assert response == "Ответ с контекстом"
+