agent-lab-sdk 0.1.34__tar.gz → 0.1.35__tar.gz

{agent_lab_sdk-0.1.34 → agent_lab_sdk-0.1.35}/PKG-INFO

@@ -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 → agent_lab_sdk-0.1.35}/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 → agent_lab_sdk-0.1.35}/agent_lab_sdk.egg-info/PKG-INFO

@@ -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 → agent_lab_sdk-0.1.35}/pyproject.toml

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