feat: add E2B sandbox support

examples/e2b_repl_example.py

@@ -0,0 +1,28 @@
+import os
+
+from dotenv import load_dotenv
+
+from rlm import RLM
+from rlm.logger import RLMLogger
+
+load_dotenv()
+
+logger = RLMLogger(log_dir="./logs")
+
+rlm = RLM(
+    backend="openai",
+    backend_kwargs={
+        "api_key": os.getenv("OPENAI_API_KEY"),
+        "model_name": "gpt-5-nano",
+    },
+    environment="e2b",
+    environment_kwargs={
+        "timeout": 300,
+    },
+    max_depth=1,
+    logger=logger,
+    verbose=True,
+)
+
+result = rlm.completion("Using your code, solve 2^(2^(2^(2))). Show your work in Python.")
+print(result.response)

pyproject.toml

@@ -36,6 +36,7 @@ Issues = "https://github.com/alexzhang13/rlm/issues"
 
 [project.optional-dependencies]
 modal = ["modal>=0.73.0", "dill>=0.3.7"]
+e2b = ["e2b-code-interpreter>=0.0.11", "dill>=0.3.7"]
 daytona = ["daytona>=0.128.1", "dill>=0.3.7"]
 prime = ["prime-sandboxes>=0.2.0", "dill>=0.3.7"]
 

rlm/core/types.py

@@ -13,7 +13,7 @@
     "azure_openai",
     "gemini",
 ]
-EnvironmentType = Literal["local", "docker", "modal", "prime", "daytona"]
+EnvironmentType = Literal["local", "docker", "modal", "prime", "daytona", "e2b"]
 
 
 def _serialize_value(value: Any) -> Any:

rlm/environments/__init__.py

@@ -7,12 +7,12 @@
 
 
 def get_environment(
-    environment: Literal["local", "modal", "docker", "daytona", "prime"],
+    environment: Literal["local", "modal", "docker", "daytona", "prime", "e2b"],
     environment_kwargs: dict[str, Any],
 ) -> BaseEnv:
     """
     Routes a specific environment and the args (as a dict) to the appropriate environment if supported.
-    Currently supported environments: ['local', 'modal', 'docker', 'daytona', 'prime']
+    Currently supported environments: ['local', 'modal', 'docker', 'daytona', 'prime', 'e2b']
     """
     if environment == "local":
         return LocalREPL(**environment_kwargs)
@@ -32,7 +32,11 @@ def get_environment(
         from rlm.environments.prime_repl import PrimeREPL
 
         return PrimeREPL(**environment_kwargs)
+    elif environment == "e2b":
+        from rlm.environments.e2b_repl import E2BREPL
+
+        return E2BREPL(**environment_kwargs)
     else:
         raise ValueError(
-            f"Unknown environment: {environment}. Supported: ['local', 'modal', 'docker', 'daytona', 'prime']"
+            f"Unknown environment: {environment}. Supported: ['local', 'modal', 'docker', 'daytona', 'prime', 'e2b']"
         )