alita-sdk 0.3.128__tar.gz → 0.3.130__tar.gz

{alita_sdk-0.3.128 → alita_sdk-0.3.130}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.128
-Summary: SDK for building langchain agents using resouces from Alita
+Version: 0.3.130
+Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedjik@gmail.com>
 Project-URL: Homepage, https://projectalita.ai
 Project-URL: Issues, https://github.com/ProjectAlita/alita-sdk/issues

{alita_sdk-0.3.128 → alita_sdk-0.3.130}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "alita_sdk"
-version = "0.3.128"
-description = "SDK for building langchain agents using resouces from Alita"
+version = "0.3.130"
+description = "SDK for building langchain agents using resources from Alita"
 readme = "README.md"
 requires-python = ">=3.10"
 classifiers = [ "Programming Language :: Python :: 3", "License :: OSI Approved :: Apache Software License", "Operating System :: OS Independent",]

{alita_sdk-0.3.128 → alita_sdk-0.3.130}/src/alita_sdk/clients/client.py

@@ -348,7 +348,7 @@ class AlitaClient:
             response_data = response.json()
             response_messages = []
             for message in response_data['messages']:
-                if message.get('type') == 'user':
+                if message.get('role') == 'user':
                     response_messages.append(HumanMessage(content=message['content']))
                 else:
                     response_messages.append(AIMessage(content=message['content']))

{alita_sdk-0.3.128 → alita_sdk-0.3.130}/src/alita_sdk/langchain/langraph_agent.py

@@ -399,7 +399,8 @@ def create_graph(
                                 input_variables=node.get('input', ['messages'])))
                         elif node_type == 'agent':
                             input_params = node.get('input', ['messages'])
-                            input_mapping = {'task': {'type': 'fstring', 'value': f"{node.get('task', '')}"}}
+                            input_mapping = {'task': {'type': 'fstring', 'value': f"{node.get('task', '')}"},
+                                                      'chat_history': {'type': 'fixed', 'value': []}}
                             # Add 'chat_history' to input_mapping only if 'messages' is in input_params
                             if 'messages' in input_params:
                                 input_mapping['chat_history'] = {'type': 'variable', 'value': 'messages'}

{alita_sdk-0.3.128 → alita_sdk-0.3.130}/src/alita_sdk/llms/alita.py

@@ -150,14 +150,33 @@ class AlitaChatModel(BaseChatModel):
                 logger.debug(f"message before getting to ChatGenerationChunk: {message}")
                 yield ChatGenerationChunk(message=message, generation_info=generation_info)
     
-    async def _astrem(
+    async def _astream(
         self,
         messages: List[BaseMessage],
         stop: Optional[List[str]] = None,
-        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        run_manager: Optional[AsyncCallbackManagerForLLMRun] = None,
         **kwargs: Any,
     ) -> AsyncIterator[ChatGenerationChunk]:
-        raise NotImplementedError("Streaming is not implemented")
+        iterator = await run_in_executor(
+            None,
+            self._stream,
+            messages,
+            stop,
+            run_manager.get_sync() if run_manager else None,
+            **kwargs,
+        )
+        done = object()
+        while True:
+            item: ChatGenerationChunk | object = await run_in_executor(
+                None,
+                next,
+                iterator,
+                done,
+            )
+            if item is done:
+                break
+            if isinstance(item, ChatGenerationChunk):
+                yield item
     
     def _create_chat_result(self, response: list[BaseMessage]) -> ChatResult:
         token_usage = 0

{alita_sdk-0.3.128 → alita_sdk-0.3.130}/src/alita_sdk/toolkits/tools.py

@@ -137,7 +137,7 @@ def _init_single_mcp_tool(toolkit_name, available_tool, alita, toolkit_settings)
             ),
             client=alita,
             server=toolkit_name,
-            tool_timeout_sec=toolkit_settings["timeout"]
+            tool_timeout_sec=toolkit_settings.get("timeout", 90)
         )
     except Exception as e:
         logger.error(f"Failed to create McpServerTool for '{toolkit_name}.{tool_name}': {e}")

{alita_sdk-0.3.128 → alita_sdk-0.3.130}/src/alita_sdk/tools/loop_output.py

@@ -39,8 +39,8 @@ Tool arguments schema:
 """
     unstructured_output: str = """Expected output is JSON that to be used as a KWARGS for the tool call like {{"key": "value"}} 
 in case your key is "chat_history" value should be a list of messages with roles like {{"chat_history": [{{"role": "user", "content": "input"}}, {{"role": "assistant", "content": "output"}}]}}.
-Tool won't have access to convesation so all keys and values need to be actual and independant. 
-Anwer must be JSON only extractable by JSON.LOADS."""
+Tool won't have access to conversation so all keys and values need to be actual and independent.
+Answer must be JSON only extractable by JSON.LOADS."""
 
     def invoke(
             self,

{alita_sdk-0.3.128 → alita_sdk-0.3.130}/src/alita_sdk/tools/tool.py

@@ -37,8 +37,8 @@ Tool arguments schema:
 """
     unstructured_output: str = """Expected output is JSON that to be used as a KWARGS for the tool call like {{"key": "value"}} 
 in case your key is "chat_history" value should be a list of messages with roles like {{"chat_history": [{{"role": "user", "content": "input"}}, {{"role": "assistant", "content": "output"}}]}}.
-Tool won't have access to convesation so all keys and values need to be actual and independant. 
-Anwer must be JSON only extractable by JSON.LOADS."""
+Tool won't have access to conversation so all keys and values need to be actual and independent.
+Answer must be JSON only extractable by JSON.LOADS."""
 
     def invoke(
             self,

{alita_sdk-0.3.128 → alita_sdk-0.3.130}/src/alita_sdk.egg-info/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.128
-Summary: SDK for building langchain agents using resouces from Alita
+Version: 0.3.130
+Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedjik@gmail.com>
 Project-URL: Homepage, https://projectalita.ai
 Project-URL: Issues, https://github.com/ProjectAlita/alita-sdk/issues