alita-sdk 0.3.321__tar.gz → 0.3.321b1__tar.gz

{alita_sdk-0.3.321 → alita_sdk-0.3.321b1}/PKG-INFO

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

{alita_sdk-0.3.321 → alita_sdk-0.3.321b1}/alita_sdk/runtime/langchain/langraph_agent.py

@@ -584,11 +584,19 @@ def create_graph(
             entry_point = clean_string(schema['entry_point'])
         except KeyError:
             raise ToolException("Entry point is not defined in the schema. Please define 'entry_point' in the schema.")
-        if state.items():
-            state_default_node = StateDefaultNode(default_vars=set_defaults(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))
+        # if state.items():
+        #     state_default_node = StateDefaultNode(default_vars=set_defaults(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))
+        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)
@@ -710,7 +718,7 @@ class LangGraphAgentRunnable(CompiledStateGraph):
         else:
             result = super().invoke(input, config=config, *args, **kwargs)
         try:
-            if self.output_variables and self.output_variables[0] != "messages":
+            if self.output_variables and self.output_variables[0] in result 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:

{alita_sdk-0.3.321 → alita_sdk-0.3.321b1}/alita_sdk/tools/non_code_indexer_toolkit.py

@@ -1,4 +1,5 @@
 from langchain_core.documents import Document
+from langchain_core.tools import ToolException
 
 from alita_sdk.runtime.utils.utils import IndexerKeywords
 from alita_sdk.tools.base_indexer_toolkit import BaseIndexerToolkit
@@ -6,6 +7,9 @@ from alita_sdk.tools.base_indexer_toolkit import BaseIndexerToolkit
 
 class NonCodeIndexerToolkit(BaseIndexerToolkit):
     def _get_indexed_data(self, collection_suffix: str):
+        if not self.vector_adapter:
+            raise ToolException("Vector adapter is not initialized. "
+                             "Check your configuration: embedding_model and vectorstore_type.")
         return self.vector_adapter.get_indexed_data(self, collection_suffix)
 
     def key_fn(self, document: Document):

{alita_sdk-0.3.321 → alita_sdk-0.3.321b1}/alita_sdk.egg-info/PKG-INFO

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

{alita_sdk-0.3.321 → alita_sdk-0.3.321b1}/alita_sdk.egg-info/top_level.txt

@@ -1,2 +1,3 @@
 alita_sdk
 docs
+src

{alita_sdk-0.3.321 → alita_sdk-0.3.321b1}/pyproject.toml

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