alita-sdk 0.3.121__tar.gz → 0.3.123__tar.gz

{alita_sdk-0.3.121 → alita_sdk-0.3.123}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.121
+Version: 0.3.123
 Summary: SDK for building langchain agents using resouces 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

{alita_sdk-0.3.121 → alita_sdk-0.3.123}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "alita_sdk"
-version = "0.3.121"
+version = "0.3.123"
 description = "SDK for building langchain agents using resouces from Alita"
 readme = "README.md"
 requires-python = ">=3.10"

{alita_sdk-0.3.121 → alita_sdk-0.3.123}/src/alita_sdk/langchain/document_loaders/AlitaConfluenceLoader.py

@@ -10,7 +10,7 @@ from pdf2image import convert_from_bytes
 from reportlab.graphics import renderPM
 from svglib.svglib import svg2rlg
 
-from src.alita_sdk.langchain.tools.utils import image_to_byte_array, bytes_to_base64
+from ..tools.utils import image_to_byte_array, bytes_to_base64
 
 Image.MAX_IMAGE_PIXELS = 300_000_000
 

{alita_sdk-0.3.121 → alita_sdk-0.3.123}/src/alita_sdk/langchain/interfaces/llm_processor.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import importlib
+from json import dumps
 from traceback import format_exc
 from langchain.chains.llm import LLMChain
 
@@ -33,6 +34,7 @@ from ..retrievers.AlitaRetriever import AlitaRetriever
 from ..tools.log import print_log
 
 
+
 def get_model(model_type: str, model_params: dict):
     """ Get LLM or ChatLLM """
     if model_type is None:
@@ -181,6 +183,11 @@ def add_documents(vectorstore, documents):
     metadata = []
     for document in documents:
         texts.append(document.page_content)
+        for key in document.metadata:
+            if isinstance(document.metadata[key], list):
+                document.metadata[key] = "; ".join(document.metadata[key])
+            if isinstance(document.metadata[key], dict):
+                document.metadata[key] = dumps(document.metadata[key])
         metadata.append(document.metadata)
     vectorstore.add_texts(texts, metadatas=metadata)
 

{alita_sdk-0.3.121 → alita_sdk-0.3.123}/src/alita_sdk/langchain/langraph_agent.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Union, Any, Optional, Annotated
+from typing import Union, Any, Optional, Annotated, get_type_hints
 from uuid import uuid4
 
 import yaml
@@ -329,11 +329,17 @@ def create_graph(
                                 structured_output=node.get('structured_output', False)))
                         break
             elif node_type == 'llm':
+                output_vars = node.get('output', [])
+                output_vars_dict = {
+                    var: get_type_hints(state_class).get(var, str).__name__
+                    for var in output_vars
+                }
                 lg_builder.add_node(node_id, LLMNode(
                     client=client, prompt=node.get('prompt', {}),
                     name=node['id'], return_type='dict',
                     response_key=node.get('response_key', 'messages'),
-                    output_variables=node.get('output', []),
+                    structured_output_dict=output_vars_dict,
+                    output_variables=output_vars,
                     input_variables=node.get('input', ['messages']),
                     structured_output=node.get('structured_output', False)))
             elif node_type == 'router':

{alita_sdk-0.3.121 → alita_sdk-0.3.123}/src/alita_sdk/tools/function.py

@@ -47,11 +47,10 @@ class FunctionTool(BaseTool):
             )
             logger.info(f"ToolNode response: {tool_result}")
             if not self.output_variables:
-                return {"messages": [{"role": "assistant", "content": tool_result}]}
+                return {"messages": [{"role": "assistant", "content": dumps(tool_result)}]}
             else:
                 return {
-                    self.output_variables[0]: tool_result,
-                    "messages": [{"role": "assistant", "content": tool_result}]
+                    self.output_variables[0]: tool_result
                 }
         except ValidationError:
             return {"messages": [

{alita_sdk-0.3.121 → alita_sdk-0.3.123}/src/alita_sdk/tools/llm.py

@@ -25,6 +25,7 @@ class LLMNode(BaseTool):
     client: Any = None
     return_type: str = "str"
     response_key: str = "messages"
+    structured_output_dict: Optional[dict[str, str]] = None
     output_variables: Optional[List[str]] = None
     input_variables: Optional[List[str]] = None
     structured_output: Optional[bool] = False
@@ -35,11 +36,19 @@ class LLMNode(BaseTool):
         llm_input = create_llm_input(self.prompt, params, kwargs)
         try:
             if self.structured_output and len(self.output_variables) > 0:
-                struct_params = {var: {"type": "str", "description": ""} for var in self.output_variables}
+                struct_params = {
+                    key: {
+                        "type": 'list[str]' if 'list' in value else value,
+                        "description": ""
+                    }
+                    for key, value in (self.structured_output_dict or {}).items()
+                }
                 stuct_model = create_pydantic_model(f"LLMOutput", struct_params)
                 llm = self.client.with_structured_output(stuct_model)
                 completion = llm.invoke(llm_input)
                 result = completion.model_dump()
+                if result.get('messages') and isinstance(result['messages'], list):
+                    result['messages'] = [{'role': 'assistant', 'content': '\n'.join(result['messages'])}]
                 return result
             else:
                 completion = self.client.invoke(llm_input)

{alita_sdk-0.3.121 → alita_sdk-0.3.123}/src/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.121
+Version: 0.3.123
 Summary: SDK for building langchain agents using resouces 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