Rename CodeInterpreter to Sandbox

Author: jakubno

js/src/index.ts

@@ -1,6 +1,6 @@
 export * from 'e2b'
 
-export { CodeInterpreter, JupyterExtension } from './codeInterpreter'
+export { Sandbox, JupyterExtension } from './sandbox'
 
 export type {
   Logs,
@@ -27,6 +27,6 @@ export type {
   SuperGraph,
   PointData,
 } from './graphs'
-import { CodeInterpreter } from './codeInterpreter'
+import { Sandbox } from './sandbox'
 
-export default CodeInterpreter
+export default Sandbox

js/src/codeInterpreter.ts renamed to js/src/sandbox.ts

@@ -1,4 +1,4 @@
-import { ConnectionConfig, Sandbox, TimeoutError } from 'e2b'
+import { ConnectionConfig, Sandbox as BaseSandbox, TimeoutError } from 'e2b'
 
 import { Result, Execution, OutputMessage, parseOutput, extractError } from './messaging'
 
@@ -292,12 +292,12 @@ export class JupyterExtension {
 /**
  * Code interpreter module for executing code in a stateful context.
  */
-export class CodeInterpreter extends Sandbox {
+export class Sandbox extends BaseSandbox {
   protected static override readonly defaultTemplate: string = 'code-interpreter-beta'
   protected static readonly jupyterPort = 49999
 
   readonly notebook = new JupyterExtension(
-    `${this.connectionConfig.debug ? 'http' : 'https'}://${this.getHost(CodeInterpreter.jupyterPort)}`,
+    `${this.connectionConfig.debug ? 'http' : 'https'}://${this.getHost(Sandbox.jupyterPort)}`,
     this.connectionConfig,
   )
 }

js/tests/benchmarking.js

@@ -1,4 +1,4 @@
-const { CodeInterpreter } = require('../dist')
+const { Sandbox } = require('../dist')
 const dotenv = require('dotenv')
 dotenv.config()
 
@@ -11,7 +11,7 @@ async function main() {
   for (let i = 0; i < iterations; i++) {
     console.log('Iteration:', i + 1)
     let startTime = new Date()
-    const sandbox = await CodeInterpreter.create()
+    const sandbox = await Sandbox.create()
     createSandboxTime += new Date() - startTime
 
     startTime = new Date()

js/tests/envVars.test.ts

@@ -1,11 +1,11 @@
 import { expect } from 'vitest'
 
 import { isDebug, sandboxTest } from './setup'
-import { CodeInterpreter } from '../src'
+import { Sandbox } from '../src'
 
 // Skip this test if we are running in debug mode — the pwd and user in the testing docker container are not the same as in the actual sandbox.
 sandboxTest.skipIf(isDebug)('env vars', async () => {
-  const sandbox = await CodeInterpreter.create({
+  const sandbox = await Sandbox.create({
     envs: { TEST_ENV_VAR: 'supertest' },
   })
   const result = await sandbox.notebook.execCell(
@@ -25,7 +25,7 @@ sandboxTest('env vars on sandbox', async ({ sandbox }) => {
 })
 
 sandboxTest('env vars on sandbox override', async () => {
-  const sandbox = await CodeInterpreter.create({
+  const sandbox = await Sandbox.create({
     envs: { FOO: 'bar', SBX: 'value' },
   })
   await sandbox.notebook.execCell(

js/tests/reconnect.test.ts

@@ -1,10 +1,10 @@
 import { expect } from 'vitest'
 
-import { CodeInterpreter } from '../src'
+import { Sandbox } from '../src'
 import { sandboxTest } from './setup'
 
 sandboxTest('reconnect', async ({ sandbox }) => {
-  sandbox = await CodeInterpreter.connect(sandbox.sandboxId)
+  sandbox = await Sandbox.connect(sandbox.sandboxId)
 
   const result = await sandbox.notebook.execCell('x =1; x')
 

js/tests/runtimes/bun/run.test.ts

@@ -1,9 +1,9 @@
 import { expect, test } from 'bun:test'
 
-import { CodeInterpreter } from '../../../src'
+import { Sandbox } from '../../../src'
 
 test('Bun test', async () => {
-  const sbx = await CodeInterpreter.create({ timeoutMs: 5_000 })
+  const sbx = await Sandbox.create({ timeoutMs: 5_000 })
   try {
     const result = await sbx.notebook.execCell('print("Hello, World!")')
     expect(result.logs.stdout.join('')).toEqual('Hello, World!\n')

js/tests/runtimes/deno/run.test.ts

@@ -3,10 +3,10 @@ import { load } from 'https://deno.land/[email protected]/dotenv/mod.ts'
 
 await load({ envPath: '.env', export: true })
 
-import { CodeInterpreter } from '../../../dist/index.mjs'
+import { Sandbox } from '../../../dist/index.mjs'
 
 Deno.test('Deno test', async () => {
-  const sbx = await CodeInterpreter.create({ timeoutMs: 5_000 })
+  const sbx = await Sandbox.create({ timeoutMs: 5_000 })
   try {
     const result = await sbx.notebook.execCell('print("Hello, World!")')
     assertEquals(result.logs.stdout.join(''), 'Hello, World!\n')

js/tests/setup.ts

@@ -1,16 +1,16 @@
-import { CodeInterpreter } from '../src'
+import { Sandbox } from '../src'
 import { test as base } from 'vitest'
 
 const timeoutMs = 60_000
 
 interface SandboxFixture {
-  sandbox: CodeInterpreter
+  sandbox: Sandbox
 }
 
 export const sandboxTest = base.extend<SandboxFixture>({
   sandbox: [
     async ({}, use) => {
-      const sandbox = await CodeInterpreter.create({ timeoutMs })
+      const sandbox = await Sandbox.create({ timeoutMs })
       try {
         await use(sandbox)
       } finally {

python/e2b_code_interpreter/__init__.py

@@ -1,6 +1,6 @@
 from e2b import *
-from .code_interpreter_sync import CodeInterpreter
-from .code_interpreter_async import AsyncCodeInterpreter
+from .code_interpreter_sync import Sandbox
+from .code_interpreter_async import AsyncSandbox
 from .models import (
     Execution,
     ExecutionError,

python/e2b_code_interpreter/code_interpreter_async.py

@@ -4,7 +4,7 @@
 from typing import Optional, List, Dict
 from httpx import AsyncHTTPTransport, AsyncClient
 
-from e2b import AsyncSandbox, ConnectionConfig
+from e2b import AsyncSandbox as BaseAsyncSandbox, ConnectionConfig
 
 from e2b_code_interpreter.constants import (
     DEFAULT_KERNEL_ID,
@@ -236,7 +236,7 @@ async def list_kernels(
             raise format_request_timeout_error()
 
 
-class AsyncCodeInterpreter(AsyncSandbox):
+class AsyncSandbox(BaseAsyncSandbox):
     default_template = DEFAULT_TEMPLATE
     _jupyter_port = JUPYTER_PORT