agent-lab-sdk 0.1.29__py3-none-any.whl → 0.1.31__py3-none-any.whl

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.dist-info → agent_lab_sdk-0.1.31.dist-info}/METADATA

@@ -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.dist-info → agent_lab_sdk-0.1.31.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=XyY0i_xb71Mt-zsZ6VSXltRK11Bi7CeWVR3mxbIRr4I,7272
+agent_lab_sdk/llm/throttled.py,sha256=wkZGw2lYjcg8yqfGxM4IEP2dcEf85bskRxINtrR5cn8,7530
 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.29.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
-agent_lab_sdk-0.1.29.dist-info/METADATA,sha256=uyc3EoyiOojFugnXayZuNI2TX6qppx1Te93pUGWEtrk,17884
-agent_lab_sdk-0.1.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agent_lab_sdk-0.1.29.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
-agent_lab_sdk-0.1.29.dist-info/RECORD,,
+agent_lab_sdk-0.1.31.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
+agent_lab_sdk-0.1.31.dist-info/METADATA,sha256=imuoRasBdDLTDL_UCeUQ-wF0z7lHct5ZD5eUy6ySi5c,17884
+agent_lab_sdk-0.1.31.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agent_lab_sdk-0.1.31.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
+agent_lab_sdk-0.1.31.dist-info/RECORD,,