alita-sdk 0.3.122__py3-none-any.whl → 0.3.123__py3-none-any.whl

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/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/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/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.122.dist-info → alita_sdk-0.3.123.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.122
+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.122.dist-info → alita_sdk-0.3.123.dist-info}/RECORD

@@ -19,7 +19,7 @@ alita_sdk/langchain/assistant.py,sha256=V0MpZG9IQRlKMr1SfAabVbUsIp-gNPO9A1SWxuFi
 alita_sdk/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/langchain/constants.py,sha256=eHVJ_beJNTf1WJo4yq7KMK64fxsRvs3lKc34QCXSbpk,3319
 alita_sdk/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
-alita_sdk/langchain/langraph_agent.py,sha256=vxsbjOE0UWZIuyXNGEMR1QJ8t5yHS7Kzjt8a8nLb-jw,20626
+alita_sdk/langchain/langraph_agent.py,sha256=f9rGk6QGQbfbGudwpM5ax9yS-xDlElFqGAOsZbGvrtI,20919
 alita_sdk/langchain/mixedAgentParser.py,sha256=M256lvtsL3YtYflBCEp-rWKrKtcY1dJIyRGVv7KW9ME,2611
 alita_sdk/langchain/mixedAgentRenderes.py,sha256=asBtKqm88QhZRILditjYICwFVKF5KfO38hu2O-WrSWE,5964
 alita_sdk/langchain/utils.py,sha256=Npferkn10dvdksnKzLJLBI5bNGQyVWTBwqp3vQtUqmY,6631
@@ -41,7 +41,7 @@ alita_sdk/langchain/document_loaders/constants.py,sha256=QpgMiKo-riIBuU67DGhXi7Z
 alita_sdk/langchain/document_loaders/utils.py,sha256=ifh0UJiweIb2iEb2_THJy2pbAnTupOffGtTdRA7vjeE,874
 alita_sdk/langchain/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/langchain/interfaces/kwextractor.py,sha256=kSJA9L8g8UArmHu7Bd9dIO0Rrq86JPUb8RYNlnN68FQ,3072
-alita_sdk/langchain/interfaces/llm_processor.py,sha256=oX5feXsQ5D-8q3Mf40Txltv3kkHKb2xBeNPwecy5yDQ,8442
+alita_sdk/langchain/interfaces/llm_processor.py,sha256=AbuWlZ54Cu5zy6AExRN0ou3r15i4pY3WSIVPJx8ltkA,8764
 alita_sdk/langchain/interfaces/loaders.py,sha256=li-O2dubiDNYn-qfVcDsuD4LqP_IZ61cV2vHUZAqeXc,3337
 alita_sdk/langchain/interfaces/splitters.py,sha256=tW65-Ejj9VYyxXFZNgPts_CKILQ18bWp_1bZ-24FKGc,3630
 alita_sdk/langchain/retrievers/AlitaRetriever.py,sha256=osChtJxUlfpsFESpJSE5mnJAkxTXnzgFZnC6l5mUlbo,6148
@@ -75,9 +75,9 @@ alita_sdk/tools/application.py,sha256=UJlYd3Sub10LpAoKkKEpvd4miWyrS-yYE5NKyqx-H4
 alita_sdk/tools/artifact.py,sha256=uTa6K5d-NCDRnuLJVd6vA5TNIPH39onyPIyW5Thz4C0,6160
 alita_sdk/tools/datasource.py,sha256=pvbaSfI-ThQQnjHG-QhYNSTYRnZB0rYtZFpjCfpzxYI,2443
 alita_sdk/tools/echo.py,sha256=spw9eCweXzixJqHnZofHE1yWiSUa04L4VKycf3KCEaM,486
-alita_sdk/tools/function.py,sha256=g5VaLtgYHoEMgid1s1zU2NzavqT2IErjhSIWqqPw3Yk,2617
+alita_sdk/tools/function.py,sha256=wzPS5Y8ScmsXspmn73MQnkCcAghsllAg7BLh61ZNuVw,2543
 alita_sdk/tools/indexer_tool.py,sha256=P9S_omk5TZkizb6UXyxMO87Pzj4UCaye0CuXBgCnhTU,4258
-alita_sdk/tools/llm.py,sha256=oiaecQNvFLqokmImyfbBKQ_bNMD8hrD8lhTFpiUytdA,2779
+alita_sdk/tools/llm.py,sha256=JA0OnSU13CLdkS5NFv6iRk8P7k-B47L-oPjk8xrzk48,3223
 alita_sdk/tools/loop.py,sha256=uds0WhZvwMxDVFI6MZHrcmMle637cQfBNg682iLxoJA,8335
 alita_sdk/tools/loop_output.py,sha256=NoGIGYc42wY3NNcWRijYzRnUVXcCn5cRVd8QmuIpoHU,8068
 alita_sdk/tools/pgvector_search.py,sha256=NN2BGAnq4SsDHIhUcFZ8d_dbEOM8QwB0UwpsWCYruXU,11692
@@ -91,10 +91,10 @@ alita_sdk/utils/evaluate.py,sha256=iM1P8gzBLHTuSCe85_Ng_h30m52hFuGuhNXJ7kB1tgI,1
 alita_sdk/utils/logging.py,sha256=hBE3qAzmcLMdamMp2YRXwOOK9P4lmNaNhM76kntVljs,3124
 alita_sdk/utils/streamlit.py,sha256=zp8owZwHI3HZplhcExJf6R3-APtWx-z6s5jznT2hY_k,29124
 alita_sdk/utils/utils.py,sha256=dM8whOJAuFJFe19qJ69-FLzrUp6d2G-G6L7d4ss2XqM,346
-alita_sdk-0.3.122.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.123.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_jira_analysis.py,sha256=I0cErH5R_dHVyutpXrM1QEo7jfBuKWTmDQvJBPjx18I,3281
-alita_sdk-0.3.122.dist-info/METADATA,sha256=_tp0Oa-uXYLh9RmhhlYGM4dazECkrNy0vtUUFMce_Cg,7075
-alita_sdk-0.3.122.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-alita_sdk-0.3.122.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
-alita_sdk-0.3.122.dist-info/RECORD,,
+alita_sdk-0.3.123.dist-info/METADATA,sha256=UBv__Ag0bAjUYgjkYIE3pjx2tzIWlSfHZSc-XxiaFkg,7075
+alita_sdk-0.3.123.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+alita_sdk-0.3.123.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
+alita_sdk-0.3.123.dist-info/RECORD,,