agent-lab-sdk 0.1.34__py3-none-any.whl → 0.1.35__py3-none-any.whl

agent_lab_sdk/llm/throttled.py

@@ -110,6 +110,7 @@ _semaphores = {
 
 class ThrottledGigaChatEmbeddings(GigaChatEmbeddings):
     _manage_access_token: bool = PrivateAttr(default=True)
+    _base_kwargs: dict = PrivateAttr(default={})
 
     def __init__(self, *args, manage_access_token=True, **kwargs):
         if manage_access_token and "access_token" not in kwargs:
@@ -119,11 +120,13 @@ class ThrottledGigaChatEmbeddings(GigaChatEmbeddings):
             super().__init__(**kwargs)
 
         self._manage_access_token = manage_access_token
+        self._base_kwargs = dict(kwargs)
 
     def _fresh(self) -> GigaChatEmbeddings:
         if self._manage_access_token:
-            self.access_token = GigaChatTokenManager.get_token()
-            return self
+            if "access_token" in self._base_kwargs:
+                self._base_kwargs.pop("access_token")
+            return GigaChatEmbeddings(access_token=GigaChatTokenManager.get_token(), **self._base_kwargs)
         else:    
             return self
 
@@ -146,6 +149,7 @@ class ThrottledGigaChatEmbeddings(GigaChatEmbeddings):
 
 class ThrottledGigaChat(GigaChat):
     _manage_access_token: bool = PrivateAttr(default=True)
+    _base_kwargs: dict = PrivateAttr(default={})
 
     def __init__(self, *args, manage_access_token=True, **kwargs):
         if manage_access_token and "access_token" not in kwargs:
@@ -155,11 +159,16 @@ class ThrottledGigaChat(GigaChat):
             super().__init__(**kwargs)
 
         self._manage_access_token = manage_access_token
+        self._base_kwargs = dict(kwargs)
 
     def _fresh(self) -> GigaChat:
         if self._manage_access_token:
-            self.access_token = GigaChatTokenManager.get_token()
-            return self
+            if "access_token" in self._base_kwargs:
+                self._base_kwargs.pop("access_token")
+            new_gigachat = GigaChat(access_token=GigaChatTokenManager.get_token(), **self._base_kwargs)
+            # это поле которое управляется классом BaseChatModel, по другому его пока никак не прокинуть
+            new_gigachat.disable_streaming = self.disable_streaming
+            return new_gigachat
         else:    
             return self
 

{agent_lab_sdk-0.1.34.dist-info → agent_lab_sdk-0.1.35.dist-info}/METADATA

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

{agent_lab_sdk-0.1.34.dist-info → agent_lab_sdk-0.1.35.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=nNYKoGgat_0DReEtgH4W3pvTQ8yTOkbIooLap75K9Lg,7326
+agent_lab_sdk/llm/throttled.py,sha256=faccDXiKkmFFfZkVrxFQDHxHutAAivsKDEpYlygfuqU,8019
 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=e75nRW7Dz_RHk5Yia8DkFfbqMafsLQsQrJPfz
 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.34.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
-agent_lab_sdk-0.1.34.dist-info/METADATA,sha256=TLOQ-pmBjPtgav3fhYZnK354s8jUDnnltStkZgGKmsA,17884
-agent_lab_sdk-0.1.34.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agent_lab_sdk-0.1.34.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
-agent_lab_sdk-0.1.34.dist-info/RECORD,,
+agent_lab_sdk-0.1.35.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
+agent_lab_sdk-0.1.35.dist-info/METADATA,sha256=U8K4oMgtKBD5uMVi3RhDRkkdSUIjZF1y3GcLgvCzN-g,17884
+agent_lab_sdk-0.1.35.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agent_lab_sdk-0.1.35.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
+agent_lab_sdk-0.1.35.dist-info/RECORD,,