alita-sdk 0.3.609__py3-none-any.whl → 0.3.611__py3-none-any.whl

alita_sdk/runtime/langchain/langraph_agent.py

@@ -230,18 +230,32 @@ class StateDefaultNode(Runnable):
         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")
+                declared_type = value.get('type', '').lower()
+
+                # If the declared type is 'str' or 'string', preserve the string value
+                # Don't auto-convert even if it looks like a valid Python literal
+                if declared_type in ('str', 'string'):
                     result[key] = temp_value
+                else:
+                    # For other types, try to evaluate as Python literal
+                    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 PrinterNode(Runnable):
     name = "PrinterNode"
+    DEFAULT_FINAL_MSG = "How to proceed? To resume the pipeline - type anything..."
 
-    def __init__(self, input_mapping: Optional[dict[str, dict]]):
+    def __init__(self, input_mapping: Optional[dict[str, dict]], final_message: Optional[str] = None):
         self.input_mapping = input_mapping
+        # Apply fallback logic for empty/None values
+        if final_message and final_message.strip():
+            self.final_message = final_message.strip()
+        else:
+            self.final_message = self.DEFAULT_FINAL_MSG
 
     def invoke(self, state: BaseStore, config: Optional[RunnableConfig] = None) -> dict:
         logger.info(f"Printer Node - Current state variables: {state}")
@@ -261,7 +275,7 @@ class PrinterNode(Runnable):
             # convert formatted output to string if it's not
             if not isinstance(formatted_output, str):
                 formatted_output = str(formatted_output)
-            formatted_output += f"\n\n-----\n*How to proceed?*\n* *to resume the pipeline - type anything...*"
+            formatted_output += f"\n\n-----\n*{self.final_message}*"
         logger.debug(f"Formatted output: {formatted_output}")
         result[PRINTER_NODE_RS] = formatted_output
         return result
@@ -745,6 +759,7 @@ def create_graph(
             elif node_type == 'printer':
                 lg_builder.add_node(node_id, PrinterNode(
                     input_mapping=node.get('input_mapping', {'printer': {'type': 'fixed', 'value': ''}}),
+                    final_message=node.get('final_message'),
                 ))
 
                 # add interrupts after printer node if specified

alita_sdk/runtime/tools/function.py

@@ -1,3 +1,4 @@
+import base64
 import json
 import logging
 from copy import deepcopy
@@ -12,6 +13,7 @@ from langchain_core.utils.function_calling import convert_to_openai_tool
 from pydantic import ValidationError
 
 from ..langchain.utils import propagate_the_input_mapping
+from ..utils.serialization import safe_serialize
 
 logger = logging.getLogger(__name__)
 
@@ -40,15 +42,32 @@ class FunctionTool(BaseTool):
     alita_client: Optional[Any] = None
 
     def _prepare_pyodide_input(self, state: Union[str, dict, ToolCall]) -> str:
-        """Prepare input for PyodideSandboxTool by injecting state into the code block."""
-        # add state into the code block here since it might be changed during the execution of the code
+        """Prepare input for PyodideSandboxTool by injecting state into the code block.
+
+        Uses base64 encoding to avoid string escaping issues when passing JSON
+        through multiple layers of parsing (Python -> Deno -> Pyodide).
+        """
         state_copy = replace_escaped_newlines(deepcopy(state))
 
-        del state_copy['messages']  # remove messages to avoid issues with pickling without langchain-core
-        # inject state into the code block as alita_state variable
-        state_json = json.dumps(state_copy, ensure_ascii=False)
-        pyodide_predata = f'#state dict\nimport json\nalita_state = json.loads({json.dumps(state_json)})\n'
+        # remove messages to avoid issues with pickling without langchain-core
+        if 'messages' in state_copy:
+            del state_copy['messages']
+
+        # Use safe_serialize to handle Pydantic models, datetime, and other non-JSON types
+        state_json = safe_serialize(state_copy)
 
+        # Use base64 encoding to avoid all string escaping issues
+        # This is more robust than repr() when the code passes through multiple parsers
+        state_json_b64 = base64.b64encode(state_json.encode('utf-8')).decode('ascii')
+
+        # Generate code that decodes base64 and parses JSON inside Pyodide
+        pyodide_predata = f'''#state dict
+import json
+import base64
+_state_json_b64 = "{state_json_b64}"
+_state_json = base64.b64decode(_state_json_b64).decode('utf-8')
+alita_state = json.loads(_state_json)
+'''
         return pyodide_predata
 
     def _handle_pyodide_output(self, tool_result: Any) -> dict:

alita_sdk/runtime/utils/serialization.py

@@ -0,0 +1,155 @@
+"""
+Serialization utilities for safe JSON encoding of complex objects.
+
+Handles Pydantic models, LangChain messages, datetime objects, and other
+non-standard types that may appear in state variables.
+"""
+import json
+import logging
+from datetime import datetime, date
+from typing import Any
+
+logger = logging.getLogger(__name__)
+
+
+def _convert_to_serializable(obj: Any, _seen: set = None) -> Any:
+    """
+    Recursively convert an object to JSON-serializable primitives.
+
+    Handles nested dicts and lists that may contain non-serializable objects.
+    Uses a seen set to prevent infinite recursion with circular references.
+
+    Args:
+        obj: Any object to convert
+        _seen: Internal set to track seen object ids (for circular reference detection)
+
+    Returns:
+        JSON-serializable representation of the object
+    """
+    # Initialize seen set for circular reference detection
+    if _seen is None:
+        _seen = set()
+
+    # Check for circular references (only for mutable objects)
+    obj_id = id(obj)
+    if isinstance(obj, (dict, list, set)) and obj_id in _seen:
+        return f"<circular reference: {type(obj).__name__}>"
+
+    # Primitives - return as-is
+    if obj is None or isinstance(obj, (str, int, float, bool)):
+        return obj
+
+    # Add to seen set for mutable containers
+    if isinstance(obj, (dict, list, set)):
+        _seen = _seen | {obj_id}  # Create new set to avoid mutation issues
+
+    # Dict - recursively process all values
+    if isinstance(obj, dict):
+        return {
+            _convert_to_serializable(k, _seen): _convert_to_serializable(v, _seen)
+            for k, v in obj.items()
+        }
+
+    # List/tuple - recursively process all items
+    if isinstance(obj, (list, tuple)):
+        return [_convert_to_serializable(item, _seen) for item in obj]
+
+    # Set - convert to list and process
+    if isinstance(obj, set):
+        return [_convert_to_serializable(item, _seen) for item in obj]
+
+    # Bytes - decode to string
+    if isinstance(obj, bytes):
+        try:
+            return obj.decode('utf-8')
+        except UnicodeDecodeError:
+            return obj.decode('utf-8', errors='replace')
+
+    # Datetime objects
+    if isinstance(obj, datetime):
+        return obj.isoformat()
+    if isinstance(obj, date):
+        return obj.isoformat()
+
+    # Pydantic BaseModel (v2) - check for model_dump method
+    if hasattr(obj, 'model_dump') and callable(getattr(obj, 'model_dump')):
+        try:
+            return _convert_to_serializable(obj.model_dump(), _seen)
+        except Exception as e:
+            logger.debug(f"Failed to call model_dump on {type(obj).__name__}: {e}")
+
+    # Pydantic BaseModel (v1) - check for dict method
+    if hasattr(obj, 'dict') and callable(getattr(obj, 'dict')) and hasattr(obj, '__fields__'):
+        try:
+            return _convert_to_serializable(obj.dict(), _seen)
+        except Exception as e:
+            logger.debug(f"Failed to call dict on {type(obj).__name__}: {e}")
+
+    # LangChain BaseMessage - extract key fields
+    if hasattr(obj, 'type') and hasattr(obj, 'content'):
+        try:
+            result = {
+                "type": obj.type,
+                "content": _convert_to_serializable(obj.content, _seen),
+            }
+            if hasattr(obj, 'additional_kwargs') and obj.additional_kwargs:
+                result["additional_kwargs"] = _convert_to_serializable(obj.additional_kwargs, _seen)
+            if hasattr(obj, 'name') and obj.name:
+                result["name"] = obj.name
+            return result
+        except Exception as e:
+            logger.debug(f"Failed to extract message fields from {type(obj).__name__}: {e}")
+
+    # Objects with __dict__ attribute (custom classes)
+    if hasattr(obj, '__dict__'):
+        try:
+            return _convert_to_serializable(obj.__dict__, _seen)
+        except Exception as e:
+            logger.debug(f"Failed to serialize __dict__ of {type(obj).__name__}: {e}")
+
+    # UUID objects
+    if hasattr(obj, 'hex') and hasattr(obj, 'int'):
+        return str(obj)
+
+    # Enum objects
+    if hasattr(obj, 'value') and hasattr(obj, 'name') and hasattr(obj.__class__, '__members__'):
+        return obj.value
+
+    # Last resort - convert to string
+    try:
+        return str(obj)
+    except Exception:
+        return f"<non-serializable: {type(obj).__name__}>"
+
+
+def safe_serialize(obj: Any, **kwargs) -> str:
+    """
+    Safely serialize any object to a JSON string.
+
+    Pre-processes the entire object tree to convert non-serializable
+    objects before passing to json.dumps. This ensures nested dicts
+    and lists with non-standard objects are handled correctly.
+
+    Args:
+        obj: Any object to serialize
+        **kwargs: Additional arguments passed to json.dumps
+            (e.g., indent, sort_keys)
+
+    Returns:
+        JSON string representation of the object
+
+    Example:
+        >>> from pydantic import BaseModel
+        >>> class User(BaseModel):
+        ...     name: str
+        >>> state = {"user": User(name="Alice"), "count": 5}
+        >>> safe_serialize(state)
+        '{"user": {"name": "Alice"}, "count": 5}'
+    """
+    # Pre-process the entire object tree
+    serializable = _convert_to_serializable(obj)
+
+    # Set defaults
+    kwargs.setdefault('ensure_ascii', False)
+
+    return json.dumps(serializable, **kwargs)

{alita_sdk-0.3.609.dist-info → alita_sdk-0.3.611.dist-info}/METADATA

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

@@ -119,7 +119,7 @@ alita_sdk/runtime/langchain/assistant.py,sha256=RcU82QyxtLW8yWeamEBqYJ6B243ehRya
 alita_sdk/runtime/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/runtime/langchain/constants.py,sha256=GxZP0dupwaVq0iTKE2nNF-Vp0wzMlkh_aiBAhCuy84E,29451
 alita_sdk/runtime/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
-alita_sdk/runtime/langchain/langraph_agent.py,sha256=-cMdn3LdCImV82HiSQdgb6f7jA4LeEWjTso9f-KHsng,63140
+alita_sdk/runtime/langchain/langraph_agent.py,sha256=7mkOpl_cBCb6hZqNCN-B9TePRQjnblLMWDOzcGni-rY,63957
 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
@@ -199,7 +199,7 @@ alita_sdk/runtime/tools/artifact.py,sha256=Us-NM1VTfqDNBzs8nMsI3Inu-_V9SoZOqjfUA
 alita_sdk/runtime/tools/data_analysis.py,sha256=PHQ0xa2eDkw6FsHAHVTWB58wO8tg76tHrp4lXRQZ0jQ,6396
 alita_sdk/runtime/tools/datasource.py,sha256=pvbaSfI-ThQQnjHG-QhYNSTYRnZB0rYtZFpjCfpzxYI,2443
 alita_sdk/runtime/tools/echo.py,sha256=spw9eCweXzixJqHnZofHE1yWiSUa04L4VKycf3KCEaM,486
-alita_sdk/runtime/tools/function.py,sha256=HSMO1nBTRKMvWC_m0M8TOLGaZ2k_7ksPgLqzuRh6kV4,7083
+alita_sdk/runtime/tools/function.py,sha256=x5gaaCrxgA5vwq29pUpAgGA5l6THriOJx9d58-G9Ybo,7679
 alita_sdk/runtime/tools/graph.py,sha256=7jImBBSEdP5Mjnn2keOiyUwdGDFhEXLUrgUiugO3mgA,3503
 alita_sdk/runtime/tools/image_generation.py,sha256=waxxFIAgmh9-COcljL9uZ7e_s7EL9OWveUxYk0ulEUM,7855
 alita_sdk/runtime/tools/indexer_tool.py,sha256=whSLPevB4WD6dhh2JDXEivDmTvbjiMV1MrPl9cz5eLA,4375
@@ -230,6 +230,7 @@ alita_sdk/runtime/utils/mcp_oauth.py,sha256=YfMFZOs1nGSR7ZvGKSE6nKlpJRNkr2h0BZXk
 alita_sdk/runtime/utils/mcp_sse_client.py,sha256=wU_fmZbh2sXUYjkJmBjmpfhFLpncjAXE5Wg3fqyNhmk,18318
 alita_sdk/runtime/utils/mcp_tools_discovery.py,sha256=pL1bMUEaZmiABFGsphed0LvhR-wGfkNnCpD_89R01Ws,3908
 alita_sdk/runtime/utils/save_dataframe.py,sha256=i-E1wp-t4wb17Zq3nA3xYwgSILjoXNizaQAA9opWvxY,1576
+alita_sdk/runtime/utils/serialization.py,sha256=J7cQS7-SYD8haUv5Inn44EfsTzmAzOquacgn_t2GU-g,5327
 alita_sdk/runtime/utils/streamlit.py,sha256=0TotNKnvMPHuwBdhMEpM5DhIedQQa1AUz9BlmXFBhAU,107179
 alita_sdk/runtime/utils/toolkit_runtime.py,sha256=MU63Fpxj0b5_r1IUUc0Q3-PN9VwL7rUxp2MRR4tmYR8,5136
 alita_sdk/runtime/utils/toolkit_utils.py,sha256=c56a9jzlOXYYdl67N3arGl5sOqO3ly60bne2OBYuQKc,6758
@@ -458,9 +459,9 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=ppJayzkKRhTbQVVd2EhQmvADwdosl
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=cUSc0ZhpGmWnTQ3ZjllU9QmNlCfaHZ21HCFjfenSMH8,3081
 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.609.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.609.dist-info/METADATA,sha256=hEWxQjw26WUFFy2f56DL-k9cuqD3Xxxszb6f4YEFtC8,24339
-alita_sdk-0.3.609.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.609.dist-info/entry_points.txt,sha256=VijN0h4alp1WXm8tfS3P7vuGxN4a5RZqHjXAoEIBZnI,49
-alita_sdk-0.3.609.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.609.dist-info/RECORD,,
+alita_sdk-0.3.611.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.611.dist-info/METADATA,sha256=VbmEWybS3DmRjXF8BIB5vG8dLgO5HS7Vm9_qbMG7miQ,24339
+alita_sdk-0.3.611.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.611.dist-info/entry_points.txt,sha256=VijN0h4alp1WXm8tfS3P7vuGxN4a5RZqHjXAoEIBZnI,49
+alita_sdk-0.3.611.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.611.dist-info/RECORD,,