agent-lab-sdk 0.1.28__py3-none-any.whl → 0.1.30__py3-none-any.whl

agent_lab_sdk/llm/throttled.py

@@ -110,7 +110,7 @@ _semaphores = {
 
 class ThrottledGigaChatEmbeddings(GigaChatEmbeddings):
     _manage_access_token: bool = PrivateAttr(default=True)
-    _base_kwargs: bool = PrivateAttr(default=True)
+    _base_kwargs: dict = PrivateAttr(default={})
 
     def __init__(self, *args, manage_access_token=True, **kwargs):
         print(self)
@@ -125,7 +125,7 @@ class ThrottledGigaChatEmbeddings(GigaChatEmbeddings):
 
     def _fresh(self) -> GigaChatEmbeddings:
         if self._manage_access_token:
-            return GigaChatEmbeddings(access_token=GigaChatTokenManager, **self._base_kwargs)
+            return GigaChatEmbeddings(access_token=GigaChatTokenManager.get_token(), **self._base_kwargs)
         else:    
             return self
 
@@ -148,7 +148,7 @@ class ThrottledGigaChatEmbeddings(GigaChatEmbeddings):
 
 class ThrottledGigaChat(GigaChat):
     _manage_access_token: bool = PrivateAttr(default=True)
-    _base_kwargs: bool = PrivateAttr(default=True)
+    _base_kwargs: dict = PrivateAttr(default={})
 
     def __init__(self, *args, manage_access_token=True, **kwargs):
         print(self)
@@ -163,7 +163,7 @@ class ThrottledGigaChat(GigaChat):
 
     def _fresh(self) -> GigaChat:
         if self._manage_access_token:
-            return GigaChat(access_token=GigaChatTokenManager, **self._base_kwargs)
+            return GigaChat(access_token=GigaChatTokenManager.get_token(), **self._base_kwargs)
         else:    
             return self
 
@@ -197,6 +197,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.28.dist-info → agent_lab_sdk-0.1.30.dist-info}/METADATA

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

{agent_lab_sdk-0.1.28.dist-info → agent_lab_sdk-0.1.30.dist-info}/RECORD

@@ -5,7 +5,7 @@ agent_lab_sdk/llm/__init__.py,sha256=Yo9MbYdHS1iX05A9XiJGwWN1Hm4IARGav9mNFPrtDeA
 agent_lab_sdk/llm/agw_token_manager.py,sha256=_bPPI8muaEa6H01P8hHQOJHiiivaLd8N_d3OT9UT_80,4787
 agent_lab_sdk/llm/gigachat_token_manager.py,sha256=nlOxHcwJovsmM4cpI4fwMrYjoSeMjelDaHTipXsrUuA,8282
 agent_lab_sdk/llm/llm.py,sha256=HIp97-rs8Xcxokgptgh6bCfkIx7UJUFOYdGbpi7SsSg,1949
-agent_lab_sdk/llm/throttled.py,sha256=QTiwFKPSC_UOJV4Zlba_7R-RMDkL0W3mUTTI24CZ9eY,7252
+agent_lab_sdk/llm/throttled.py,sha256=RkDbtCzemaQnQdq3S2obkJ8nSb1ubmcLeZk-0s5xzgc,7582
 agent_lab_sdk/metrics/__init__.py,sha256=G4VSlzKwupPMM4c6vZaF1rnd0KusKarezDMjli9pVFw,57
 agent_lab_sdk/metrics/metrics.py,sha256=_XTT9vMG7T0u_D2pL371wm8GoBU5fodJ45D2RACnBJw,3439
 agent_lab_sdk/schema/__init__.py,sha256=bHSyXQYkcB9fWBlziWodXR_IzC5nKrdKzrCpyVWNY9o,521
@@ -13,8 +13,8 @@ agent_lab_sdk/schema/input_types.py,sha256=A8yRSFdCo-BSZM50sj-oXge8HdcVz1AkqbBWg
 agent_lab_sdk/schema/log_message.py,sha256=nadi6lZGRuDSPmfbYs9QPpRJUT9Pfy8Y7pGCvyFF5Mw,638
 agent_lab_sdk/storage/__init__.py,sha256=ik1_v1DMTwehvcAEXIYxuvLuCjJCa3y5qAuJqoQpuSA,81
 agent_lab_sdk/storage/storage.py,sha256=ELpt7GRwFD-aWa6ctinfA_QwcvzWLvKS0Wz8FlxVqAs,2075
-agent_lab_sdk-0.1.28.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
-agent_lab_sdk-0.1.28.dist-info/METADATA,sha256=0O503Vk6Jk940UyR0lYqRdM7hEzOaBWdRsPvhsWHdsk,17884
-agent_lab_sdk-0.1.28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agent_lab_sdk-0.1.28.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
-agent_lab_sdk-0.1.28.dist-info/RECORD,,
+agent_lab_sdk-0.1.30.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
+agent_lab_sdk-0.1.30.dist-info/METADATA,sha256=8wnIgMKSD-rLDmCdS-hApr7BwFTd8mkDqOT6XJsrozg,17884
+agent_lab_sdk-0.1.30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agent_lab_sdk-0.1.30.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
+agent_lab_sdk-0.1.30.dist-info/RECORD,,