e2b-haystack 1.0.0__tar.gz → 1.1.0__tar.gz

e2b_haystack-1.1.0/CHANGELOG.md

@@ -0,0 +1,9 @@
+# Changelog
+
+## [integrations/e2b-v1.0.0] - 2026-04-30
+
+### 🚀 Features
+
+- Add e2b-haystack integration for E2B cloud sandbox tools (#3195)
+
+<!-- generated by git-cliff -->

{e2b_haystack-1.0.0 → e2b_haystack-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: e2b-haystack
-Version: 1.0.0
+Version: 1.1.0
 Summary: Haystack integration for E2B cloud sandboxes
 Project-URL: Documentation, https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/e2b#readme
 Project-URL: Issues, https://github.com/deepset-ai/haystack-core-integrations/issues
@@ -29,6 +29,7 @@ Description-Content-Type: text/markdown
 [![PyPI - Version](https://img.shields.io/pypi/v/e2b-haystack.svg)](https://pypi.org/project/e2b-haystack)
 [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/e2b-haystack.svg)](https://pypi.org/project/e2b-haystack)
 
+- [Integration page](https://haystack.deepset.ai/integrations/e2b)
 - [Changelog](https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/e2b/CHANGELOG.md)
 
 ---
@@ -37,4 +38,4 @@ Description-Content-Type: text/markdown
 
 Refer to the general [Contribution Guidelines](https://github.com/deepset-ai/haystack-core-integrations/blob/main/CONTRIBUTING.md).
 
-To run integration tests locally, set the `E2B_API_KEY` environment variable (get one at <https://e2b.dev/>).
+To run integration tests locally, set the `E2B_API_KEY` environment variable (get one at <https://e2b.dev/>).

{e2b_haystack-1.0.0 → e2b_haystack-1.1.0}/README.md

@@ -3,6 +3,7 @@
 [![PyPI - Version](https://img.shields.io/pypi/v/e2b-haystack.svg)](https://pypi.org/project/e2b-haystack)
 [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/e2b-haystack.svg)](https://pypi.org/project/e2b-haystack)
 
+- [Integration page](https://haystack.deepset.ai/integrations/e2b)
 - [Changelog](https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/e2b/CHANGELOG.md)
 
 ---
@@ -11,4 +12,4 @@
 
 Refer to the general [Contribution Guidelines](https://github.com/deepset-ai/haystack-core-integrations/blob/main/CONTRIBUTING.md).
 
-To run integration tests locally, set the `E2B_API_KEY` environment variable (get one at <https://e2b.dev/>).
+To run integration tests locally, set the `E2B_API_KEY` environment variable (get one at <https://e2b.dev/>).

{e2b_haystack-1.0.0 → e2b_haystack-1.1.0}/pyproject.toml

@@ -131,6 +131,7 @@ ignore = [
     "PLR0912",
     "PLR0913",
     "PLR0915",
+    "PLR0917",
     # Allow `Any` type - used legitimately for dynamic types and SDK boundaries
     "ANN401",
 ]

{e2b_haystack-1.0.0 → e2b_haystack-1.1.0}/src/haystack_integrations/tools/e2b/bash_tool.py

@@ -80,6 +80,14 @@ class RunBashCommandTool(Tool):
         )
         self._e2b_sandbox = sandbox
 
+    def warm_up(self) -> None:
+        """Start the sandbox if needed."""
+        self._e2b_sandbox.warm_up()
+
+    def close(self) -> None:
+        """Close the sandbox."""
+        self._e2b_sandbox.close()
+
     def to_dict(self) -> dict[str, Any]:
         """Serialize this tool to a dictionary."""
         return {

{e2b_haystack-1.0.0 → e2b_haystack-1.1.0}/src/haystack_integrations/tools/e2b/e2b_sandbox.py

@@ -8,11 +8,9 @@ from typing import Any, ClassVar
 
 from haystack import logging
 from haystack.core.serialization import generate_qualified_class_name
-from haystack.lazy_imports import LazyImport
 from haystack.utils import Secret, deserialize_secrets_inplace
 
-with LazyImport(message="Run 'pip install e2b'") as e2b_import:
-    from e2b import Sandbox
+from e2b import Sandbox
 
 logger = logging.getLogger(__name__)
 
@@ -52,15 +50,6 @@ class E2BSandbox:
         ],
     )
     ```
-
-    Lifecycle is handled automatically by the Agent's pipeline. If you use the
-    tools standalone, call :meth:`warm_up` before the first tool invocation:
-
-    ```python
-    sandbox.warm_up()
-    # ... use tools ...
-    sandbox.close()
-    ```
     """
 
     # Process-wide cache used during deserialization to keep tools that
@@ -115,7 +104,6 @@ class E2BSandbox:
         if self._sandbox is not None:
             return
 
-        e2b_import.check()
         resolved_key = self.api_key.resolve_value()
         try:
             logger.info(

{e2b_haystack-1.0.0 → e2b_haystack-1.1.0}/src/haystack_integrations/tools/e2b/list_directory_tool.py

@@ -67,6 +67,14 @@ class ListDirectoryTool(Tool):
         )
         self._e2b_sandbox = sandbox
 
+    def warm_up(self) -> None:
+        """Start the sandbox if needed."""
+        self._e2b_sandbox.warm_up()
+
+    def close(self) -> None:
+        """Close the sandbox."""
+        self._e2b_sandbox.close()
+
     def to_dict(self) -> dict[str, Any]:
         """Serialize this tool to a dictionary."""
         return {

{e2b_haystack-1.0.0 → e2b_haystack-1.1.0}/src/haystack_integrations/tools/e2b/read_file_tool.py

@@ -62,6 +62,14 @@ class ReadFileTool(Tool):
         )
         self._e2b_sandbox = sandbox
 
+    def warm_up(self) -> None:
+        """Start the sandbox if needed."""
+        self._e2b_sandbox.warm_up()
+
+    def close(self) -> None:
+        """Close the sandbox."""
+        self._e2b_sandbox.close()
+
     def to_dict(self) -> dict[str, Any]:
         """Serialize this tool to a dictionary."""
         return {

{e2b_haystack-1.0.0 → e2b_haystack-1.1.0}/src/haystack_integrations/tools/e2b/write_file_tool.py

@@ -62,6 +62,14 @@ class WriteFileTool(Tool):
         )
         self._e2b_sandbox = sandbox
 
+    def warm_up(self) -> None:
+        """Start the sandbox if needed."""
+        self._e2b_sandbox.warm_up()
+
+    def close(self) -> None:
+        """Close the sandbox."""
+        self._e2b_sandbox.close()
+
     def to_dict(self) -> dict[str, Any]:
         """Serialize this tool to a dictionary."""
         return {

{e2b_haystack-1.0.0 → e2b_haystack-1.1.0}/tests/test_integration.py

@@ -6,9 +6,13 @@
 Integration tests for the E2B sandbox tools.
 
 These tests require a valid E2B_API_KEY environment variable and will
-spin up a real cloud sandbox on each run.
+spin up a real cloud sandbox on each run. Every test in this module skips when
+the variable is unset, so the suite can always be invoked — including on fork
+PRs, which have no access to repository secrets.
 """
 
+import os
+
 import pytest
 
 from haystack_integrations.tools.e2b import (
@@ -20,6 +24,8 @@ from haystack_integrations.tools.e2b import (
     WriteFileTool,
 )
 
+pytestmark = pytest.mark.skipif(not os.environ.get("E2B_API_KEY"), reason="E2B_API_KEY not set")
+
 
 @pytest.fixture(scope="module")
 def sandbox():

{e2b_haystack-1.0.0 → e2b_haystack-1.1.0}/tests/test_sandbox_toolset.py

@@ -73,10 +73,8 @@ class TestE2BSandboxInit:
 
 
 class TestE2BSandboxWarmUp:
-    @patch("haystack_integrations.tools.e2b.e2b_sandbox.e2b_import")
     @patch("haystack_integrations.tools.e2b.e2b_sandbox.Sandbox.create")
-    def test_warm_up_creates_sandbox(self, mock_sandbox_create, mock_e2b_import):
-        mock_e2b_import.check.return_value = None
+    def test_warm_up_creates_sandbox(self, mock_sandbox_create):
         mock_instance = _make_sandbox_mock()
         mock_sandbox_create.return_value = mock_instance
 
@@ -91,10 +89,8 @@ class TestE2BSandboxWarmUp:
         )
         assert sb._sandbox is mock_instance
 
-    @patch("haystack_integrations.tools.e2b.e2b_sandbox.e2b_import")
     @patch("haystack_integrations.tools.e2b.e2b_sandbox.Sandbox.create")
-    def test_warm_up_passes_environment_vars(self, mock_sandbox_create, mock_e2b_import):
-        mock_e2b_import.check.return_value = None
+    def test_warm_up_passes_environment_vars(self, mock_sandbox_create):
         mock_sandbox_create.return_value = _make_sandbox_mock()
 
         sb = _make_sandbox(environment_vars={"MY_VAR": "value"})
@@ -103,10 +99,8 @@ class TestE2BSandboxWarmUp:
         _, kwargs = mock_sandbox_create.call_args
         assert kwargs["envs"] == {"MY_VAR": "value"}
 
-    @patch("haystack_integrations.tools.e2b.e2b_sandbox.e2b_import")
     @patch("haystack_integrations.tools.e2b.e2b_sandbox.Sandbox.create")
-    def test_warm_up_is_idempotent(self, mock_sandbox_create, mock_e2b_import):
-        mock_e2b_import.check.return_value = None
+    def test_warm_up_is_idempotent(self, mock_sandbox_create):
         mock_sandbox_create.return_value = _make_sandbox_mock()
 
         sb = _make_sandbox()
@@ -115,10 +109,8 @@ class TestE2BSandboxWarmUp:
 
         mock_sandbox_create.assert_called_once()
 
-    @patch("haystack_integrations.tools.e2b.e2b_sandbox.e2b_import")
     @patch("haystack_integrations.tools.e2b.e2b_sandbox.Sandbox.create")
-    def test_warm_up_raises_on_sandbox_error(self, mock_sandbox_create, mock_e2b_import):
-        mock_e2b_import.check.return_value = None
+    def test_warm_up_raises_on_sandbox_error(self, mock_sandbox_create):
         mock_sandbox_create.side_effect = Exception("connection refused")
 
         sb = _make_sandbox()
@@ -343,12 +335,47 @@ class TestToolClasses:
         ts = E2BToolset()
         assert ts.sandbox.api_key is not None
 
-    def test_tools_from_same_sandbox_share_state(self):
-        """Tools instantiated with the same sandbox share state."""
-        sb = _make_sandbox()
-        bash_tool = RunBashCommandTool(sandbox=sb)
-        read_tool = ReadFileTool(sandbox=sb)
-        assert bash_tool._e2b_sandbox is read_tool._e2b_sandbox
+    @pytest.mark.parametrize("tool_class", [RunBashCommandTool, ReadFileTool, WriteFileTool, ListDirectoryTool])
+    def test_warm_up_and_close(self, tool_class):
+        sandbox = MagicMock(spec=E2BSandbox)
+        tool = tool_class(sandbox=sandbox)
+
+        tool.warm_up()
+        sandbox.warm_up.assert_called_once_with()
+        tool.close()
+        sandbox.close.assert_called_once_with()
+
+    @patch("haystack_integrations.tools.e2b.e2b_sandbox.Sandbox.create")
+    def test_tools_share_sandbox_lifecycle(self, create):
+        first, second = _make_sandbox_mock(), _make_sandbox_mock()
+        create.side_effect = [first, second]
+        sandbox = _make_sandbox()
+        tools = [ReadFileTool(sandbox=sandbox), ListDirectoryTool(sandbox=sandbox)]
+
+        for tool in tools:
+            tool.close()
+        create.assert_not_called()
+
+        tools[1].invoke(path="/")
+        first.files.list.assert_called_once_with("/")
+        for tool in tools:
+            tool.warm_up()
+        create.assert_called_once()
+
+        for tool in tools:
+            tool.close()
+        first.kill.assert_called_once_with()
+        assert sandbox._sandbox is None
+
+        for tool in tools:
+            tool.warm_up()
+        assert create.call_count == 2
+        assert sandbox._sandbox is second
+
+        for tool in tools:
+            tool.close()
+        first.kill.assert_called_once_with()
+        second.kill.assert_called_once_with()
 
 
 # ---------------------------------------------------------------------------
@@ -378,10 +405,8 @@ class TestRunBashCommandTool:
 
         mock.commands.run.assert_called_once_with("sleep 5", timeout=30)
 
-    @patch("haystack_integrations.tools.e2b.e2b_sandbox.e2b_import")
     @patch("haystack_integrations.tools.e2b.e2b_sandbox.Sandbox.create")
-    def test_wraps_warm_up_failure(self, mock_sandbox_create, mock_e2b_import):
-        mock_e2b_import.check.return_value = None
+    def test_wraps_warm_up_failure(self, mock_sandbox_create):
         mock_sandbox_create.side_effect = Exception("connection refused")
         sb = _make_sandbox()
         tool = RunBashCommandTool(sandbox=sb)
@@ -421,10 +446,8 @@ class TestReadFileTool:
 
         assert result == "binary content"
 
-    @patch("haystack_integrations.tools.e2b.e2b_sandbox.e2b_import")
     @patch("haystack_integrations.tools.e2b.e2b_sandbox.Sandbox.create")
-    def test_wraps_warm_up_failure(self, mock_sandbox_create, mock_e2b_import):
-        mock_e2b_import.check.return_value = None
+    def test_wraps_warm_up_failure(self, mock_sandbox_create):
         mock_sandbox_create.side_effect = Exception("connection refused")
         sb = _make_sandbox()
         tool = ReadFileTool(sandbox=sb)
@@ -454,10 +477,8 @@ class TestWriteFileTool:
         assert "/output/result.txt" in result
         mock.files.write.assert_called_once_with("/output/result.txt", "hello")
 
-    @patch("haystack_integrations.tools.e2b.e2b_sandbox.e2b_import")
     @patch("haystack_integrations.tools.e2b.e2b_sandbox.Sandbox.create")
-    def test_wraps_warm_up_failure(self, mock_sandbox_create, mock_e2b_import):
-        mock_e2b_import.check.return_value = None
+    def test_wraps_warm_up_failure(self, mock_sandbox_create):
         mock_sandbox_create.side_effect = Exception("connection refused")
         sb = _make_sandbox()
         tool = WriteFileTool(sandbox=sb)
@@ -507,10 +528,8 @@ class TestListDirectoryTool:
 
         assert result == "(empty directory)"
 
-    @patch("haystack_integrations.tools.e2b.e2b_sandbox.e2b_import")
     @patch("haystack_integrations.tools.e2b.e2b_sandbox.Sandbox.create")
-    def test_wraps_warm_up_failure(self, mock_sandbox_create, mock_e2b_import):
-        mock_e2b_import.check.return_value = None
+    def test_wraps_warm_up_failure(self, mock_sandbox_create):
         mock_sandbox_create.side_effect = Exception("connection refused")
         sb = _make_sandbox()
         tool = ListDirectoryTool(sandbox=sb)