alita-sdk 0.3.337__py3-none-any.whl → 0.3.338__py3-none-any.whl

alita_sdk/runtime/tools/sandbox.py

@@ -1,6 +1,7 @@
 import logging
 import asyncio
 import subprocess
+import os
 from typing import Any, Type, Optional, Union, Dict
 from langchain_core.tools import BaseTool
 from pydantic import BaseModel, Field, create_model
@@ -22,6 +23,42 @@ def _is_deno_available() -> bool:
     except (subprocess.TimeoutExpired, subprocess.CalledProcessError, FileNotFoundError):
         return False
 
+
+def _setup_pyodide_cache_env() -> None:
+    """Setup Pyodide caching environment variables for performance optimization"""
+    try:
+        # Check if cache environment file exists and source it
+        cache_env_file = os.path.expanduser("~/.pyodide_cache_env")
+        if os.path.exists(cache_env_file):
+            with open(cache_env_file, 'r') as f:
+                for line in f:
+                    line = line.strip()
+                    if line.startswith('export ') and '=' in line:
+                        # Parse export VAR=value format
+                        var_assignment = line[7:]  # Remove 'export '
+                        if '=' in var_assignment:
+                            key, value = var_assignment.split('=', 1)
+                            # Remove quotes if present
+                            value = value.strip('"').strip("'")
+                            os.environ[key] = value
+                            logger.debug(f"Set Pyodide cache env: {key}={value}")
+        
+        # Set default caching environment variables if not already set
+        cache_defaults = {
+            'PYODIDE_PACKAGES_PATH': os.path.expanduser('~/.cache/pyodide'),
+            'DENO_DIR': os.path.expanduser('~/.cache/deno'),
+            'PYODIDE_CACHE_DIR': os.path.expanduser('~/.cache/pyodide'),
+        }
+        
+        for key, default_value in cache_defaults.items():
+            if key not in os.environ:
+                os.environ[key] = default_value
+                logger.debug(f"Set default Pyodide env: {key}={default_value}")
+                
+    except Exception as e:
+        logger.warning(f"Could not setup Pyodide cache environment: {e}")
+
+
 # Create input schema for the sandbox tool
 sandbox_tool_input = create_model(
     "SandboxToolInput", 
@@ -33,6 +70,7 @@ class PyodideSandboxTool(BaseTool):
     """
     A tool that provides secure Python code execution using Pyodide (Python compiled to WebAssembly).
     This tool leverages langchain-sandbox to provide a safe environment for running untrusted Python code.
+    Optimized for performance with caching and stateless execution by default.
     """
     
     name: str = "pyodide_sandbox"
@@ -46,9 +84,10 @@ class PyodideSandboxTool(BaseTool):
     
     The sandbox supports most Python standard library modules and can install additional packages.
     Note: File access and some system operations are restricted for security.
+    Optimized for performance with local caching (stateless by default for faster execution).
     """
     args_schema: Type[BaseModel] = sandbox_tool_input
-    stateful: bool = True
+    stateful: bool = False  # Default to stateless for better performance
     allow_net: bool = True
     session_bytes: Optional[bytes] = None
     session_metadata: Optional[Dict] = None
@@ -56,26 +95,32 @@ class PyodideSandboxTool(BaseTool):
     def __init__(self, **kwargs: Any) -> None:
         super().__init__(**kwargs)
         self._sandbox = None
+        # Setup caching environment for optimal performance
+        _setup_pyodide_cache_env()
         self._initialize_sandbox()
     
     def _initialize_sandbox(self) -> None:
-        """Initialize the PyodideSandbox instance"""
+        """Initialize the PyodideSandbox instance with optimized settings"""
         try:
             # Check if Deno is available
             if not _is_deno_available():
                 error_msg = (
                     "Deno is required for PyodideSandbox but is not installed. "
-                    "Please install Deno manually or ensure it's available in PATH."
+                    "Please run the bootstrap.sh script or install Deno manually."
                 )
                 logger.error(error_msg)
                 raise RuntimeError(error_msg)
             
             from langchain_sandbox import PyodideSandbox
+            
+            # Configure sandbox with performance optimizations
             self._sandbox = PyodideSandbox(
                 stateful=self.stateful,
-                allow_net=self.allow_net
+                allow_net=self.allow_net,
+                # Use auto node_modules_dir for better caching
+                node_modules_dir="auto"
             )
-            logger.info("PyodideSandbox initialized successfully")
+            logger.info(f"PyodideSandbox initialized successfully (stateful={self.stateful})")
         except ImportError as e:
             if "langchain_sandbox" in str(e):
                 error_msg = (
@@ -205,10 +250,10 @@ def create_sandbox_tool(stateful: bool = False, allow_net: bool = True) -> BaseT
     Factory function to create sandbox tools with specified configuration.
     
     Note: This tool requires Deno to be installed and available in PATH.
-    For installation, use the bootstrap.sh script or install manually.
+    For installation and optimization, run the bootstrap.sh script.
     
     Args:
-        stateful: Whether to maintain state between executions
+        stateful: Whether to maintain state between executions (default: False for better performance)
         allow_net: Whether to allow network access (for package installation)
     
     Returns:
@@ -217,6 +262,11 @@ def create_sandbox_tool(stateful: bool = False, allow_net: bool = True) -> BaseT
     Raises:
         ImportError: If langchain-sandbox is not installed
         RuntimeError: If Deno is not found in PATH
+        
+    Performance Notes:
+        - Stateless mode (default) is faster and avoids session state overhead
+        - Run bootstrap.sh script to enable local caching and reduce initialization time
+        - Cached wheels reduce package download time from ~4.76s to near-instant
     """
     if stateful:
         return StatefulPyodideSandboxTool(allow_net=allow_net)

{alita_sdk-0.3.337.dist-info → alita_sdk-0.3.338.dist-info}/METADATA

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

@@ -119,7 +119,7 @@ alita_sdk/runtime/tools/mcp_server_tool.py,sha256=MhLxZJ44LYrB_0GrojmkyqKoDRaqIH
 alita_sdk/runtime/tools/pgvector_search.py,sha256=NN2BGAnq4SsDHIhUcFZ8d_dbEOM8QwB0UwpsWCYruXU,11692
 alita_sdk/runtime/tools/prompt.py,sha256=nJafb_e5aOM1Rr3qGFCR-SKziU9uCsiP2okIMs9PppM,741
 alita_sdk/runtime/tools/router.py,sha256=wCvZjVkdXK9dMMeEerrgKf5M790RudH68pDortnHSz0,1517
-alita_sdk/runtime/tools/sandbox.py,sha256=_v8ACXV4POFRIuM7iU1NTm3-OL5pBmONl5z7bjDmBtw,9053
+alita_sdk/runtime/tools/sandbox.py,sha256=WNz-aUMtkGCPg84dDy_0BPkyp-6YjoYB-xjIEFFrtKw,11601
 alita_sdk/runtime/tools/tool.py,sha256=lE1hGi6qOAXG7qxtqxarD_XMQqTghdywf261DZawwno,5631
 alita_sdk/runtime/tools/vectorstore.py,sha256=UFBAJ_N2F6uB0xxIy1VMx581tHco-xDl7v2Hl6u0Xzw,34468
 alita_sdk/runtime/tools/vectorstore_base.py,sha256=qmidtzD2SFZdUDrmKNkUWnuBHqJDjPbQQJ_z1TqIl0g,27283
@@ -350,8 +350,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.337.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.337.dist-info/METADATA,sha256=BWEeyZsFO8oG3JOc7z8vy-nS-xJZxruNuEE1zm29HQc,18901
-alita_sdk-0.3.337.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.337.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.337.dist-info/RECORD,,
+alita_sdk-0.3.338.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.338.dist-info/METADATA,sha256=JCh2biWyamVC1ZnDrAOGqrxctL38ArPoq4Uu_QTLw9w,18901
+alita_sdk-0.3.338.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.338.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.338.dist-info/RECORD,,