agent-lab-sdk 0.1.16__tar.gz → 0.1.19__tar.gz

{agent_lab_sdk-0.1.16 → agent_lab_sdk-0.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.16
+Version: 0.1.19
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved
@@ -19,7 +19,7 @@ Requires-Python: <4.0,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: requests
-Requires-Dist: langgraph~=0.4.1
+Requires-Dist: langgraph
 Requires-Dist: langchain_gigachat
 Requires-Dist: prometheus-client
 Requires-Dist: langchain
@@ -58,7 +58,19 @@ from agent_lab_sdk.llm import get_model
 # Использует токен из окружения по умолчанию
 model = get_model()
 
-# Передача явных параметров
+# Получить модель GigaChat, использует токен из окружения по умолчанию
+model = get_model("chat")
+
+# Получить модель ThrottledGigaChat, использует токен из окружения по умолчанию
+model = get_model("chat", throttled=True)
+
+# Получить модель EmbeddingsGigaChat, использует токен из окружения по умолчанию
+model = get_model("embeddings")
+
+# Получить модель ThrottledEmbeddingsGigaChat, использует токен из окружения по умолчанию
+model = get_model("embeddings", throttled=True)
+
+# Передача явных параметров GigaChat, токен из окружения не использует
 model = get_model(
     access_token="YOUR_TOKEN",
     timeout=60,
@@ -68,6 +80,8 @@ model = get_model(
 
 > если не передавать access_token, токен будет выбран через GigaChatTokenManager
 
+> указание флага throttled=True включает ограничения и метрики для GigaChat и GigaChatEmbeddings. Подробнее [здесь](#2-модуль-agent_lab_sdkllmthrottled)
+
 ### 1.2. Менеджеры токенов
 
 | Класс                  | Описание                                                                                | Пример использования                            |
@@ -80,13 +94,15 @@ model = get_model(
 | Переменная                               | Описание                                                 | Значение по умолчанию / Пример                      |
 | ---------------------------------------- | -------------------------------------------------------- | --------------------------------------------------- |
 | `GIGACHAT_SCOPE`                         | Scope GigaChat API                               | `GIGACHAT_API_PERS`                                 |
-| `GLOBAL_GIGACHAT_TIMEOUT`                | Таймаут запросов к GigaChat (секунды)                    | `120`                                               |
-| `USE_TOKEN_PROVIDER_AGW`                 | Использовать `AgwTokenManager`                           | `true`                                              |
+| `GIGACHAT_TIMEOUT`                | Таймаут запросов к GigaChat (секунды)                    | `120`                                               |
+| `USE_TOKEN_PROVIDER_AGW`                 | Использовать `AgwTokenManager` для получения токена GigaChat                          | `true`                                              |
 | `GIGACHAT_CREDENTIALS`                   | Базовые креды для GigaChat (`b64(clientId:secretId)`)    | `Y2xpZW50SWQ6c2VjcmV0SWQ=`                          |
+| `GIGACHAT_USER`                   | Имя пользователя GigaChat advanced    | `user`                          |
+| `GIGACHAT_PASSWORD`                   | Пароль пользователя GigaChat advanced)    | `password`                          |
 | `GIGACHAT_TOKEN_PATH`                    | Путь к файлу кеша токена GigaChat                        | `/tmp/gigachat_token.json`                          |
 | `GIGACHAT_TOKEN_FETCH_RETRIES`           | Количество попыток получения токена (GigaChat)           | `3`                                                 |
 | `USE_GIGACHAT_ADVANCED`                  | Включает запрос токена GigaChat API в продвинутом режиме | `true`                                              |
-| `GIGACHAT_BASE_URL`                      | Базовый URL GigaChat для расширенного режима             | `https://ngw.devices.sberbank.ru:9443/api/v2/oauth` |
+| `GIGACHAT_BASE_URL`                      | Базовый URL GigaChat (важно чтобы заканчивался на символ `/`)                                    | `https://gigachat.sberdevices.ru/v1/` |
 | `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-0.1.16 → agent_lab_sdk-0.1.19}/README.md

@@ -29,7 +29,19 @@ from agent_lab_sdk.llm import get_model
 # Использует токен из окружения по умолчанию
 model = get_model()
 
-# Передача явных параметров
+# Получить модель GigaChat, использует токен из окружения по умолчанию
+model = get_model("chat")
+
+# Получить модель ThrottledGigaChat, использует токен из окружения по умолчанию
+model = get_model("chat", throttled=True)
+
+# Получить модель EmbeddingsGigaChat, использует токен из окружения по умолчанию
+model = get_model("embeddings")
+
+# Получить модель ThrottledEmbeddingsGigaChat, использует токен из окружения по умолчанию
+model = get_model("embeddings", throttled=True)
+
+# Передача явных параметров GigaChat, токен из окружения не использует
 model = get_model(
     access_token="YOUR_TOKEN",
     timeout=60,
@@ -39,6 +51,8 @@ model = get_model(
 
 > если не передавать access_token, токен будет выбран через GigaChatTokenManager
 
+> указание флага throttled=True включает ограничения и метрики для GigaChat и GigaChatEmbeddings. Подробнее [здесь](#2-модуль-agent_lab_sdkllmthrottled)
+
 ### 1.2. Менеджеры токенов
 
 | Класс                  | Описание                                                                                | Пример использования                            |
@@ -51,13 +65,15 @@ model = get_model(
 | Переменная                               | Описание                                                 | Значение по умолчанию / Пример                      |
 | ---------------------------------------- | -------------------------------------------------------- | --------------------------------------------------- |
 | `GIGACHAT_SCOPE`                         | Scope GigaChat API                               | `GIGACHAT_API_PERS`                                 |
-| `GLOBAL_GIGACHAT_TIMEOUT`                | Таймаут запросов к GigaChat (секунды)                    | `120`                                               |
-| `USE_TOKEN_PROVIDER_AGW`                 | Использовать `AgwTokenManager`                           | `true`                                              |
+| `GIGACHAT_TIMEOUT`                | Таймаут запросов к GigaChat (секунды)                    | `120`                                               |
+| `USE_TOKEN_PROVIDER_AGW`                 | Использовать `AgwTokenManager` для получения токена GigaChat                          | `true`                                              |
 | `GIGACHAT_CREDENTIALS`                   | Базовые креды для GigaChat (`b64(clientId:secretId)`)    | `Y2xpZW50SWQ6c2VjcmV0SWQ=`                          |
+| `GIGACHAT_USER`                   | Имя пользователя GigaChat advanced    | `user`                          |
+| `GIGACHAT_PASSWORD`                   | Пароль пользователя GigaChat advanced)    | `password`                          |
 | `GIGACHAT_TOKEN_PATH`                    | Путь к файлу кеша токена GigaChat                        | `/tmp/gigachat_token.json`                          |
 | `GIGACHAT_TOKEN_FETCH_RETRIES`           | Количество попыток получения токена (GigaChat)           | `3`                                                 |
 | `USE_GIGACHAT_ADVANCED`                  | Включает запрос токена GigaChat API в продвинутом режиме | `true`                                              |
-| `GIGACHAT_BASE_URL`                      | Базовый URL GigaChat для расширенного режима             | `https://ngw.devices.sberbank.ru:9443/api/v2/oauth` |
+| `GIGACHAT_BASE_URL`                      | Базовый URL GigaChat (важно чтобы заканчивался на символ `/`)                                    | `https://gigachat.sberdevices.ru/v1/` |
 | `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-0.1.16 → agent_lab_sdk-0.1.19}/agent_lab_sdk/llm/gigachat_token_manager.py

@@ -7,6 +7,7 @@ import requests
 import fcntl
 import random
 import time
+import base64
 import threading
 from datetime import datetime, timedelta, timezone
 from .agw_token_manager import AgwTokenManager
@@ -41,14 +42,17 @@ class GigaChatTokenManager:
             data = None
         else:
             gigachat_scope = os.environ.get("GIGACHAT_SCOPE", "GIGACHAT_API_PERS")
-
-            url = "https://ngw.devices.sberbank.ru:9443/api/v2/oauth"
+            url = os.getenv("GIGACHAT_AUTH_URL", "https://ngw.devices.sberbank.ru:9443/api/v2/oauth")
             data = {"scope": gigachat_scope}
 
-        print(url)
         gigachat_credentials = os.environ.get("GIGACHAT_CREDENTIALS")
         if not gigachat_credentials:
-            raise ValueError("Переменная окружения GIGACHAT_CREDENTIALS не установлена.")
+            user = os.getenv("GIGACHAT_USER")
+            password = os.getenv("GIGACHAT_PASSWORD")
+            if user and password:
+                gigachat_credentials = base64.b64encode(f"{user}:{password}")
+        if not gigachat_credentials:
+            raise ValueError("Переменная окружения GIGACHAT_CREDENTIALS (или GIGACHAT_USER в комбинации с GIGACHAT_PASSWORD) не установлена.")
         headers = {
             "Content-Type": "application/x-www-form-urlencoded",
             "Accept": "application/json",

agent_lab_sdk-0.1.19/agent_lab_sdk/llm/llm.py

@@ -0,0 +1,38 @@
+from langchain_gigachat.chat_models import GigaChat
+from langchain_gigachat.embeddings import GigaChatEmbeddings
+from agent_lab_sdk.llm.gigachat_token_manager import GigaChatTokenManager
+from agent_lab_sdk.llm.throttled import ThrottledGigaChat, ThrottledGigaChatEmbeddings
+from typing import Union
+import os
+
+def get_model(type="chat", throttled = False, **kwargs) -> Union[GigaChat, GigaChatEmbeddings, ThrottledGigaChat, ThrottledGigaChatEmbeddings]:
+    """
+    * type - определяет тип моледи : chat | embeddings
+    * throttled - оборачивает модель в класс обертку с дополнительным регулированием через семафор
+    * kwargs - прокидываются в модель
+    """
+    access_token = kwargs.pop("access_token", None)
+    if not access_token:
+        user = kwargs.get("user", None)
+        password = kwargs.get("password", None)
+        if not user and not password:
+            access_token = GigaChatTokenManager.get_token()
+
+    verify_ssl_certs = kwargs.pop("verify_ssl_certs", False)
+    verify_ssl_certs = os.getenv("GIGACHAT_VERIFY_SSL_CERTS", verify_ssl_certs)
+
+    if type == "chat":
+        _class = GigaChat
+        if throttled:
+            _class = ThrottledGigaChat
+    elif type == "embeddings":
+        _class = GigaChatEmbeddings
+        if throttled:
+            _class = ThrottledGigaChatEmbeddings
+    else:
+        raise ValueError(f"unsupported type {type}. possible values: chat, embeddings")
+    return _class(
+        access_token=access_token,
+        verify_ssl_certs=verify_ssl_certs,
+        **kwargs
+        )

{agent_lab_sdk-0.1.16 → agent_lab_sdk-0.1.19}/agent_lab_sdk/metrics/metrics.py

@@ -34,7 +34,7 @@ def get_metric(
     buckets: Optional[List[float]] = None,
     states: Optional[List[str]] = None,
     **kwargs
-) -> Any:
+) -> MetricType:
     """
     Возвращает метрику по name, создаёт её при первом запросе.
     

{agent_lab_sdk-0.1.16 → agent_lab_sdk-0.1.19}/agent_lab_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.16
+Version: 0.1.19
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved
@@ -19,7 +19,7 @@ Requires-Python: <4.0,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: requests
-Requires-Dist: langgraph~=0.4.1
+Requires-Dist: langgraph
 Requires-Dist: langchain_gigachat
 Requires-Dist: prometheus-client
 Requires-Dist: langchain
@@ -58,7 +58,19 @@ from agent_lab_sdk.llm import get_model
 # Использует токен из окружения по умолчанию
 model = get_model()
 
-# Передача явных параметров
+# Получить модель GigaChat, использует токен из окружения по умолчанию
+model = get_model("chat")
+
+# Получить модель ThrottledGigaChat, использует токен из окружения по умолчанию
+model = get_model("chat", throttled=True)
+
+# Получить модель EmbeddingsGigaChat, использует токен из окружения по умолчанию
+model = get_model("embeddings")
+
+# Получить модель ThrottledEmbeddingsGigaChat, использует токен из окружения по умолчанию
+model = get_model("embeddings", throttled=True)
+
+# Передача явных параметров GigaChat, токен из окружения не использует
 model = get_model(
     access_token="YOUR_TOKEN",
     timeout=60,
@@ -68,6 +80,8 @@ model = get_model(
 
 > если не передавать access_token, токен будет выбран через GigaChatTokenManager
 
+> указание флага throttled=True включает ограничения и метрики для GigaChat и GigaChatEmbeddings. Подробнее [здесь](#2-модуль-agent_lab_sdkllmthrottled)
+
 ### 1.2. Менеджеры токенов
 
 | Класс                  | Описание                                                                                | Пример использования                            |
@@ -80,13 +94,15 @@ model = get_model(
 | Переменная                               | Описание                                                 | Значение по умолчанию / Пример                      |
 | ---------------------------------------- | -------------------------------------------------------- | --------------------------------------------------- |
 | `GIGACHAT_SCOPE`                         | Scope GigaChat API                               | `GIGACHAT_API_PERS`                                 |
-| `GLOBAL_GIGACHAT_TIMEOUT`                | Таймаут запросов к GigaChat (секунды)                    | `120`                                               |
-| `USE_TOKEN_PROVIDER_AGW`                 | Использовать `AgwTokenManager`                           | `true`                                              |
+| `GIGACHAT_TIMEOUT`                | Таймаут запросов к GigaChat (секунды)                    | `120`                                               |
+| `USE_TOKEN_PROVIDER_AGW`                 | Использовать `AgwTokenManager` для получения токена GigaChat                          | `true`                                              |
 | `GIGACHAT_CREDENTIALS`                   | Базовые креды для GigaChat (`b64(clientId:secretId)`)    | `Y2xpZW50SWQ6c2VjcmV0SWQ=`                          |
+| `GIGACHAT_USER`                   | Имя пользователя GigaChat advanced    | `user`                          |
+| `GIGACHAT_PASSWORD`                   | Пароль пользователя GigaChat advanced)    | `password`                          |
 | `GIGACHAT_TOKEN_PATH`                    | Путь к файлу кеша токена GigaChat                        | `/tmp/gigachat_token.json`                          |
 | `GIGACHAT_TOKEN_FETCH_RETRIES`           | Количество попыток получения токена (GigaChat)           | `3`                                                 |
 | `USE_GIGACHAT_ADVANCED`                  | Включает запрос токена GigaChat API в продвинутом режиме | `true`                                              |
-| `GIGACHAT_BASE_URL`                      | Базовый URL GigaChat для расширенного режима             | `https://ngw.devices.sberbank.ru:9443/api/v2/oauth` |
+| `GIGACHAT_BASE_URL`                      | Базовый URL GigaChat (важно чтобы заканчивался на символ `/`)                                    | `https://gigachat.sberdevices.ru/v1/` |
 | `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-0.1.16 → agent_lab_sdk-0.1.19}/agent_lab_sdk.egg-info/requires.txt

@@ -1,5 +1,5 @@
 requests
-langgraph~=0.4.1
+langgraph
 langchain_gigachat
 prometheus-client
 langchain

{agent_lab_sdk-0.1.16 → agent_lab_sdk-0.1.19}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agent-lab-sdk"
-version = "0.1.16"
+version = "0.1.19"
 description = "SDK для работы с Agent Lab"
 readme = "README.md"
 license = { text = "Proprietary and Confidential — All Rights Reserved" }
@@ -13,7 +13,7 @@ authors = [
 ]
 requires-python = ">=3.11,<4.0"
 dependencies = [
-  "requests", "langgraph~=0.4.1", "langchain_gigachat", "prometheus-client", "langchain", "httpx~=0.27.2", "orjson>=3.9.7"
+  "requests", "langgraph", "langchain_gigachat", "prometheus-client", "langchain", "httpx~=0.27.2", "orjson>=3.9.7"
 ]
 keywords = ["agent", "lab", "sdk"]
 classifiers = [

agent_lab_sdk-0.1.16/agent_lab_sdk/llm/llm.py

@@ -1,28 +0,0 @@
-from langchain_gigachat.chat_models import GigaChat
-from agent_lab_sdk.llm.gigachat_token_manager import GigaChatTokenManager
-import os
-
-def get_model(**kwargs) -> GigaChat:
-    access_token = kwargs.pop("access_token", None)
-    if not access_token:
-        access_token = GigaChatTokenManager.get_token()
-    timeout = kwargs.pop("timeout", None)
-    if not timeout:
-        timeout=int(os.getenv("GLOBAL_GIGACHAT_TIMEOUT", "120"))
-        
-    scope = kwargs.pop("scope", None)
-    if not scope:
-        scope = os.getenv("GIGACHAT_SCOPE")
-
-    verify_ssl_certs = kwargs.pop("verify_ssl_certs", False)
-
-    if not scope:
-        raise ValueError("GIGACHAT_SCOPE environment variable is not set.")
-
-    return GigaChat(
-        access_token=access_token,
-        verify_ssl_certs=verify_ssl_certs,
-        scope=scope,
-        timeout=timeout,
-        **kwargs
-        )