alita-sdk 0.3.266__tar.gz → 0.3.267__tar.gz

{alita_sdk-0.3.266 → alita_sdk-0.3.267}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.266
+Version: 0.3.267
 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>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.266 → alita_sdk-0.3.267}/alita_sdk/runtime/langchain/langraph_agent.py

@@ -422,7 +422,7 @@ def create_graph(
                             if not pipeline_name:
                                 raise ValueError("Subgraph must have a 'tool' node: add required tool to the subgraph node")
                             node_fn = SubgraphRunnable(
-                                inner=tool,
+                                inner=tool.graph,
                                 name=pipeline_name,
                                 input_mapping=node.get('input_mapping', {}),
                                 output_mapping=node.get('output_mapping', {}),
@@ -666,7 +666,9 @@ class LangGraphAgentRunnable(CompiledStateGraph):
                config: Optional[RunnableConfig] = None,
                *args, **kwargs):
         logger.info(f"Incomming Input: {input}")
-        if not config.get("configurable", {}).get("thread_id"):
+        if config is None:
+            config = RunnableConfig()
+        if not config.get("configurable", {}).get("thread_id", ""):
             config["configurable"] = {"thread_id": str(uuid4())}
         thread_id = config.get("configurable", {}).get("thread_id")
         # Handle chat history and current input properly

{alita_sdk-0.3.266 → alita_sdk-0.3.267}/alita_sdk/runtime/toolkits/subgraph.py

@@ -1,8 +1,11 @@
 from typing import List, Any
 
+from langchain_core.tools import BaseTool
+from langgraph.checkpoint.memory import MemorySaver
 from langgraph.graph.state import CompiledStateGraph
 
 from ..langchain.langraph_agent import create_graph, SUBGRAPH_REGISTRY
+from ..tools.graph import GraphTool
 from ..utils.utils import clean_string
 
 
@@ -16,7 +19,7 @@ class SubgraphToolkit:
         llm,
         app_api_key: str,
         selected_tools: list[str] = []
-    ) -> List[CompiledStateGraph]:
+    ) -> List[BaseTool]:
         from .tools import get_tools
         # from langgraph.checkpoint.memory import MemorySaver
 
@@ -36,18 +39,20 @@ class SubgraphToolkit:
         
         # For backward compatibility, still create a compiled graph stub
         # This is mainly used for identification in the parent graph's tools list
+        # For now the graph toolkit will have its own ephemeral in memory checkpoint memory.
         graph = create_graph(
             client=llm,
             tools=tools,
             yaml_schema=version_details['instructions'],
             debug=False,
             store=None,
-            memory=None,
-            for_subgraph=True,  # compile as raw subgraph
+            memory=MemorySaver(),
+            # for_subgraph=True,  # compile as raw subgraph
         )
-        
+
+        cleaned_subgraph_name = clean_string(subgraph_name)
         # Tag the graph stub for parent lookup
-        graph.name = clean_string(subgraph_name)
+        graph.name = cleaned_subgraph_name
         
         # Return the compiled graph stub for backward compatibility
-        return [graph]
+        return [GraphTool(description=app_details['description'], name=subgraph_name, graph=graph)]

alita_sdk-0.3.267/alita_sdk/runtime/tools/graph.py

@@ -0,0 +1,75 @@
+import json
+
+from langgraph.graph.state import CompiledStateGraph
+
+from ..utils.utils import clean_string
+from langchain_core.tools import BaseTool
+from langchain_core.messages import BaseMessage, AIMessage, ToolCall
+from typing import Any, Type, Optional, Union
+from pydantic import create_model, field_validator, BaseModel
+from pydantic.fields import FieldInfo
+from ..langchain.mixedAgentRenderes import convert_message_to_json
+from logging import getLogger
+
+logger = getLogger(__name__)
+
+graphToolSchema = create_model(
+    "graphToolSchema",
+    input=(str, FieldInfo(description="User Input for Graph")),
+    chat_history=(Optional[list[BaseMessage]],
+                  FieldInfo(description="Chat History relevant for Graph in format [{'role': '<user| assistant | etc>', 'content': '<content of the respected message>'}]", default=[]))
+)
+
+
+def formulate_query(kwargs):
+    chat_history = []
+    if kwargs.get('chat_history'):
+        if isinstance(kwargs.get('chat_history')[-1], BaseMessage):
+            chat_history = convert_message_to_json(kwargs.get('chat_history')[:])
+        elif isinstance(kwargs.get('chat_history')[-1], dict):
+            if all([True if message.get('role') and message.get('content') else False for message in
+                    kwargs.get('chat_history')]):
+                chat_history = kwargs.get('chat_history')[:]
+            else:
+                for each in kwargs.get('chat_history')[:]:
+                    chat_history.append(AIMessage(json.dumps(each)))
+        elif isinstance(kwargs.get('chat_history')[-1], str):
+            chat_history = []
+            for each in kwargs.get('chat_history')[:]:
+                chat_history.append(AIMessage(each))
+    elif kwargs.get('messages'):
+        chat_history = convert_message_to_json(kwargs.get('messages')[:])
+    result = {"input": kwargs.get('input'), "chat_history": chat_history}
+    for key, value in kwargs.items():
+        if key not in ("input", "chat_history"):
+            result[key] = value
+    return result
+
+
+class GraphTool(BaseTool):
+    name: str
+    description: str
+    graph: CompiledStateGraph
+    args_schema: Type[BaseModel] = graphToolSchema
+    return_type: str = "str"
+
+    @field_validator('name', mode='before')
+    @classmethod
+    def remove_spaces(cls, v):
+        return clean_string(v)
+
+    def invoke(self, input: Any, config: Optional[dict] = None, **kwargs: Any) -> Any:
+        """Override default invoke to preserve all fields, not just args_schema"""
+        schema_values = self.args_schema(**input).model_dump() if self.args_schema else {}
+        extras = {k: v for k, v in input.items() if k not in schema_values}
+        all_kwargs = {**kwargs, **extras, **schema_values}
+        if config is None:
+            config = {}
+        return self._run(*config, **all_kwargs)
+
+    def _run(self, *args, **kwargs):
+        response = self.graph.invoke(formulate_query(kwargs))
+        if self.return_type == "str":
+            return response["output"]
+        else:
+            return {"messages": [{"role": "assistant", "content": response["output"]}]}

{alita_sdk-0.3.266 → alita_sdk-0.3.267}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.266
+Version: 0.3.267
 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>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.266 → alita_sdk-0.3.267}/alita_sdk.egg-info/SOURCES.txt

@@ -105,6 +105,7 @@ alita_sdk/runtime/tools/artifact.py
 alita_sdk/runtime/tools/datasource.py
 alita_sdk/runtime/tools/echo.py
 alita_sdk/runtime/tools/function.py
+alita_sdk/runtime/tools/graph.py
 alita_sdk/runtime/tools/indexer_tool.py
 alita_sdk/runtime/tools/llm.py
 alita_sdk/runtime/tools/loop.py

{alita_sdk-0.3.266 → alita_sdk-0.3.267}/pyproject.toml

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