fix

sandbox/code_runner.py

@@ -10,7 +10,6 @@ import time
 import gradio as gr
 import re
 import base64
-from e2b_code_interpreter import Sandbox as CodeSandbox
 from gradio_sandboxcomponent import SandboxComponent
 
 from sandbox.sandbox_state import ChatbotSandboxState
@@ -377,7 +376,7 @@ def run_code_interpreter(code: str, code_language: str | None, install_command:
     Args:
         code (str): The code to be executed.
     """
-    sandbox = CodeSandbox()
+    sandbox = create_sandbox(is_code_sandbox=True)
 
     stderrs = []
 
@@ -393,8 +392,14 @@ def run_code_interpreter(code: str, code_language: str | None, install_command:
         if not is_success:
             print(f"Install command failed: {stderr}")
 
+    # Add virtual environment to Python path and run user code
     execution = sandbox.run_code(
-        code=code,
+        code=f"""
+import sys, glob
+for path in glob.glob("/home/user/.venv/lib/python*/site-packages"):
+    if path not in sys.path: sys.path.insert(0, path)
+
+{code}""",
         language=code_language
     )
 

sandbox/sandbox_manager.py

@@ -5,6 +5,7 @@ Facades for interacting with the e2b sandbox.
 import json
 from typing import Literal
 from e2b import Sandbox
+from e2b_code_interpreter import Sandbox as CodeSandbox
 from e2b.sandbox.commands.command_handle import CommandExitException
 from e2b.exceptions import TimeoutException
 import time
@@ -15,14 +16,20 @@ import queue
 from .constants import SANDBOX_TEMPLATE_ID, SANDBOX_NGINX_PORT, SANDBOX_RETRY_COUNT, SANDBOX_TIMEOUT_SECONDS, INSTALLED_PYPI_PACKAGES
 
 
-def create_sandbox(template: str = SANDBOX_TEMPLATE_ID) -> Sandbox:
+def create_sandbox(template: str = SANDBOX_TEMPLATE_ID, is_code_sandbox: bool = False) -> Sandbox:
     '''
     Create a new sandbox.
     Will retry if the sandbox creation fails.
     '''
     for attempt in range(1, SANDBOX_RETRY_COUNT + 1):
         try:
-            return Sandbox(
+            if is_code_sandbox:
+                return CodeSandbox.create(
+                    domain="e2b-foxtrot.dev",
+                    template=template,
+                    timeout=SANDBOX_TIMEOUT_SECONDS,
+                )
+            return Sandbox.create(
                 domain="e2b-foxtrot.dev",
                 template=template,
                 timeout=SANDBOX_TIMEOUT_SECONDS,