alita-sdk 0.3.127__py3-none-any.whl → 0.3.128__py3-none-any.whl

alita_sdk/langchain/langraph_agent.py

@@ -387,7 +387,7 @@ def create_graph(
             if toolkit_name:
                 tool_name = f"{clean_string(toolkit_name)}{TOOLKIT_SPLITTER}{tool_name}"
             logger.info(f"Node: {node_id} : {node_type} - {tool_name}")
-            if node_type in ['function', 'tool', 'loop', 'loop_from_tool', 'indexer', 'subgraph']:
+            if node_type in ['function', 'tool', 'loop', 'loop_from_tool', 'indexer', 'subgraph', 'pipeline', 'agent']:
                 for tool in tools:
                     if tool.name == tool_name:
                         if node_type == 'function':
@@ -397,14 +397,30 @@ def create_graph(
                                 input_mapping=node.get('input_mapping',
                                                        {'messages': {'type': 'variable', 'value': 'messages'}}),
                                 input_variables=node.get('input', ['messages'])))
-                        elif node_type == 'subgraph':
+                        elif node_type == 'agent':
+                            input_params = node.get('input', ['messages'])
+                            input_mapping = {'task': {'type': 'fstring', 'value': f"{node.get('task', '')}"}}
+                            # 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'}
+                            lg_builder.add_node(node_id, FunctionTool(
+                                client=client, tool=tool,
+                                name=node['id'], return_type='dict',
+                                output_variables=node.get('output', []),
+                                input_variables=input_params,
+                                input_mapping= input_mapping
+                            ))
+                        elif node_type == 'subgraph' or node_type == 'pipeline':
                             # assign parent memory/store
                             # tool.checkpointer = memory
                             # tool.store = store
                             # wrap with mappings
+                            pipeline_name = node.get('tool', None)
+                            if not pipeline_name:
+                                raise ValueError("Subgraph must have a 'tool' node: add required tool to the subgraph node")
                             node_fn = SubgraphRunnable(
                                 inner=tool,
-                                name=node['id'],
+                                name=pipeline_name,
                                 input_mapping=node.get('input_mapping', {}),
                                 output_mapping=node.get('output_mapping', {}),
                             )
@@ -419,6 +435,15 @@ def create_graph(
                                 structured_output=node.get('structured_output', False),
                                 task=node.get('task')
                             ))
+                        # TODO: decide on struct output for agent nodes
+                        # elif node_type == 'agent':
+                        #     lg_builder.add_node(node_id, AgentNode(
+                        #         client=client, tool=tool,
+                        #         name=node['id'], return_type='dict',
+                        #         output_variables=node.get('output', []),
+                        #         input_variables=node.get('input', ['messages']),
+                        #         task=node.get('task')
+                        #     ))
                         elif node_type == 'loop':
                             lg_builder.add_node(node_id, LoopNode(
                                 client=client, tool=tool,

alita_sdk/toolkits/tools.py

@@ -52,7 +52,7 @@ def get_tools(tools_list: list, alita_client, llm) -> list:
                 selected_tools=tool['settings']['selected_tools'],
                 toolkit_name=tool.get('toolkit_name', '') or tool.get('name', '')
             ).get_tools())
-        elif tool['type'] == 'application':
+        elif tool['type'] == 'application' and tool.get('agent_type', '') != 'pipeline' :
             tools.extend(ApplicationToolkit.get_toolkit(
                 alita_client,
                 application_id=int(tool['settings']['application_id']),
@@ -60,7 +60,7 @@ def get_tools(tools_list: list, alita_client, llm) -> list:
                 app_api_key=alita_client.auth_token,
                 selected_tools=[]
             ).get_tools())
-        elif tool['type'] == 'subgraph':
+        elif tool['type'] == 'application' and tool.get('agent_type', '') == 'pipeline':
             # static get_toolkit returns a list of CompiledStateGraph stubs
             tools.extend(SubgraphToolkit.get_toolkit(
                 alita_client,

alita_sdk/tools/agent.py

@@ -0,0 +1,74 @@
+import logging
+from json import dumps
+from traceback import format_exc
+from typing import Any, Optional, Union
+
+from langchain_core.callbacks import dispatch_custom_event
+from langchain_core.messages import ToolCall
+from langchain_core.runnables import RunnableConfig
+from langchain_core.tools import BaseTool
+from langchain_core.utils.function_calling import convert_to_openai_tool
+from pydantic import ValidationError
+
+logger = logging.getLogger(__name__)
+
+
+class AgentNode(BaseTool):
+    name: str = 'AgentNode'
+    description: str = 'This is agent node for tools'
+    client: Any = None
+    tool: BaseTool = None
+    return_type: str = "str"
+    input_variables: Optional[list[str]] = None
+    output_variables: Optional[list[str]] = None
+    structured_output: Optional[bool] = False
+    task: Optional[str] = None
+
+    def invoke(
+            self,
+            state: Union[str, dict, ToolCall],
+            config: Optional[RunnableConfig] = None,
+            **kwargs: Any,
+    ) -> Any:
+        params = convert_to_openai_tool(self.tool).get(
+            'function', {'parameters': {}}).get(
+            'parameters', {'properties': {}}).get('properties', {})
+        input_ = []
+        last_message = {}
+        logger.debug(f"AgentNode input: {self.input_variables}")
+        logger.debug(f"Output variables: {self.output_variables}")
+        for var in self.input_variables:
+            if var != 'messages':
+                last_message[var] = state[var]
+        if self.task:
+            task = self.task.format(**last_message, last_message=dumps(last_message))
+        else:
+            task = 'Input from user: {last_message}'.format(last_message=dumps(last_message))
+        try:
+            agent_input = {'task': task, 'chat_history': state.get('messages', [])[:] if 'messages' in self.input_variables else None}
+            tool_result = self.tool.invoke(agent_input, config=config, kwargs=kwargs)
+            dispatch_custom_event(
+                "on_tool_node", {
+                    "input_variables": self.input_variables,
+                    "tool_result": tool_result,
+                    "state": state,
+                }, config=config
+            )
+            message_result = tool_result
+            if isinstance(tool_result, dict) or isinstance(tool_result, list):
+                message_result = dumps(tool_result)
+            logger.info(f"AgentNode response: {tool_result}")
+            if not self.output_variables:
+                return {"messages": [{"role": "assistant", "content": message_result}]}
+            else:
+                return {self.output_variables[0]: tool_result,
+                        "messages": [{"role": "assistant", "content": message_result}]}
+        except ValidationError:
+            logger.error(f"ValidationError: {format_exc()}")
+            return {
+                "messages": [{"role": "assistant", "content": f"""Tool input to the {self.tool.name} with value {agent_input} raised ValidationError. 
+        \n\nTool schema is {dumps(params)} \n\nand the input to LLM was 
+        {input_[-1].content}"""}]}
+
+    def _run(self, *args, **kwargs):
+        return self.invoke(**kwargs)

{alita_sdk-0.3.127.dist-info → alita_sdk-0.3.128.dist-info}/METADATA

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

@@ -19,7 +19,7 @@ alita_sdk/langchain/assistant.py,sha256=J_xhwbNl934BgDKSpAMC9a1u6v03DZQcTYaamCzt
 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=PrD_9XEX7_LDOT_SuohW_nhqLMlzc14xmByHPrO0V6E,37951
+alita_sdk/langchain/langraph_agent.py,sha256=5TQQ1S2UhRY0PYCdr-W282LgPqLM2HA9xr8MOh185BY,39747
 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
@@ -69,9 +69,10 @@ alita_sdk/toolkits/artifact.py,sha256=7zb17vhJ3CigeTqvzQ4VNBsU5UOCJqAwz7fOJGMYqX
 alita_sdk/toolkits/datasource.py,sha256=v3FQu8Gmvq7gAGAnFEbA8qofyUhh98rxgIjY6GHBfyI,2494
 alita_sdk/toolkits/prompt.py,sha256=WIpTkkVYWqIqOWR_LlSWz3ug8uO9tm5jJ7aZYdiGRn0,1192
 alita_sdk/toolkits/subgraph.py,sha256=ZYqI4yVLbEPAjCR8dpXbjbL2ipX598Hk3fL6AgaqFD4,1758
-alita_sdk/toolkits/tools.py,sha256=gk3nvQBdab3QM8v93ff2nrN4ZfcT779yae2RygkTl8s,5834
+alita_sdk/toolkits/tools.py,sha256=eb4UFaSky0N42cTXHiJ9KojaReivcRNr2RM4UABQf8o,5928
 alita_sdk/toolkits/vectorstore.py,sha256=di08-CRl0KJ9xSZ8_24VVnPZy58iLqHtXW8vuF29P64,2893
 alita_sdk/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+alita_sdk/tools/agent.py,sha256=m98QxOHwnCRTT9j18Olbb5UPS8-ZGeQaGiUyZJSyFck,3162
 alita_sdk/tools/application.py,sha256=UJlYd3Sub10LpAoKkKEpvd4miWyrS-yYE5NKyqx-H4Q,2194
 alita_sdk/tools/artifact.py,sha256=uTa6K5d-NCDRnuLJVd6vA5TNIPH39onyPIyW5Thz4C0,6160
 alita_sdk/tools/datasource.py,sha256=pvbaSfI-ThQQnjHG-QhYNSTYRnZB0rYtZFpjCfpzxYI,2443
@@ -93,10 +94,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.127.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.128.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.127.dist-info/METADATA,sha256=Ox_VkvvGHqTNfe_wFqkVXU0etHQmW92EziEoEM5D158,7075
-alita_sdk-0.3.127.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.127.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
-alita_sdk-0.3.127.dist-info/RECORD,,
+alita_sdk-0.3.128.dist-info/METADATA,sha256=zWqqG-EN_cA9WGMHZvsejLGCpSTb1rrT2x2rdHugVoE,7075
+alita_sdk-0.3.128.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.128.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
+alita_sdk-0.3.128.dist-info/RECORD,,