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

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

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

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

@@ -49,8 +49,7 @@ class SearchPlugin:
     def __init__(self, base_url=None, api_key=None, query_domains=None):
         self.query_domains = query_domains or None
         self._api_key = api_key or self._get_setting_or_raise("SEARCH_PLUGIN_KEY")
-        self._retrieval_url = base_url or self._get_setting_or_raise("SEARCH_PLUGIN_URL")
-        self._rambler_proxy_url = base_url or self._get_setting_or_raise("SEARCH_IMAGES_PLUGIN_URL")
+        self._base_url = base_url
 
     def _get_setting_or_raise(self, name):
         value = getattr(settings, name, None)
@@ -122,11 +121,12 @@ class SearchPlugin:
 
         search_response = []
         retrieve_response = RetrieveResponse(documents=[], sources={}, images=[], mapped_documents=[])
+        url = self._base_url or self._get_setting_or_raise("SEARCH_PLUGIN_URL")
 
         try:
             with httpx.Client() as client:
                 response = client.post(
-                    url=self._retrieval_url,
+                    url=url,
                     headers=headers,
                     json=body,
                     timeout=settings.SEARCH_PLUGIN_RETRIEVAL_TIMEOUT
@@ -180,11 +180,12 @@ class SearchPlugin:
             "User-Agent": "agent-toolkit",
             **headers
         }
+        url = self._base_url or self._get_setting_or_raise("SEARCH_IMAGES_PLUGIN_URL")
 
         try:
             with httpx.Client(verify=True) as client:
                 response = client.get(
-                    url=self._rambler_proxy_url,
+                    url=url,
                     headers=headers,
                     params=params,
                     timeout=settings.SEARCH_PLUGIN_RAMBLER_PROXY_TIMEOUT

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

@@ -0,0 +1,4 @@
+from .search_image import get_search_image_tool
+from .search_plugin_web_search import get_search_plugin_tool
+
+__all__ = ["get_search_image_tool", "get_search_plugin_tool"]

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

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

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

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

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

@@ -1,4 +0,0 @@
-from .search_image import search_image
-from .search_plugin_web_search import search_plugin_web_search
-
-__all__ = ["search_image", "search_plugin_web_search"]