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

alita_sdk/runtime/langchain/assistant.py

@@ -176,6 +176,22 @@ class Assistant:
         # Exclude compiled graph runnables from simple tool agents
         simple_tools = [t for t in self.tools if isinstance(t, (BaseTool, CompiledStateGraph))]
         
+        # Add sandbox tool by default for react agents
+        try:
+            from ..tools.sandbox import create_sandbox_tool
+            sandbox_tool = create_sandbox_tool(stateful=True, allow_net=True)
+            simple_tools.append(sandbox_tool)
+            logger.info("Added PyodideSandboxTool to react agent")
+        except ImportError as e:
+            logger.warning(f"Failed to add PyodideSandboxTool: {e}. Install langchain-sandbox to enable this feature.")
+        except RuntimeError as e:
+            if "Deno" in str(e):
+                logger.warning("Failed to add PyodideSandboxTool: Deno is required. Install from https://docs.deno.com/runtime/getting_started/installation/")
+            else:
+                logger.warning(f"Failed to add PyodideSandboxTool: {e}")
+        except Exception as e:
+            logger.error(f"Error adding PyodideSandboxTool: {e}")
+        
         # Set up memory/checkpointer if available
         checkpointer = None
         if self.memory is not None:

alita_sdk/runtime/tools/__init__.py

@@ -0,0 +1,14 @@
+"""
+Runtime tools module for Alita SDK.
+This module provides various tools that can be used within LangGraph agents.
+"""
+
+from .sandbox import PyodideSandboxTool, StatefulPyodideSandboxTool, create_sandbox_tool
+from .echo import EchoTool
+
+__all__ = [
+    "PyodideSandboxTool",
+    "StatefulPyodideSandboxTool", 
+    "create_sandbox_tool",
+    "EchoTool"
+]

alita_sdk/runtime/tools/sandbox.py

@@ -0,0 +1,224 @@
+import logging
+import asyncio
+import subprocess
+from typing import Any, Type, Optional, Union, Dict
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel, Field, create_model
+from pydantic.fields import FieldInfo
+
+logger = logging.getLogger(__name__)
+
+
+def _is_deno_available() -> bool:
+    """Check if Deno is available in the PATH"""
+    try:
+        result = subprocess.run(
+            ["deno", "--version"], 
+            capture_output=True, 
+            text=True, 
+            timeout=10
+        )
+        return result.returncode == 0
+    except (subprocess.TimeoutExpired, subprocess.CalledProcessError, FileNotFoundError):
+        return False
+
+# Create input schema for the sandbox tool
+sandbox_tool_input = create_model(
+    "SandboxToolInput", 
+    code=(str, FieldInfo(description="Python code to execute in the sandbox environment"))
+)
+
+
+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.
+    """
+    
+    name: str = "pyodide_sandbox"
+    description: str = """Execute Python code in a secure sandbox environment using Pyodide.
+    This tool allows safe execution of Python code without access to the host system.
+    Use this tool when you need to:
+    - Execute Python code snippets
+    - Perform calculations or data analysis
+    - Test Python algorithms
+    - Run code that requires isolation from the host system
+    
+    The sandbox supports most Python standard library modules and can install additional packages.
+    Note: File access and some system operations are restricted for security.
+    """
+    args_schema: Type[BaseModel] = sandbox_tool_input
+    stateful: bool = True
+    allow_net: bool = True
+    session_bytes: Optional[bytes] = None
+    session_metadata: Optional[Dict] = None
+    
+    def __init__(self, **kwargs: Any) -> None:
+        super().__init__(**kwargs)
+        self._sandbox = None
+        self._initialize_sandbox()
+    
+    def _initialize_sandbox(self) -> None:
+        """Initialize the PyodideSandbox instance"""
+        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."
+                )
+                logger.error(error_msg)
+                raise RuntimeError(error_msg)
+            
+            from langchain_sandbox import PyodideSandbox
+            self._sandbox = PyodideSandbox(
+                stateful=self.stateful,
+                allow_net=self.allow_net
+            )
+            logger.info("PyodideSandbox initialized successfully")
+        except ImportError as e:
+            if "langchain_sandbox" in str(e):
+                error_msg = (
+                    "langchain-sandbox is required for the PyodideSandboxTool. "
+                    "Please install it with: pip install langchain-sandbox"
+                )
+                logger.error(error_msg)
+                raise ImportError(error_msg) from e
+            else:
+                logger.error(f"Failed to import required module: {e}")
+                raise
+        except Exception as e:
+            logger.error(f"Failed to initialize PyodideSandbox: {e}")
+            raise
+    
+    def _run(self, code: str) -> str:
+        """
+        Synchronous version - runs the async method in a new event loop
+        """
+        try:
+            # Check if sandbox is initialized, if not try to initialize
+            if self._sandbox is None:
+                self._initialize_sandbox()
+            
+            # Check if we're already in an async context
+            try:
+                loop = asyncio.get_running_loop()
+                # We're in an async context, but _run is supposed to be sync
+                # We'll need to use a different approach
+                import concurrent.futures
+                with concurrent.futures.ThreadPoolExecutor() as executor:
+                    future = executor.submit(asyncio.run, self._arun(code))
+                    return future.result()
+            except RuntimeError:
+                # No running loop, safe to use asyncio.run
+                return asyncio.run(self._arun(code))
+        except (ImportError, RuntimeError) as e:
+            # Handle specific dependency errors gracefully
+            error_msg = str(e)
+            if "langchain-sandbox" in error_msg:
+                return "❌ PyodideSandboxTool requires langchain-sandbox. Install with: pip install langchain-sandbox"
+            elif "Deno" in error_msg:
+                return "❌ PyodideSandboxTool requires Deno. Install from: https://docs.deno.com/runtime/getting_started/installation/"
+            else:
+                return f"❌ PyodideSandboxTool initialization failed: {error_msg}"
+        except Exception as e:
+            logger.error(f"Error executing code in sandbox: {e}")
+            return f"Error executing code: {str(e)}"
+    
+    async def _arun(self, code: str) -> str:
+        """
+        Execute Python code in the Pyodide sandbox
+        """
+        try:
+            if self._sandbox is None:
+                self._initialize_sandbox()
+            
+            # Execute the code with session state if available
+            result = await self._sandbox.execute(
+                code,
+                session_bytes=self.session_bytes,
+                session_metadata=self.session_metadata
+            )
+            
+            # Update session state for stateful execution
+            if self.stateful:
+                self.session_bytes = result.session_bytes
+                self.session_metadata = result.session_metadata
+            
+            # Format the output
+            output_parts = []
+            
+            if result.result is not None:
+                output_parts.append(f"Result: {result.result}")
+            
+            if result.stdout:
+                output_parts.append(f"Output: {result.stdout}")
+            
+            if result.stderr:
+                output_parts.append(f"Error: {result.stderr}")
+            
+            if result.status == 'error':
+                output_parts.append(f"Execution failed with status: {result.status}")
+            
+            execution_info = f"Execution time: {result.execution_time:.2f}s"
+            if result.session_metadata and 'packages' in result.session_metadata:
+                packages = result.session_metadata.get('packages', [])
+                if packages:
+                    execution_info += f", Packages: {', '.join(packages)}"
+            
+            output_parts.append(execution_info)
+            
+            return "\n".join(output_parts) if output_parts else "Code executed successfully (no output)"
+            
+        except Exception as e:
+            logger.error(f"Error executing code in sandbox: {e}")
+            return f"Error executing code: {str(e)}"
+
+
+class StatefulPyodideSandboxTool(PyodideSandboxTool):
+    """
+    A stateful version of the PyodideSandboxTool that maintains state between executions.
+    This version preserves variables, imports, and function definitions across multiple tool calls.
+    """
+    
+    name: str = "stateful_pyodide_sandbox"
+    description: str = """Execute Python code in a stateful sandbox environment using Pyodide.
+    This tool maintains state between executions, preserving variables, imports, and function definitions.
+    Use this tool when you need to:
+    - Build upon previous code executions
+    - Maintain variables across multiple calls
+    - Develop complex programs step by step
+    - Preserve imported libraries and defined functions
+    
+    The sandbox supports most Python standard library modules and can install additional packages.
+    Note: File access and some system operations are restricted for security.
+    """
+    
+    def __init__(self, **kwargs: Any) -> None:
+        kwargs['stateful'] = True  # Force stateful mode
+        super().__init__(**kwargs)
+
+
+# Factory function for creating sandbox tools
+def create_sandbox_tool(stateful: bool = False, allow_net: bool = True) -> BaseTool:
+    """
+    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.
+    
+    Args:
+        stateful: Whether to maintain state between executions
+        allow_net: Whether to allow network access (for package installation)
+    
+    Returns:
+        Configured sandbox tool instance
+        
+    Raises:
+        ImportError: If langchain-sandbox is not installed
+        RuntimeError: If Deno is not found in PATH
+    """
+    if stateful:
+        return StatefulPyodideSandboxTool(allow_net=allow_net)
+    else:
+        return PyodideSandboxTool(stateful=False, allow_net=allow_net)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.335
+Version: 0.3.337
 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
@@ -25,15 +25,17 @@ Requires-Dist: httpcore==1.0.7
 Requires-Dist: urllib3>=2
 Requires-Dist: certifi==2024.8.30
 Provides-Extra: runtime
-Requires-Dist: langchain_core==0.3.49; extra == "runtime"
-Requires-Dist: langchain<=0.3.22; extra == "runtime"
-Requires-Dist: langchain_community~=0.3.7; extra == "runtime"
-Requires-Dist: langchain-openai~=0.3.0; extra == "runtime"
+Requires-Dist: langchain_core<0.4.0,>=0.3.76; extra == "runtime"
+Requires-Dist: langchain<0.4.0,>=0.3.22; extra == "runtime"
+Requires-Dist: langchain_community<0.4.0,>=0.3.7; extra == "runtime"
+Requires-Dist: langchain-openai<0.4.0,>=0.3.0; extra == "runtime"
+Requires-Dist: langchain-anthropic<0.4.0,>=0.3.10; extra == "runtime"
+Requires-Dist: anthropic>=0.57.0; extra == "runtime"
 Requires-Dist: langgraph-checkpoint-sqlite~=2.0.0; extra == "runtime"
 Requires-Dist: langgraph-checkpoint-postgres==2.0.21; extra == "runtime"
 Requires-Dist: langsmith>=0.3.45; extra == "runtime"
 Requires-Dist: langgraph<0.5,>=0.4.8; extra == "runtime"
-Requires-Dist: langchain_chroma~=0.2.2; extra == "runtime"
+Requires-Dist: langchain_chroma<0.3.0,>=0.2.2; extra == "runtime"
 Requires-Dist: langchain-unstructured~=0.1.6; extra == "runtime"
 Requires-Dist: langchain-postgres~=0.0.13; extra == "runtime"
 Requires-Dist: keybert==0.8.3; extra == "runtime"
@@ -68,6 +70,7 @@ Requires-Dist: opentelemetry_instrumentation==0.46b0; extra == "runtime"
 Requires-Dist: grpcio_status==1.63.0rc1; extra == "runtime"
 Requires-Dist: protobuf==4.25.7; extra == "runtime"
 Requires-Dist: sentence-transformers==2.7.0; extra == "runtime"
+Requires-Dist: langchain-sandbox>=0.0.6; extra == "runtime"
 Provides-Extra: tools
 Requires-Dist: dulwich==0.21.6; extra == "tools"
 Requires-Dist: paramiko==3.3.1; extra == "tools"
@@ -96,8 +99,6 @@ Requires-Dist: azure-search-documents==11.5.2; extra == "tools"
 Requires-Dist: PyMySQL==1.1.1; extra == "tools"
 Requires-Dist: psycopg2-binary==2.9.10; extra == "tools"
 Requires-Dist: Office365-REST-Python-Client==2.5.14; extra == "tools"
-Requires-Dist: python-docx==1.1.2; extra == "tools"
-Requires-Dist: python-pptx==1.0.2; extra == "tools"
 Requires-Dist: pypdf2~=3.0.1; extra == "tools"
 Requires-Dist: FigmaPy==2018.1.0; extra == "tools"
 Requires-Dist: pandas==2.2.3; extra == "tools"
@@ -114,7 +115,6 @@ Requires-Dist: playwright>=1.52.0; extra == "tools"
 Requires-Dist: google-api-python-client==2.154.0; extra == "tools"
 Requires-Dist: wikipedia==1.4.0; extra == "tools"
 Requires-Dist: lxml==5.2.2; extra == "tools"
-Requires-Dist: beautifulsoup4; extra == "tools"
 Requires-Dist: pymupdf==1.24.9; extra == "tools"
 Requires-Dist: googlemaps==4.10.0; extra == "tools"
 Requires-Dist: yagmail==0.15.293; extra == "tools"

{alita_sdk-0.3.335.dist-info → alita_sdk-0.3.337.dist-info}/RECORD

@@ -40,7 +40,7 @@ alita_sdk/runtime/clients/client.py,sha256=q9dRNrwDQJmb3dlFa6iWp-rB-XsgNLJh7gtrF
 alita_sdk/runtime/clients/datasource.py,sha256=HAZovoQN9jBg0_-lIlGBQzb4FJdczPhkHehAiVG3Wx0,1020
 alita_sdk/runtime/clients/prompt.py,sha256=li1RG9eBwgNK_Qf0qUaZ8QNTmsncFrAL2pv3kbxZRZg,1447
 alita_sdk/runtime/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/runtime/langchain/assistant.py,sha256=HuAlm5XLBEu-JZmyC2WBMWdwTa74LY-d-Gu0dCbB1nU,13727
+alita_sdk/runtime/langchain/assistant.py,sha256=ljmDSxgpbextmmK6PuA77eunFPqz5-OSrcsnKwbTu2c,14622
 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
@@ -103,7 +103,7 @@ alita_sdk/runtime/toolkits/prompt.py,sha256=WIpTkkVYWqIqOWR_LlSWz3ug8uO9tm5jJ7aZ
 alita_sdk/runtime/toolkits/subgraph.py,sha256=wwUK8JjPXkGzyVZ3tAukmvST6eGbqx_U11rpnmbrvtg,2105
 alita_sdk/runtime/toolkits/tools.py,sha256=YkDvbN1TdWNO-wHbbJbNeonGCwCPlxsuegH33mCfE-I,8303
 alita_sdk/runtime/toolkits/vectorstore.py,sha256=BGppQADa1ZiLO17fC0uCACTTEvPHlodEDYEzUcBRbAA,2901
-alita_sdk/runtime/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+alita_sdk/runtime/tools/__init__.py,sha256=7OA8YPKlEOfXu3-gJA08cyR-VymjSPL-OmbXI-B2xVA,355
 alita_sdk/runtime/tools/agent.py,sha256=m98QxOHwnCRTT9j18Olbb5UPS8-ZGeQaGiUyZJSyFck,3162
 alita_sdk/runtime/tools/application.py,sha256=j6H2OFteb7h2Yhxk1ipA-i5g3BvfmTfPuMxbfLPEa0E,2891
 alita_sdk/runtime/tools/artifact.py,sha256=yIn-kfI9OWoaxbBeqdqF0M1HPeMtNnvZ_pCPoUIwnCk,8708
@@ -119,6 +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/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
@@ -349,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.335.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.335.dist-info/METADATA,sha256=IgNDzh7u2L6Ra07WYt4VymAQJAIt-3gYmENtFvlxL6w,18835
-alita_sdk-0.3.335.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.335.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.335.dist-info/RECORD,,
+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,,