Added code samples for AI-Agents

services/token_manager.py

@@ -0,0 +1,69 @@
+"""Управление токенами GigaChat."""
+import os
+import time
+from typing import Optional
+
+import aiohttp
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+class TokenManager:
+    """Менеджер токенов для GigaChat API."""
+
+    def __init__(
+        self,
+        client_id: Optional[str] = None,
+        client_secret: Optional[str] = None,
+        auth_url: Optional[str] = None,
+    ):
+        self.client_id = client_id or os.getenv("GIGACHAT_CLIENT_ID")
+        self.client_secret = client_secret or os.getenv("GIGACHAT_CLIENT_SECRET")
+        self.auth_url = auth_url or os.getenv(
+            "GIGACHAT_AUTH_URL", "https://ngw.devices.sberbank.ru:9443/api/v2/oauth"
+        )
+        self._access_token: Optional[str] = None
+        self._expires_at: float = 0
+
+    async def get_token(self, force_refresh: bool = False) -> str:
+        """
+        Получить актуальный токен доступа.
+
+        Args:
+            force_refresh: Принудительно обновить токен
+
+        Returns:
+            Токен доступа
+        """
+        if not force_refresh and self._access_token and time.time() < self._expires_at:
+            return self._access_token
+
+        async with aiohttp.ClientSession() as session:
+            auth = aiohttp.BasicAuth(self.client_id, self.client_secret)
+            async with session.post(
+                self.auth_url,
+                auth=auth,
+                data={"scope": "GIGACHAT_API_PERS"},
+            ) as response:
+                if response.status != 200:
+                    error_text = await response.text()
+                    raise Exception(f"Failed to get token: {response.status} - {error_text}")
+
+                data = await response.json()
+                self._access_token = data["access_token"]
+                # Токен обычно действителен 30 минут, обновляем за 5 минут до истечения
+                expires_in = data.get("expires_in", 1800)
+                self._expires_at = time.time() + expires_in - 300
+
+                return self._access_token
+
+    def is_token_valid(self) -> bool:
+        """Проверить, действителен ли текущий токен."""
+        return self._access_token is not None and time.time() < self._expires_at
+
+    def clear_token(self):
+        """Очистить токен (для тестирования)."""
+        self._access_token = None
+        self._expires_at = 0
+