alita-sdk 0.3.126__tar.gz → 0.3.127__tar.gz

{alita_sdk-0.3.126 → alita_sdk-0.3.127}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.126
+Version: 0.3.127
 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.126 → alita_sdk-0.3.127}/pyproject.toml

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

{alita_sdk-0.3.126 → alita_sdk-0.3.127}/src/alita_sdk/langchain/langraph_agent.py

@@ -4,6 +4,7 @@ from uuid import uuid4
 from typing import Dict
 
 import yaml
+import ast
 from langchain_core.callbacks import dispatch_custom_event
 from langchain_core.messages import HumanMessage
 from langchain_core.runnables import Runnable
@@ -211,6 +212,25 @@ class TransitionalEdge(Runnable):
         )
         return self.next_step if self.next_step != 'END' else END
 
+class StateDefaultNode(Runnable):
+    name = "StateDefaultNode"
+
+    def __init__(self, default_vars: dict = {}):
+        self.default_vars = default_vars
+
+    def invoke(self, state: BaseStore, config: Optional[RunnableConfig] = None) -> dict:
+        logger.info("Setting default state variables")
+        result = {}
+        for key, value in self.default_vars.items():
+            if isinstance(value, dict) and 'value' in value:
+                temp_value = value['value']
+                try:
+                    result[key] = ast.literal_eval(temp_value)
+                except:
+                    logger.debug("Unable to evaluate value, using as is")
+                    result[key] = temp_value
+        return result
+
 
 class StateModifierNode(Runnable):
     name = "StateModifierNode"
@@ -353,7 +373,8 @@ def create_graph(
     logger.debug(f"Schema: {schema}")
     logger.debug(f"Tools: {tools}")
     logger.info(f"Tools: {[tool.name for tool in tools]}")
-    state_class = create_state(schema.get('state', {}))
+    state = schema.get('state', {})
+    state_class = create_state(state)
     lg_builder = StateGraph(state_class)
     interrupt_before = [clean_string(every) for every in schema.get('interrupt_before', [])]
     interrupt_after = [clean_string(every) for every in schema.get('interrupt_after', [])]
@@ -503,9 +524,20 @@ def create_graph(
                     conditional_outputs=node['condition'].get('conditional_outputs', []),
                     default_output=node['condition'].get('default_output', 'END')))
 
-        lg_builder.set_entry_point(clean_string(schema['entry_point']))
+        # set default value for state variable at START
+        entry_point = clean_string(schema['entry_point'])
+        for key, value in state.items():
+            if 'type' in value and 'value' in value:
+                # set default value for state variable if it is defined in the schema
+                state_default_node = StateDefaultNode(default_vars=state)
+                lg_builder.add_node(state_default_node.name, state_default_node)
+                lg_builder.set_entry_point(state_default_node.name)
+                lg_builder.add_conditional_edges(state_default_node.name, TransitionalEdge(entry_point))
+                break
+        else:
+            # if no state variables are defined, set the entry point directly
+            lg_builder.set_entry_point(entry_point)
 
-        # assign default values
         interrupt_before = interrupt_before or []
         interrupt_after = interrupt_after or []
 

{alita_sdk-0.3.126 → alita_sdk-0.3.127}/src/alita_sdk/tools/tool.py

@@ -10,6 +10,7 @@ from langchain_core.tools import BaseTool
 from langchain_core.utils.function_calling import convert_to_openai_tool
 from pydantic import ValidationError, BaseModel, create_model
 
+from .application import Application
 from ..langchain.utils import _extract_json
 
 logger = logging.getLogger(__name__)
@@ -74,8 +75,9 @@ Anwer must be JSON only extractable by JSON.LOADS."""
             ))
         ]
         if self.structured_output:
-            # cut defaults from schema
-            fields = {name: (field.annotation, ...) for name, field in self.tool.args_schema.model_fields.items()}
+            # cut defaults from schema and remove chat_history for application as a tool
+            fields = {name: (field.annotation, ...) for name, field
+                      in self.tool.args_schema.model_fields.items() if name != 'chat_history'}
             input_schema = create_model('NewModel', **fields)
 
             llm = self.client.with_structured_output(input_schema)
@@ -87,6 +89,10 @@ Anwer must be JSON only extractable by JSON.LOADS."""
             result = _extract_json(completion.content.strip())
             logger.info(f"ToolNode tool params: {result}")
         try:
+            # handler for application added as a tool
+            if isinstance(self.tool, Application):
+                # set empty chat history
+                result['chat_history'] = None
             tool_result = self.tool.invoke(result, config=config, kwargs=kwargs)
             dispatch_custom_event(
                 "on_tool_node", {

{alita_sdk-0.3.126 → alita_sdk-0.3.127}/src/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.126
+Version: 0.3.127
 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