Auto-retry on stale E2B sandbox — recreate and re-execute

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

sandbox_e2b.py

@@ -52,6 +52,32 @@ def close_sandbox() -> None:
         _active_sandbox = None
 
 
+def _execute(sandbox: Sandbox, code: str, timeout: int) -> dict:
+    execution = sandbox.run_code(code, timeout=timeout)
+
+    stdout = "\n".join(
+        line if isinstance(line, str) else line.text
+        for line in execution.logs.stdout
+    )
+    stderr = "\n".join(
+        line if isinstance(line, str) else line.text
+        for line in execution.logs.stderr
+    )
+
+    if execution.error:
+        stderr += f"\n{execution.error.name}: {execution.error.value}"
+        exit_code = 1
+    else:
+        exit_code = 0
+
+    return {
+        "stdout": stdout,
+        "stderr": stderr,
+        "exit_code": exit_code,
+        "timed_out": False,
+    }
+
+
 def run_python(
     code: str,
     *,
@@ -63,33 +89,26 @@ def run_python(
     sandbox = get_or_create_sandbox(workspace, scratch_dir)
 
     try:
-        execution = sandbox.run_code(code, timeout=timeout)
-
-        stdout = "\n".join(
-            line if isinstance(line, str) else line.text
-            for line in execution.logs.stdout
-        )
-        stderr = "\n".join(
-            line if isinstance(line, str) else line.text
-            for line in execution.logs.stderr
-        )
-
-        if execution.error:
-            stderr += f"\n{execution.error.name}: {execution.error.value}"
-            exit_code = 1
-        else:
-            exit_code = 0
-
-        return {
-            "stdout": stdout,
-            "stderr": stderr,
-            "exit_code": exit_code,
-            "timed_out": False,
-        }
-    except TimeoutError:
-        return {
-            "stdout": "",
-            "stderr": "Execution timed out.",
-            "exit_code": -1,
-            "timed_out": True,
-        }
+        return _execute(sandbox, code, timeout)
+    except Exception as exc:
+        if "sandbox" in str(exc).lower() and "not found" in str(exc).lower():
+            # Stale sandbox — recreate and retry
+            close_sandbox()
+            sandbox = get_or_create_sandbox(workspace, scratch_dir)
+            try:
+                return _execute(sandbox, code, timeout)
+            except TimeoutError:
+                return {
+                    "stdout": "",
+                    "stderr": "Execution timed out.",
+                    "exit_code": -1,
+                    "timed_out": True,
+                }
+        if isinstance(exc, TimeoutError):
+            return {
+                "stdout": "",
+                "stderr": "Execution timed out.",
+                "exit_code": -1,
+                "timed_out": True,
+            }
+        raise