alita-sdk 0.3.319__py3-none-any.whl → 0.3.321__py3-none-any.whl

alita_sdk/runtime/langchain/langraph_agent.py

@@ -584,14 +584,11 @@ 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.")
-        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
+        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))
         else:
             # if no state variables are defined, set the entry point directly
             lg_builder.set_entry_point(entry_point)
@@ -633,6 +630,24 @@ def create_graph(
     )
     return compiled.validate()
 
+def set_defaults(d):
+    """Set default values for dictionary entries based on their type."""
+    type_defaults = {
+        'str': '',
+        'list': [],
+        'int': 0,
+        'float': 0.0,
+        'bool': False,
+        # add more types as needed
+    }
+    for k, v in d.items():
+        # Skip 'input' key as it is not a state initial variable
+        if k == 'input':
+            continue
+        # set value or default if type is defined
+        if 'value' not in v:
+            v['value'] = type_defaults.get(v['type'], None)
+    return d
 
 def convert_dict_to_message(msg_dict):
     """Convert a dictionary message to a LangChain message object."""

alita_sdk/tools/figma/api_wrapper.py

@@ -320,6 +320,10 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
             )
         ]
 
+        if not node_ids:
+            yield from ()
+            return
+
         images = self._client.get_file_images(file_key, node_ids).images or {}
         total_images = len(images)
         if total_images == 0:

alita_sdk/tools/utils/content_parser.py

@@ -207,7 +207,7 @@ def process_document_by_type(content, extension_source: str, document: Document
     try:
         chunks = process_content_by_type(content, extension_source, llm, chunking_config)
     except Exception as e:
-        msg = f"Error during content for file {extension_source}:\n{e}"
+        msg = f"Error during content parsing for file {extension_source}:\n{e}"
         logger.warning(msg)
         yield Document(
             page_content=msg,
@@ -218,7 +218,7 @@ def process_document_by_type(content, extension_source: str, document: Document
         yield Document(
             page_content=sanitize_for_postgres(chunk.page_content),
             metadata={**document.metadata, **chunk.metadata}
-                )
+        )
 
 
 def process_content_by_type(content, filename: str, llm=None, chunking_config=None) -> \
@@ -233,7 +233,7 @@ def process_content_by_type(content, filename: str, llm=None, chunking_config=No
             if content is None:
                 logger.warning(
                     f"'{IndexerKeywords.CONTENT_IN_BYTES.value}' ie expected but not found in document metadata.")
-                return
+                return []
 
             temp_file.write(content)
             temp_file.flush()
@@ -241,7 +241,7 @@ def process_content_by_type(content, filename: str, llm=None, chunking_config=No
             loader_config = loaders_map.get(extension)
             if not loader_config:
                 logger.warning(f"No loader found for file extension: {extension}. File: {temp_file_path}")
-                return
+                return []
 
             loader_cls = loader_config['class']
             loader_kwargs = loader_config['kwargs']

{alita_sdk-0.3.319.dist-info → alita_sdk-0.3.321.dist-info}/METADATA

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

@@ -44,7 +44,7 @@ alita_sdk/runtime/langchain/assistant.py,sha256=0Cv30SJULxlrvD3hzWmTnxZyzqPEaSgp
 alita_sdk/runtime/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/runtime/langchain/constants.py,sha256=eHVJ_beJNTf1WJo4yq7KMK64fxsRvs3lKc34QCXSbpk,3319
 alita_sdk/runtime/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
-alita_sdk/runtime/langchain/langraph_agent.py,sha256=3s1QLEPu0U0pz-PVaLBIEZLp7dhqAxy2wNUIjvIt49I,43925
+alita_sdk/runtime/langchain/langraph_agent.py,sha256=5fTT2FyRFSLIbQxm8KkxT_-bkoABmZzZA_V3-9nPse0,44296
 alita_sdk/runtime/langchain/mixedAgentParser.py,sha256=M256lvtsL3YtYflBCEp-rWKrKtcY1dJIyRGVv7KW9ME,2611
 alita_sdk/runtime/langchain/mixedAgentRenderes.py,sha256=asBtKqm88QhZRILditjYICwFVKF5KfO38hu2O-WrSWE,5964
 alita_sdk/runtime/langchain/store_manager.py,sha256=i8Fl11IXJhrBXq1F1ukEVln57B1IBe-tqSUvfUmBV4A,2218
@@ -235,7 +235,7 @@ alita_sdk/tools/custom_open_api/api_wrapper.py,sha256=sDSFpvEqpSvXHGiBISdQQcUecf
 alita_sdk/tools/elastic/__init__.py,sha256=iwnSRppRpzvJ1da2K3Glu8Uu41MhBDCYbguboLkEbW0,2818
 alita_sdk/tools/elastic/api_wrapper.py,sha256=pl8CqQxteJAGwyOhMcld-ZgtOTFwwbv42OITQVe8rM0,1948
 alita_sdk/tools/figma/__init__.py,sha256=W6vIMMkZI2Lmpg6_CRRV3oadaIbVI-qTLmKUh6enqWs,4509
-alita_sdk/tools/figma/api_wrapper.py,sha256=s7AR6janhehyZi2_u3-I90_8MG_VO_LHo4WQucR3zHI,26103
+alita_sdk/tools/figma/api_wrapper.py,sha256=SjTKCq8FrW1nCP1uFuRw8A2WVKnzEyRmKEr8e3W4QjE,26174
 alita_sdk/tools/github/__init__.py,sha256=2rHu0zZyZGnLC5CkHgDIhe14N9yCyaEfrrt7ydH8478,5191
 alita_sdk/tools/github/api_wrapper.py,sha256=uDwYckdnpYRJtb0uZnDkaz2udvdDLVxuCh1tSwspsiU,8411
 alita_sdk/tools/github/github_client.py,sha256=nxnSXsDul2PPbWvYZS8TmAFFmR-5ALyakNoV5LN2D4U,86617
@@ -327,7 +327,7 @@ alita_sdk/tools/testrail/__init__.py,sha256=Xg4nVjULL_D8JpIXLYXppnwUfGF4-lguFwKH
 alita_sdk/tools/testrail/api_wrapper.py,sha256=PKhtf04C6PFDexGCAJm-hjA9Gpu4crx6EXKT5K-b_Pk,32985
 alita_sdk/tools/utils/__init__.py,sha256=155xepXPr4OEzs2Mz5YnjXcBpxSv1X2eznRUVoPtyK0,3268
 alita_sdk/tools/utils/available_tools_decorator.py,sha256=IbrdfeQkswxUFgvvN7-dyLMZMyXLiwvX7kgi3phciCk,273
-alita_sdk/tools/utils/content_parser.py,sha256=Z6YQ__GONqRE0Ifhhhjc3A2SgYsxbTejBUNHw6MCdJA,14057
+alita_sdk/tools/utils/content_parser.py,sha256=7k5Ddv3Nzp3UoocgslwwSXi1G9ZR7sXzj6593IDeOcM,14063
 alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=ypBEAkFRGHv5edW0N9rdo1yKurNGQ4pRVEWtrN_7SeA,17656
 alita_sdk/tools/vector_adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/tools/xray/__init__.py,sha256=eOMWP8VamFbbJgt1xrGpGPqB9ByOTA0Cd3LCaETzGk4,4376
@@ -349,8 +349,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=kT0TbmMvuKhDUZc0i7KO18O38JM9S
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=0ne8XLJEQSLOWfzd2HdnqOYmQlUliKHbBED5kW_Vias,2895
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=kmw_xol8YIYFplBLWTqP_VKPRhL_1ItDD0_vXTe_UuI,14906
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=R371waHsms4sllHCbijKYs90C-9Yu0sSR3N4SUfQOgU,5066
-alita_sdk-0.3.319.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.319.dist-info/METADATA,sha256=UnHRV2v-GuuFB5dxevPtsCNlvqenMZXte0wsRZakGjc,18897
-alita_sdk-0.3.319.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.319.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.319.dist-info/RECORD,,
+alita_sdk-0.3.321.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.321.dist-info/METADATA,sha256=NYftgHbUR29AEwXHSYD4vDsVk94-TYbo4ijPAl7FeJQ,18897
+alita_sdk-0.3.321.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.321.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.321.dist-info/RECORD,,