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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.57.dev3
+Version: 0.1.58.dev1
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved
@@ -136,6 +136,7 @@ model = get_model(
 | `SDK_AGENT_LAB_GIGACHAT_CREDENTIAL_ID`   | Credential Id для получения токена из Agent Service                                             | -                                                  |
 | `AGENT_SERVICE_NAME`                     | Имя сервиса агента (обязательно)                                                                | -                                                  |
 | `SDK_AGENT_LAB_USE_TOKEN_PROVIDER_AGS`   | Использовать `AgsTokenManager` для получения токена и лимитов GigaChat                          | `false`                                            |
+| `SDK_AGENT_LAB_GIGACHAT_USE_MTLS`        | Не использовать токен менеджеры и глобальный лимитер или использовать                           | `false`                                            |
 
 
 ---
@@ -189,6 +190,9 @@ chat_with_retry = ThrottledGigaChat(
 | `embed_waiting_tasks`     | Число задач, ожидающих слота эмбеддингов       | Gauge     |
 | `embed_wait_time_seconds` | Время ожидания слота эмбеддингов (секунды)     | Histogram |
 
+Все метрики семафоров экспортируются с label `credential_id`.
+Если credential не задан (локальный режим), используется значение `credential_id="local"`.
+
 ---
 
 ## 3. Модуль `agent_lab_sdk.metrics`
@@ -518,11 +522,30 @@ pip install -e .
 Позволяет использовать API search plugin + llm tool + лимитирование запросов
 
 ```python
-from agent_lab_sdk.tools.search_plugin.search_plugin import SearchPlugin
+from agent_lab_sdk.tools.search_plugin.search_plugin import SearchPlugin, SearchMode
+from agent_lab_sdk.tools.search_plugin.tools import get_search_plugin_tool, get_search_image_tool
+from langgraph.prebuilt import ToolNode
+from langgraph.graph import StateGraph
+from agent_lab_sdk.llm import get_model
 
 plugin = SearchPlugin()
-result = plugin.retrieval("Что такое SDK?", "actual_info_web_search")
-print(result.documents, result.sources, result.images)
+retrieval_result = plugin.retrieval("Что такое SDK?", SearchMode.actual_info_web_search)
+print(retrieval_result.documents, retrieval_result.sources, retrieval_result.images)
+
+images_result = plugin.rambler_images_proxy("Логотип питона")
+for img in images_result:
+    print(img.url, img.image_link)
+
+search_tool = get_search_plugin_tool()
+image_tool = get_search_image_tool()
+tools = [search_tool, image_tool]
+
+llm = get_model().bind_tools(tools)
+
+graph = StateGraph(State)
+graph.add_node("tools", ToolNode(tools), metadata={'description': 'Вызов инструментов'})
+
+
 ```
 
 Реализованы методы POST /retrieval, GET /rambler_proxy_images

{agent_lab_sdk-0.1.57.dev3 → agent_lab_sdk-0.1.58.dev1}/README.md

@@ -101,6 +101,7 @@ model = get_model(
 | `SDK_AGENT_LAB_GIGACHAT_CREDENTIAL_ID`   | Credential Id для получения токена из Agent Service                                             | -                                                  |
 | `AGENT_SERVICE_NAME`                     | Имя сервиса агента (обязательно)                                                                | -                                                  |
 | `SDK_AGENT_LAB_USE_TOKEN_PROVIDER_AGS`   | Использовать `AgsTokenManager` для получения токена и лимитов GigaChat                          | `false`                                            |
+| `SDK_AGENT_LAB_GIGACHAT_USE_MTLS`        | Не использовать токен менеджеры и глобальный лимитер или использовать                           | `false`                                            |
 
 
 ---
@@ -154,6 +155,9 @@ chat_with_retry = ThrottledGigaChat(
 | `embed_waiting_tasks`     | Число задач, ожидающих слота эмбеддингов       | Gauge     |
 | `embed_wait_time_seconds` | Время ожидания слота эмбеддингов (секунды)     | Histogram |
 
+Все метрики семафоров экспортируются с label `credential_id`.
+Если credential не задан (локальный режим), используется значение `credential_id="local"`.
+
 ---
 
 ## 3. Модуль `agent_lab_sdk.metrics`
@@ -483,11 +487,30 @@ pip install -e .
 Позволяет использовать API search plugin + llm tool + лимитирование запросов
 
 ```python
-from agent_lab_sdk.tools.search_plugin.search_plugin import SearchPlugin
+from agent_lab_sdk.tools.search_plugin.search_plugin import SearchPlugin, SearchMode
+from agent_lab_sdk.tools.search_plugin.tools import get_search_plugin_tool, get_search_image_tool
+from langgraph.prebuilt import ToolNode
+from langgraph.graph import StateGraph
+from agent_lab_sdk.llm import get_model
 
 plugin = SearchPlugin()
-result = plugin.retrieval("Что такое SDK?", "actual_info_web_search")
-print(result.documents, result.sources, result.images)
+retrieval_result = plugin.retrieval("Что такое SDK?", SearchMode.actual_info_web_search)
+print(retrieval_result.documents, retrieval_result.sources, retrieval_result.images)
+
+images_result = plugin.rambler_images_proxy("Логотип питона")
+for img in images_result:
+    print(img.url, img.image_link)
+
+search_tool = get_search_plugin_tool()
+image_tool = get_search_image_tool()
+tools = [search_tool, image_tool]
+
+llm = get_model().bind_tools(tools)
+
+graph = StateGraph(State)
+graph.add_node("tools", ToolNode(tools), metadata={'description': 'Вызов инструментов'})
+
+
 ```
 
 Реализованы методы POST /retrieval, GET /rambler_proxy_images

{agent_lab_sdk-0.1.57.dev3 → agent_lab_sdk-0.1.58.dev1}/agent_lab_sdk/core/settings.py

@@ -34,11 +34,13 @@ class Settings(BaseSettings):
     SEARCH_PLUGIN_KEY: str | None = Field(alias="SEARCH_PLUGIN_KEY", default=None)
     SEARCH_PLUGIN_URL: str | None = Field(alias="SEARCH_PLUGIN_URL", default=None)
     SEARCH_IMAGES_PLUGIN_URL: str | None = Field(alias="SEARCH_IMAGES_PLUGIN_URL", default=None)
-    SEARCH_PLUGIN_RETRIEVAL_TIMEOUT: int = 30  # seconds
-    SEARCH_PLUGIN_RAMBLER_PROXY_TIMEOUT: int = 30  # seconds
+    SEARCH_PLUGIN_RETRIEVAL_TIMEOUT: float = 30  # seconds
+    SEARCH_PLUGIN_RAMBLER_PROXY_TIMEOUT: float = 30  # seconds
 
     SEARCH_PLUGIN_MAX_CONCURRENCY: int = 100000
 
+    GIGACHAT_USE_MTLS: bool = False
+
     class Config:
         env_prefix = ENV_PREFIX
 

{agent_lab_sdk-0.1.57.dev3 → agent_lab_sdk-0.1.58.dev1}/agent_lab_sdk/langgraph/checkpoint/agw_saver.py

@@ -491,6 +491,16 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
         # Всё остальное — лист, через _safe_load
         return self._safe_load(obj)
 
+    # ----------------------- per-request headers --------------------
+    def _per_request_headers(self, cfg: RunnableConfig | None) -> Dict[str, str]:
+        """Extract per-request headers from RunnableConfig (e.g. x-user-id)."""
+        if not cfg:
+            return {}
+        user_id = cfg.get("configurable", {}).get("user_id")
+        if user_id:
+            return {"x-user-id": str(user_id)}
+        return {}
+
     # ----------------------- config <-> api --------------------------
     def _to_api_config(self, cfg: RunnableConfig | None) -> Dict[str, Any]:
         if not cfg:
@@ -594,10 +604,14 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
         method: str,
         path: str,
         *,
+        headers: Dict[str, str] | None = None,
         ok_statuses: Iterable[int] | None = None,
         label_path: str | None = None,
         **kw,
     ) -> httpx.Response:
+        if headers:
+            kw["headers"] = {**kw.get("headers", {}), **headers}
+
         if "json" in kw:
             payload = kw.pop("json")
             kw["data"] = orjson.dumps(payload)
@@ -746,7 +760,12 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
             path = f"/checkpoint/{tid}"
             params = None
 
-        resp = await self._http("GET", path, params=params, label_path=path_template)
+        resp = await self._http(
+            "GET", path,
+            params=params,
+            headers=self._per_request_headers(cfg),
+            label_path=path_template,
+        )
         logger.debug("AGW aget_tuple response: %s", resp.text)
 
         if not resp.text:
@@ -774,6 +793,7 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
             "POST",
             "/checkpoint/list",
             json=payload,
+            headers=self._per_request_headers(cfg),
             label_path="/checkpoint/list",
         )
         logger.debug("AGW alist response: %s", resp.text)
@@ -794,7 +814,12 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
             "metadata": self._enc_meta(get_checkpoint_metadata(cfg, metadata)),
             "newVersions": self._safe_dump_deep(new_versions),
         }
-        resp = await self._http("POST", "/checkpoint", json=payload, label_path="/checkpoint")
+        resp = await self._http(
+            "POST", "/checkpoint",
+            json=payload,
+            headers=self._per_request_headers(cfg),
+            label_path="/checkpoint",
+        )
         logger.debug("AGW aput response: %s", resp.text)
         resp.raise_for_status()
         return resp.json()["config"]
@@ -817,6 +842,7 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
             "POST",
             "/checkpoint/writes",
             json=payload,
+            headers=self._per_request_headers(cfg),
             label_path="/checkpoint/writes",
         )
         logger.debug("AGW aput_writes response: %s", resp.text)

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

@@ -8,7 +8,10 @@ from agent_lab_sdk.core.unified_semaphore import UnifiedSemaphore
 from agent_lab_sdk.llm.managers.ags_token_manager import AgsTokenManager
 from agent_lab_sdk.metrics.create import create_metrics
 
-if settings.REDIS_THROTTLE_ENABLED:
+_redis_throttle_enabled = settings.REDIS_THROTTLE_ENABLED
+if settings.GIGACHAT_USE_MTLS:
+    _redis_throttle_enabled = False
+if _redis_throttle_enabled:
     from agent_lab_sdk.llm.managers.limits_manager import get_limits_manager
 
 logger = logging.getLogger(__name__)
@@ -28,10 +31,11 @@ class Limiting:
         self._internal_semaphore = internal_semaphore
         self._model_type = model_type
 
-        self._current = 0
-        self._in_use.set(0)
-        self._waiting.set(0)
-        if settings.REDIS_THROTTLE_ENABLED:
+        self._current_by_credential = {}
+        self._waiting_by_credential = {}
+        self._set_metric_value(self._in_use, "local", 0)
+        self._set_metric_value(self._waiting, "local", 0)
+        if _redis_throttle_enabled:
             self._limits_manager = get_limits_manager()
 
     def get_semaphore_keys(self, key_type, model, agent_id, credential_id):
@@ -39,29 +43,66 @@ class Limiting:
         total_key = "{}_{}".format(key_type, model)
         return key, total_key
 
-    def _dec_waiting_inc_current(self):
-        self._waiting.dec()
+    @staticmethod
+    def _normalize_credential_id(credential_id):
+        if credential_id is None:
+            return "local"
+        value = str(credential_id).strip()
+        return value or "local"
+
+    @staticmethod
+    def _metric_with_credential(metric, credential_id):
+        if metric is None:
+            return None
+        labelnames = getattr(metric, "_labelnames", ())
+        if "credential_id" in labelnames:
+            return metric.labels(credential_id=credential_id)
+        return metric
+
+    def _set_metric_value(self, metric, credential_id, value):
+        metric_ref = self._metric_with_credential(metric, credential_id)
+        if metric_ref is not None:
+            metric_ref.set(value)
+
+    def _observe_metric_value(self, metric, credential_id, value):
+        metric_ref = self._metric_with_credential(metric, credential_id)
+        if metric_ref is not None:
+            metric_ref.observe(value)
+
+    def _inc_waiting(self, credential_id):
         with self._lock:
-            self._current += 1
+            waiting = self._waiting_by_credential.get(credential_id, 0) + 1
+            self._waiting_by_credential[credential_id] = waiting
+        self._set_metric_value(self._waiting, credential_id, waiting)
 
-    def _wait_hist_in_use_update(self, start):
+    def _dec_waiting_inc_current(self, credential_id):
+        with self._lock:
+            waiting = max(0, self._waiting_by_credential.get(credential_id, 0) - 1)
+            self._waiting_by_credential[credential_id] = waiting
+            current = self._current_by_credential.get(credential_id, 0) + 1
+            self._current_by_credential[credential_id] = current
+        self._set_metric_value(self._waiting, credential_id, waiting)
+        self._set_metric_value(self._in_use, credential_id, current)
+
+    def _wait_hist_in_use_update(self, start, credential_id):
         elapsed = time.time() - start
-        self._wait_hist.observe(elapsed)
-        self._in_use.set(self._current)
+        self._observe_metric_value(self._wait_hist, credential_id, elapsed)
 
-    def _dec_current_in_use_update(self):
+    def _dec_current_in_use_update(self, credential_id):
         with self._lock:
-            self._current -= 1
-        self._in_use.set(self._current)
+            current = max(0, self._current_by_credential.get(credential_id, 0) - 1)
+            self._current_by_credential[credential_id] = current
+        self._set_metric_value(self._in_use, credential_id, current)
 
     @contextmanager
     def acquire(self, model, credential_id=None):
-        self._waiting.inc()
+        metric_credential_id = self._normalize_credential_id(credential_id)
+        self._inc_waiting(metric_credential_id)
         start = time.time()
 
         try:
             self._internal_semaphore.acquire()
-            if settings.REDIS_THROTTLE_ENABLED:
+            if _redis_throttle_enabled:
                 logger.info("redis throttle enabled")
                 with ExitStack() as stack:
                     try:
@@ -72,27 +113,28 @@ class Limiting:
                         self._limits_manager.with_rate_limit(ags_limits.total_limits, total_key)
                         self._limits_manager.with_rate_limit(ags_limits.limits, key)
                     finally:
-                        self._dec_waiting_inc_current()
-                    self._wait_hist_in_use_update(start)
+                        self._dec_waiting_inc_current(metric_credential_id)
+                    self._wait_hist_in_use_update(start, metric_credential_id)
                     yield
             else:
-                self._dec_waiting_inc_current()
-                self._wait_hist_in_use_update(start)
+                self._dec_waiting_inc_current(metric_credential_id)
+                self._wait_hist_in_use_update(start, metric_credential_id)
                 yield
         finally:
             try:
                 self._internal_semaphore.release()
             finally:
-                self._dec_current_in_use_update()
+                self._dec_current_in_use_update(metric_credential_id)
 
     @asynccontextmanager
     async def acquire_async(self, model, credential_id=None):
-        self._waiting.inc()
+        metric_credential_id = self._normalize_credential_id(credential_id)
+        self._inc_waiting(metric_credential_id)
         start = time.time()
 
         try:
             await self._internal_semaphore.acquire_async()
-            if settings.REDIS_THROTTLE_ENABLED:
+            if _redis_throttle_enabled:
                 logger.info("redis throttle enabled")
                 async with AsyncExitStack() as stack:
                     try:
@@ -103,15 +145,15 @@ class Limiting:
                         await self._limits_manager.with_rate_limit_async(ags_limits.total_limits, total_key)
                         await self._limits_manager.with_rate_limit_async(ags_limits.limits, key)
                     finally:
-                        self._dec_waiting_inc_current()
-                    self._wait_hist_in_use_update(start)
+                        self._dec_waiting_inc_current(metric_credential_id)
+                    self._wait_hist_in_use_update(start, metric_credential_id)
                     yield
             else:
-                self._dec_waiting_inc_current()
-                self._wait_hist_in_use_update(start)
+                self._dec_waiting_inc_current(metric_credential_id)
+                self._wait_hist_in_use_update(start, metric_credential_id)
                 yield
         finally:
             try:
                 await self._internal_semaphore.release_async()
             finally:
-                self._dec_current_in_use_update()
+                self._dec_current_in_use_update(metric_credential_id)

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

@@ -1,5 +1,6 @@
 from langchain_gigachat.chat_models import GigaChat
 from langchain_gigachat.embeddings import GigaChatEmbeddings
+from agent_lab_sdk.core.settings import settings
 from agent_lab_sdk.llm.gigachat_token_manager import GigaChatTokenManager
 from agent_lab_sdk.llm.token_manager_utils import build_token_manager_kwargs
 from agent_lab_sdk.llm.throttled import ThrottledGigaChat, ThrottledGigaChatEmbeddings
@@ -34,6 +35,9 @@ def get_model(
         kwargs,
         kwargs.pop("token_manager_kwargs", None),
     )
+    if settings.GIGACHAT_USE_MTLS:
+        access_token = "mtls"
+        manage_access_token = False
     if not access_token:
         user = kwargs.get("user", None)
         password = kwargs.get("password", None)

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

@@ -71,6 +71,10 @@ class ThrottledGigaChatEmbeddings(GigaChatEmbeddings):
         model = kwargs.get("model") or settings.GIGACHAT_MODEL
         use_token_provider_ags = token_manager_kwargs.get("use_ags_token") or settings.USE_TOKEN_PROVIDER_AGS
 
+        if settings.GIGACHAT_USE_MTLS:
+            manage_access_token = False
+            kwargs["access_token"] = "mtls"
+
         if manage_access_token and "access_token" not in kwargs:
             new_kwargs = _resolve_token_kwargs(
                 model, gigachat_credential_id, use_token_provider_ags, kwargs, token_manager_kwargs
@@ -163,6 +167,10 @@ class ThrottledGigaChat(GigaChat):
         model = kwargs.get("model") or settings.GIGACHAT_MODEL
         use_token_provider_ags = token_manager_kwargs.get("use_ags_token") or settings.USE_TOKEN_PROVIDER_AGS
 
+        if settings.GIGACHAT_USE_MTLS:
+            manage_access_token = False
+            kwargs["access_token"] = "mtls"
+
         if manage_access_token and "access_token" not in kwargs:
             new_kwargs = _resolve_token_kwargs(
                 model, gigachat_credential_id, use_token_provider_ags, kwargs, token_manager_kwargs

{agent_lab_sdk-0.1.57.dev3 → agent_lab_sdk-0.1.58.dev1}/agent_lab_sdk/metrics/create.py

@@ -4,15 +4,18 @@ from agent_lab_sdk.metrics import get_metric
 def create_metrics(prefix: str):
     in_use = get_metric(
         metric_type = "gauge", name = f"{prefix}_slots_in_use",
-        documentation = f"Number of {prefix} slots currently in use"
+        documentation = f"Number of {prefix} slots currently in use",
+        labelnames=["credential_id"],
     )
     waiting = get_metric(
         metric_type = "gauge", name = f"{prefix}_waiting_tasks",
-        documentation = f"Number of tasks waiting for {prefix}"
+        documentation = f"Number of tasks waiting for {prefix}",
+        labelnames=["credential_id"],
     )
     wait_time = get_metric(
         metric_type = "histogram", name = f"{prefix}_wait_time_seconds",
         documentation = f"Time tasks wait for {prefix}",
+        labelnames=["credential_id"],
         buckets = [3, 5, 10, 15, 30, 60, 120, 240, 480, 960, 1920, float("inf")]
     )
 

agent_lab_sdk-0.1.58.dev1/agent_lab_sdk/tools/search_plugin/__init__.py

@@ -0,0 +1,4 @@
+from .search_plugin import SearchPlugin, SearchMode, RetrieveResponse, ImagesProxyResult
+from . import tools
+
+__all__ = ["SearchPlugin", "SearchMode", "RetrieveResponse", "ImagesProxyResult", "tools"]

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

@@ -1,6 +1,8 @@
 import asyncio
 import logging
+import uuid
 from dataclasses import dataclass
+from enum import StrEnum
 
 import httpx
 
@@ -41,6 +43,12 @@ class ImagesProxyResult:
     file_size: int
     mime_type: str
 
+class SearchMode(StrEnum):
+    actual_info_web_search = "actual_info_web_search"
+    safe_search = "safe_search"
+    web_search = "web_search"
+    custom = "custom"
+
 
 class SearchPlugin:
     """
@@ -54,7 +62,7 @@ class SearchPlugin:
     def _get_setting_or_raise(self, name):
         value = getattr(settings, name, None)
         if not value:
-            raise ValueError(f"SearchPlugin setting not found. Please set the {ENV_PREFIX}{name} environment variable.")
+            raise ValueError(f"SearchPlugin setting not found. Please set the {name} environment variable.")
         return value
 
     async def retrieval_async(self, query, search_mode, doc_text_max_len=None, docs_count=None, headers=None, body_params=None) -> RetrieveResponse:
@@ -109,13 +117,17 @@ class SearchPlugin:
 
         body = {
             "query": search_query,
-            "search_mode": search_mode,
+            "search_mode": str(search_mode),
             **body_params
         }
 
         headers = {
             "Authorization": f"{self._api_key}",
             "User-Agent": "agent-toolkit",
+            "X-User-ID": str(uuid.uuid4()),
+            "X-Client-ID": str(uuid.uuid4()),
+            "X-Request-ID": str(uuid.uuid4()),
+            "X-Session-ID": str(uuid.uuid4()),
             **headers
         }
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.57.dev3
+Version: 0.1.58.dev1
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved
@@ -136,6 +136,7 @@ model = get_model(
 | `SDK_AGENT_LAB_GIGACHAT_CREDENTIAL_ID`   | Credential Id для получения токена из Agent Service                                             | -                                                  |
 | `AGENT_SERVICE_NAME`                     | Имя сервиса агента (обязательно)                                                                | -                                                  |
 | `SDK_AGENT_LAB_USE_TOKEN_PROVIDER_AGS`   | Использовать `AgsTokenManager` для получения токена и лимитов GigaChat                          | `false`                                            |
+| `SDK_AGENT_LAB_GIGACHAT_USE_MTLS`        | Не использовать токен менеджеры и глобальный лимитер или использовать                           | `false`                                            |
 
 
 ---
@@ -189,6 +190,9 @@ chat_with_retry = ThrottledGigaChat(
 | `embed_waiting_tasks`     | Число задач, ожидающих слота эмбеддингов       | Gauge     |
 | `embed_wait_time_seconds` | Время ожидания слота эмбеддингов (секунды)     | Histogram |
 
+Все метрики семафоров экспортируются с label `credential_id`.
+Если credential не задан (локальный режим), используется значение `credential_id="local"`.
+
 ---
 
 ## 3. Модуль `agent_lab_sdk.metrics`
@@ -518,11 +522,30 @@ pip install -e .
 Позволяет использовать API search plugin + llm tool + лимитирование запросов
 
 ```python
-from agent_lab_sdk.tools.search_plugin.search_plugin import SearchPlugin
+from agent_lab_sdk.tools.search_plugin.search_plugin import SearchPlugin, SearchMode
+from agent_lab_sdk.tools.search_plugin.tools import get_search_plugin_tool, get_search_image_tool
+from langgraph.prebuilt import ToolNode
+from langgraph.graph import StateGraph
+from agent_lab_sdk.llm import get_model
 
 plugin = SearchPlugin()
-result = plugin.retrieval("Что такое SDK?", "actual_info_web_search")
-print(result.documents, result.sources, result.images)
+retrieval_result = plugin.retrieval("Что такое SDK?", SearchMode.actual_info_web_search)
+print(retrieval_result.documents, retrieval_result.sources, retrieval_result.images)
+
+images_result = plugin.rambler_images_proxy("Логотип питона")
+for img in images_result:
+    print(img.url, img.image_link)
+
+search_tool = get_search_plugin_tool()
+image_tool = get_search_image_tool()
+tools = [search_tool, image_tool]
+
+llm = get_model().bind_tools(tools)
+
+graph = StateGraph(State)
+graph.add_node("tools", ToolNode(tools), metadata={'description': 'Вызов инструментов'})
+
+
 ```
 
 Реализованы методы POST /retrieval, GET /rambler_proxy_images

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

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

agent_lab_sdk-0.1.57.dev3/agent_lab_sdk/tools/search_plugin/__init__.py

@@ -1,4 +0,0 @@
-from .search_plugin import SearchPlugin
-from . import tools
-
-__all__ = ["SearchPlugin", "tools"]