agent-lab-sdk 0.1.58.dev1__tar.gz → 0.1.58.dev3__tar.gz

{agent_lab_sdk-0.1.58.dev1 → agent_lab_sdk-0.1.58.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.58.dev1
+Version: 0.1.58.dev3
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved

{agent_lab_sdk-0.1.58.dev1 → agent_lab_sdk-0.1.58.dev3}/agent_lab_sdk/llm/agw_token_manager.py

@@ -10,6 +10,8 @@ import random
 import time
 from datetime import datetime, timedelta, timezone
 
+from agent_lab_sdk.core.settings import settings
+
 logger = logging.getLogger(__name__)
 
 # Путь к файлу токена (можно переопределить через ENV)
@@ -68,10 +70,12 @@ class AgwTokenManager:
                     raise
 
     @classmethod
-    def get_token(cls, provider: str, token_type: Optional[str] = None) -> str:
+    def get_token(cls, provider: str, token_type: Optional[str] = None) -> str | None:
         """
         Возвращает валидный токен, обновляя его при необходимости.
         """
+        if settings.GIGACHAT_USE_MTLS:
+            return None
         key = "{}_{}".format(provider, token_type if token_type is not None else "default")
         # Лок между потоками
         with cls._thread_lock:

{agent_lab_sdk-0.1.58.dev1 → agent_lab_sdk-0.1.58.dev3}/agent_lab_sdk/llm/gigachat_token_manager.py

@@ -195,12 +195,14 @@ class GigaChatTokenManager:
         use_token_provider_ags: Optional[bool] = None,
         model: Optional[str] = None,
         credential_id: Optional[str] = None,
-    ) -> str:
+    ) -> str | None:
         """
         Возвращает валидный токен, обновляя его при необходимости.
         Использует межпроцессную блокировку (fcntl.flock) и лок для потоков,
         чтобы избежать дублирующих запросов.
         """
+        if settings.GIGACHAT_USE_MTLS:
+            return None
         if use_token_provider_ags is None:
             use_token_provider_ags = settings.USE_TOKEN_PROVIDER_AGS
 

{agent_lab_sdk-0.1.58.dev1 → agent_lab_sdk-0.1.58.dev3}/agent_lab_sdk/llm/managers/ags_token_manager.py

@@ -50,7 +50,9 @@ class AgsTokenManager:
             return new_token
 
     @classmethod
-    def get_token_by_env(cls, model, credential_id):
+    def get_token_by_env(cls, model, credential_id) -> AgsTokenLimits | None:
+        if settings.GIGACHAT_USE_MTLS:
+            return None
         if settings.AGENT_SERVICE_NAME is None:
             raise ValueError(f"Переменная окружения {ENV_PREFIX}AGENT_SERVICE_NAME не установлена.")
         credential_id = credential_id or settings.GIGACHAT_CREDENTIAL_ID

{agent_lab_sdk-0.1.58.dev1 → agent_lab_sdk-0.1.58.dev3}/agent_lab_sdk/tools/search_plugin/tools/search_image.py

@@ -4,13 +4,19 @@ from agent_lab_sdk.tools.search_plugin.search_plugin import SearchPlugin
 
 def get_search_image_tool(headers=None, additional_params=None):
     @tool
-    def search_image(
+    def search_image_tool(
         text: str,
         image_limit: int
     ):
-        """Выполняет веб-поиск картинки через Search Plugin rambler_proxy_images"""
+        """Выполняет веб-поиск картинки через Search Plugin и возвращает картинку в markdownю.
+        text - текст запроса, image_limit - максимальное количество картинок"""
 
         search_plugin = SearchPlugin()
         images = search_plugin.rambler_images_proxy(text, headers=headers, additional_params=additional_params)
-        return list(filter(lambda image: image.url, images))[:image_limit]
-    return search_image
+
+        content = []
+        i = 1
+        for image in images[:image_limit]:
+            content.append(f"![Изображение {i}]({image.url})")
+        return "\n".join(content)
+    return search_image_tool

{agent_lab_sdk-0.1.58.dev1 → agent_lab_sdk-0.1.58.dev3}/agent_lab_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.58.dev1
+Version: 0.1.58.dev3
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved

{agent_lab_sdk-0.1.58.dev1 → agent_lab_sdk-0.1.58.dev3}/pyproject.toml

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