agent-lab-sdk 0.1.53.dev1__tar.gz → 0.1.53.dev2__tar.gz

{agent_lab_sdk-0.1.53.dev1 → agent_lab_sdk-0.1.53.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.53.dev1
+Version: 0.1.53.dev2
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved
@@ -131,9 +131,9 @@ model = get_model(
 | `TOKEN_PROVIDER_AGW_URL`                 | URL Agent Gateway для получения AGW-токена                                                      | `https://agent-gateway.apps.advosd.sberdevices.ru` |
 | `TOKEN_PROVIDER_AGW_DEFAULT_MAX_RETRIES` | Макс. попыток запроса токена (AGW)                                                              | `3`                                                |
 | `TOKEN_PROVIDER_AGW_TIMEOUT_SEC`         | Таймаут запроса к AGW (секунды)                                                                 | `5`                                                |
-| `AGENT_LAB_SDK_GIGACHAT_MODEL`           | Модель GigaChat                                                                                 | `GigaChat`                                         |
+| `GIGACHAT_MODEL`                         | Модель GigaChat                                                                                 | `GigaChat`                                         |
 | `AGENT_LAB_SDK_GIGACHAT_CREDENTIAL_ID`   | Credential Id для получения токена из Agent Service                                             | -                                                  |
-| `AGENT_LAB_SDK_AGENT_SERVICE_NAME`       | Имя сервиса агента (обязательно)                                                                | -                                                  |
+| `AGENT_SERVICE_NAME`                     | Имя сервиса агента (обязательно)                                                                | -                                                  |
 | `AGENT_LAB_SDK_USE_TOKEN_PROVIDER_AGS`   | Использовать `AgsTokenManager` для получения токена и лимитов GigaChat                          | `false`                                            |
 
 

{agent_lab_sdk-0.1.53.dev1 → agent_lab_sdk-0.1.53.dev2}/README.md

@@ -96,9 +96,9 @@ model = get_model(
 | `TOKEN_PROVIDER_AGW_URL`                 | URL Agent Gateway для получения AGW-токена                                                      | `https://agent-gateway.apps.advosd.sberdevices.ru` |
 | `TOKEN_PROVIDER_AGW_DEFAULT_MAX_RETRIES` | Макс. попыток запроса токена (AGW)                                                              | `3`                                                |
 | `TOKEN_PROVIDER_AGW_TIMEOUT_SEC`         | Таймаут запроса к AGW (секунды)                                                                 | `5`                                                |
-| `AGENT_LAB_SDK_GIGACHAT_MODEL`           | Модель GigaChat                                                                                 | `GigaChat`                                         |
+| `GIGACHAT_MODEL`                         | Модель GigaChat                                                                                 | `GigaChat`                                         |
 | `AGENT_LAB_SDK_GIGACHAT_CREDENTIAL_ID`   | Credential Id для получения токена из Agent Service                                             | -                                                  |
-| `AGENT_LAB_SDK_AGENT_SERVICE_NAME`       | Имя сервиса агента (обязательно)                                                                | -                                                  |
+| `AGENT_SERVICE_NAME`                     | Имя сервиса агента (обязательно)                                                                | -                                                  |
 | `AGENT_LAB_SDK_USE_TOKEN_PROVIDER_AGS`   | Использовать `AgsTokenManager` для получения токена и лимитов GigaChat                          | `false`                                            |
 
 

{agent_lab_sdk-0.1.53.dev1 → agent_lab_sdk-0.1.53.dev2}/agent_lab_sdk/core/settings.py

@@ -6,7 +6,7 @@ ENV_PREFIX = "SDK_AGENT_LAB_"
 
 class Settings(BaseSettings):
     USE_TOKEN_PROVIDER_AGS: bool = False
-    GIGACHAT_MODEL: str = "GigaChat"  # Взял из доки дефолт, мб надо поменять
+    GIGACHAT_MODEL: str | None = Field(alias="GIGACHAT_MODEL", default=None)
     GIGACHAT_CREDENTIAL_ID: str | None = None
     REDIS_THROTTLE_ENABLED: bool = False
     AGENT_SERVICE_NAME: str | None = Field(alias="AGENT_SERVICE_NAME", default=None)

{agent_lab_sdk-0.1.53.dev1 → agent_lab_sdk-0.1.53.dev2}/agent_lab_sdk/llm/clients/agent_service_client.py

@@ -41,7 +41,7 @@ class AgentServiceClient:
             client = self._get_client()
             url = urljoin(self.base_url, f"{self.version_prefix}/token-manager/token")
             body = {
-                "agentId": agent_id,
+                "agentServiceName": agent_id,
                 "credentialId": credential_id,
             }
             headers = {

{agent_lab_sdk-0.1.53.dev1 → agent_lab_sdk-0.1.53.dev2}/agent_lab_sdk/llm/gigachat_token_manager.py

@@ -14,6 +14,9 @@ from datetime import datetime, timedelta, timezone
 from .agw_token_manager import AgwTokenManager
 import urllib3
 
+from agent_lab_sdk.core.settings import settings
+from agent_lab_sdk.llm.managers.ags_token_manager import AgsTokenManager
+
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
@@ -189,12 +192,22 @@ class GigaChatTokenManager:
         use_gigachat_advanced: Optional[bool] = None,
         token_file_path: Optional[str] = None,
         use_token_provider_agw: Optional[bool] = None,
+        use_token_provider_ags: Optional[bool] = None,
+        model: Optional[str] = None,
+        credential_id: Optional[str] = None,
     ) -> str:
         """
         Возвращает валидный токен, обновляя его при необходимости.
         Использует межпроцессную блокировку (fcntl.flock) и лок для потоков,
         чтобы избежать дублирующих запросов.
         """
+        if use_token_provider_ags is None:
+            use_token_provider_ags = settings.USE_TOKEN_PROVIDER_AGS
+
+        if use_token_provider_ags:
+            token_limits = AgsTokenManager.get_token_by_env(model, credential_id)
+            return token_limits.token
+
         if use_token_provider_agw is None:
             use_token_provider_agw = USE_TOKEN_PROVIDER_AGW
 

{agent_lab_sdk-0.1.53.dev1 → agent_lab_sdk-0.1.53.dev2}/agent_lab_sdk/llm/managers/ags_token_manager.py

@@ -55,6 +55,8 @@ class AgsTokenManager:
             raise ValueError(f"Переменная окружения {ENV_PREFIX}AGENT_SERVICE_NAME не установлена.")
         if credential_id is None:
             raise ValueError(f"Переменная окружения {ENV_PREFIX}GIGACHAT_CREDENTIAL_ID не установлена.")
+        if model is None:
+            raise ValueError(f"Переменная окружения {ENV_PREFIX}GIGACHAT_MODEL не установлена.")
         log_key = f"{settings.AGENT_SERVICE_NAME}_{credential_id}"
         token_info = cls.get_token(settings.AGENT_SERVICE_NAME, credential_id)
         token = cls._get_value_or_raise(token_info, "token", log_key)

{agent_lab_sdk-0.1.53.dev1 → agent_lab_sdk-0.1.53.dev2}/agent_lab_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.53.dev1
+Version: 0.1.53.dev2
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved
@@ -131,9 +131,9 @@ model = get_model(
 | `TOKEN_PROVIDER_AGW_URL`                 | URL Agent Gateway для получения AGW-токена                                                      | `https://agent-gateway.apps.advosd.sberdevices.ru` |
 | `TOKEN_PROVIDER_AGW_DEFAULT_MAX_RETRIES` | Макс. попыток запроса токена (AGW)                                                              | `3`                                                |
 | `TOKEN_PROVIDER_AGW_TIMEOUT_SEC`         | Таймаут запроса к AGW (секунды)                                                                 | `5`                                                |
-| `AGENT_LAB_SDK_GIGACHAT_MODEL`           | Модель GigaChat                                                                                 | `GigaChat`                                         |
+| `GIGACHAT_MODEL`                         | Модель GigaChat                                                                                 | `GigaChat`                                         |
 | `AGENT_LAB_SDK_GIGACHAT_CREDENTIAL_ID`   | Credential Id для получения токена из Agent Service                                             | -                                                  |
-| `AGENT_LAB_SDK_AGENT_SERVICE_NAME`       | Имя сервиса агента (обязательно)                                                                | -                                                  |
+| `AGENT_SERVICE_NAME`                     | Имя сервиса агента (обязательно)                                                                | -                                                  |
 | `AGENT_LAB_SDK_USE_TOKEN_PROVIDER_AGS`   | Использовать `AgsTokenManager` для получения токена и лимитов GigaChat                          | `false`                                            |
 
 

{agent_lab_sdk-0.1.53.dev1 → agent_lab_sdk-0.1.53.dev2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agent-lab-sdk"
-version = "0.1.53.dev1"
+version = "0.1.53.dev2"
 description = "SDK для работы с Agent Lab"
 readme = "README.md"
 license = { text = "Proprietary and Confidential — All Rights Reserved" }