agent-lab-sdk 0.1.29__tar.gz → 0.1.31__tar.gz

{agent_lab_sdk-0.1.29 → agent_lab_sdk-0.1.31}/PKG-INFO

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

{agent_lab_sdk-0.1.29 → agent_lab_sdk-0.1.31}/agent_lab_sdk/llm/throttled.py

@@ -113,16 +113,15 @@ class ThrottledGigaChatEmbeddings(GigaChatEmbeddings):
     _base_kwargs: dict = PrivateAttr(default={})
 
     def __init__(self, *args, manage_access_token=True, **kwargs):
-        print(self)
-        self._manage_access_token = manage_access_token
-        self._base_kwargs = dict(kwargs)
-
-        if self._manage_access_token and "access_token" not in kwargs:
+        if manage_access_token and "access_token" not in kwargs:
             token = GigaChatTokenManager.get_token()
             super().__init__(access_token=token, **kwargs)
         else:
             super().__init__(**kwargs)
 
+        self._manage_access_token = manage_access_token
+        self._base_kwargs = dict(kwargs)
+
     def _fresh(self) -> GigaChatEmbeddings:
         if self._manage_access_token:
             return GigaChatEmbeddings(access_token=GigaChatTokenManager.get_token(), **self._base_kwargs)
@@ -151,16 +150,15 @@ class ThrottledGigaChat(GigaChat):
     _base_kwargs: dict = PrivateAttr(default={})
 
     def __init__(self, *args, manage_access_token=True, **kwargs):
-        print(self)
-        self._manage_access_token = manage_access_token
-        self._base_kwargs = dict(kwargs)
-
-        if self._manage_access_token and "access_token" not in kwargs:
+        if manage_access_token and "access_token" not in kwargs:
             token = GigaChatTokenManager.get_token()
             super().__init__(access_token=token, **kwargs)
         else:
             super().__init__(**kwargs)
 
+        self._manage_access_token = manage_access_token
+        self._base_kwargs = dict(kwargs)
+
     def _fresh(self) -> GigaChat:
         if self._manage_access_token:
             return GigaChat(access_token=GigaChatTokenManager.get_token(), **self._base_kwargs)
@@ -197,6 +195,14 @@ class ThrottledGigaChat(GigaChat):
     async def abatch(self, *args, **kwargs):
         async with _semaphores["chat"]:
             return await self._fresh().abatch(*args, **kwargs)
+        
+    def upload_file(self, *args, **kwargs):
+        with _semaphores["chat"]:
+            return self._fresh().upload_file(*args, **kwargs)
+
+    async def aupload_file(self, *args, **kwargs):
+        async with _semaphores["chat"]:
+            return await self._fresh().aupload_file(*args, **kwargs)
 
     def batch_as_completed(self, *args, **kwargs):
         with _semaphores["chat"]:

{agent_lab_sdk-0.1.29 → agent_lab_sdk-0.1.31}/agent_lab_sdk.egg-info/PKG-INFO

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

{agent_lab_sdk-0.1.29 → agent_lab_sdk-0.1.31}/pyproject.toml

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