alita-sdk 0.3.170__tar.gz → 0.3.172__tar.gz

{alita_sdk-0.3.170 → alita_sdk-0.3.172}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.170
+Version: 0.3.172
 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.170 → alita_sdk-0.3.172}/alita_sdk/runtime/langchain/langraph_agent.py

@@ -285,7 +285,7 @@ class StateModifierNode(Runnable):
 
 
 
-def prepare_output_schema(lg_builder, memory, store, debug=False, interrupt_before=None, interrupt_after=None, state_class=None):
+def prepare_output_schema(lg_builder, memory, store, debug=False, interrupt_before=None, interrupt_after=None, state_class=None, output_variables=None):
     # prepare output channels
     if interrupt_after is None:
         interrupt_after = []
@@ -328,7 +328,8 @@ def prepare_output_schema(lg_builder, memory, store, debug=False, interrupt_befo
         auto_validate=False,
         debug=debug,
         store=store,
-        schema_to_mapper=state_class
+        schema_to_mapper=state_class,
+        output_variables=output_variables
     )
 
     compiled.attach_node(START, None)
@@ -597,13 +598,16 @@ def create_graph(
         lg_builder, memory, store, debug,
         interrupt_before=interrupt_before,
         interrupt_after=interrupt_after,
-        state_class={state_class: None}
+        state_class={state_class: None},
+        output_variables=node.get('output', [])
     )
     return compiled.validate()
 
 
 class LangGraphAgentRunnable(CompiledStateGraph):
-    builder: CompiledStateGraph
+    def __init__(self, *args, output_variables=None, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.output_variables = output_variables
 
     def invoke(self, input: Union[dict[str, Any], Any],
                config: Optional[RunnableConfig] = None,
@@ -624,7 +628,11 @@ class LangGraphAgentRunnable(CompiledStateGraph):
         else:
             result = super().invoke(input, config=config, *args, **kwargs)
         try:
-            output = result['messages'][-1].content
+            if self.output_variables and self.output_variables[0] != "messages":
+                # If output_variables are specified, use the value of first one or use the last messages as default
+                output = result.get(self.output_variables[0], result['messages'][-1].content)
+            else:
+                output = result['messages'][-1].content
         except:
             output = list(result.values())[-1]
         thread_id = None

{alita_sdk-0.3.170 → alita_sdk-0.3.172}/alita_sdk/tools/ado/repos/__init__.py

@@ -4,10 +4,9 @@ from langchain_core.tools import BaseTool, BaseToolkit
 from pydantic import BaseModel, Field, create_model, SecretStr
 
 import requests
-from alita_tools.utils import check_connection_response
 from ...base.tool import BaseAction
 from .repos_wrapper import ReposApiWrapper
-from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
+from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length, check_connection_response
 
 name = "ado_repos"
 

{alita_sdk-0.3.170 → alita_sdk-0.3.172}/alita_sdk/tools/github/github_client.py

@@ -1136,7 +1136,7 @@ class GitHubClient(BaseModel):
 
         return paired_contents
 
-    def update_file(self, file_query: str, repo_name: Optional[str] = None) -> str:
+    def update_file(self, file_query: str, repo_name: Optional[str] = None, commit_message: Optional[str] = None) -> str:
         """
         Updates a file with new content.
         Parameters:
@@ -1183,7 +1183,7 @@ class GitHubClient(BaseModel):
 
             repo.update_file(
                 path=file_path,
-                message=f"Update {file_path}",
+                message=commit_message if commit_message else f"Update {file_path}",
                 content=updated_file_content,
                 branch=branch,
                 sha=repo.get_contents(file_path, ref=branch).sha,

{alita_sdk-0.3.170 → alita_sdk-0.3.172}/alita_sdk/tools/github/schemas.py

@@ -43,7 +43,8 @@ ReadFile = create_model(
 UpdateFile = create_model(
     "UpdateFile",
     file_query=(str, Field(description="File path and content to update with OLD and NEW markers")),
-    repo_name=(Optional[str], Field(default=None, description="Name of the repository (e.g., 'owner/repo'). If None, uses the default repository."))
+    repo_name=(Optional[str], Field(default=None, description="Name of the repository (e.g., 'owner/repo'). If None, uses the default repository.")),
+    commit_message=(Optional[str], Field(default=None, description="Commit message for the update operation")),
 )
 
 CreateFile = create_model(

{alita_sdk-0.3.170 → alita_sdk-0.3.172}/alita_sdk.egg-info/PKG-INFO

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

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