@mastra/e2b 0.5.1 → 0.6.0

package/CHANGELOG.md

@@ -1,5 +1,45 @@
 # @mastra/e2b
 
+## 0.6.0
+
+### Minor Changes
+
+- Add `E2BCodeModeTransport` for running Code Mode in an `E2BSandbox`. ([#19372](https://github.com/mastra-ai/mastra/pull/19372))
+
+  The default `StdioCodeModeTransport` writes the runner/program to the host tmpdir and spawns `node <hostPath>`, which only works when the sandbox shares the host filesystem (e.g. `LocalSandbox`). E2B runs the program in a remote micro-VM, so those host paths don't exist there. `E2BCodeModeTransport` writes the program and runner into the sandbox via the E2B files API, strips TypeScript on the host with esbuild (no `--experimental-strip-types`, no Node-version dependency), auto-starts the sandbox when needed, surfaces captured stderr in timeout/no-result errors, and cleans up the sandbox directory afterwards.
+
+  ```ts
+  import { createCodeMode } from '@mastra/core/tools';
+  import { E2BSandbox, E2BCodeModeTransport } from '@mastra/e2b';
+
+  const { tool, instructions } = createCodeMode({ tools, sandbox: new E2BSandbox() }, new E2BCodeModeTransport());
+  ```
+
+### Patch Changes
+
+- Updated dependencies [[`bd6d240`](https://github.com/mastra-ai/mastra/commit/bd6d2402db93dddaef0721667e7e8a030e7c6e16), [`0111486`](https://github.com/mastra-ai/mastra/commit/01114867612593eef5cfa2fda6a1194dfedda841), [`96a3749`](https://github.com/mastra-ai/mastra/commit/96a37492235f5b8076b3e3177d83ed5a5e44a640), [`fe1bda0`](https://github.com/mastra-ai/mastra/commit/fe1bda06f6af92a694a51712db747cda1e7185f0), [`25e7c12`](https://github.com/mastra-ai/mastra/commit/25e7c126a770069ae7fb7ecf1d2adb40e017b009), [`1ce5121`](https://github.com/mastra-ai/mastra/commit/1ce512155d122bb21f47d98383e82ffbf84b39e8), [`fb8aea3`](https://github.com/mastra-ai/mastra/commit/fb8aea384291e77311be3a64ee1717320d5c3c73), [`4adc391`](https://github.com/mastra-ai/mastra/commit/4adc3911075249c352bb4832d2471922826344de), [`a5c6337`](https://github.com/mastra-ai/mastra/commit/a5c6337d23c7686c81a32ce62f550f610543a240), [`3cfc47a`](https://github.com/mastra-ai/mastra/commit/3cfc47a6b89940aadd0f46fb01ae9624a73a865d), [`2bb7817`](https://github.com/mastra-ai/mastra/commit/2bb78176112fde628483de2830528f7eee911e56), [`51d9870`](https://github.com/mastra-ai/mastra/commit/51d987032c689c2855374d0f244f5d654da809d1), [`5cab274`](https://github.com/mastra-ai/mastra/commit/5cab2744250e22d12fefa7b32637dce224233cee), [`7fa27d3`](https://github.com/mastra-ai/mastra/commit/7fa27d3b6f5ed68cd34e454a4d3ad9c482a0cfbc), [`8b97958`](https://github.com/mastra-ai/mastra/commit/8b979589f9aa59ba67cac565949475f2ffeb4ac3), [`8410541`](https://github.com/mastra-ai/mastra/commit/84105412c60ecd3bb33a9838146f59c4b588228f), [`a58dcbb`](https://github.com/mastra-ai/mastra/commit/a58dcbb546d7e1d65ebdc1f39e55f0908fcd9391), [`aa38805`](https://github.com/mastra-ai/mastra/commit/aa38805b878b827403be785eb90688d7172f5a40), [`153bd3b`](https://github.com/mastra-ai/mastra/commit/153bd3b396bdfed6b74cf43de12db8fd2d83c04a), [`45a8e65`](https://github.com/mastra-ai/mastra/commit/45a8e65e1556d1362cb3f25187023c36de26661d), [`e955965`](https://github.com/mastra-ai/mastra/commit/e955965dce575a903e37cf054d28ea99aa48785e), [`2d22570`](https://github.com/mastra-ai/mastra/commit/2d22570c7dfdd02123d0ecc529efb05ccba2d9fc), [`07bb863`](https://github.com/mastra-ai/mastra/commit/07bb8631919c6f7cf377dccd45b096e0f17fbed0), [`c8ed116`](https://github.com/mastra-ai/mastra/commit/c8ed11699f62bcac70102ab4ec84d80d20541da6), [`01b338c`](https://github.com/mastra-ai/mastra/commit/01b338c56271f0219606710e3e8b26dee27ac6c2), [`a99eae8`](https://github.com/mastra-ai/mastra/commit/a99eae8908e500c1b2d12f9d277be616b98617a5), [`860ef7e`](https://github.com/mastra-ai/mastra/commit/860ef7e77d92b63469cbe5857aa1e626197e43e9), [`17e818c`](https://github.com/mastra-ai/mastra/commit/17e818c51a958ba90641b1a959dc38faf8c034e9), [`edce8d2`](https://github.com/mastra-ai/mastra/commit/edce8d2769f19e27a05737c627af2d765472a4f8), [`8a586ec`](https://github.com/mastra-ai/mastra/commit/8a586eca9a4914f31dff6140d0d45ac375b00669), [`4451dfe`](https://github.com/mastra-ai/mastra/commit/4451dfe857428e7abcc0261a507a2e186dae6d47), [`8b7361d`](https://github.com/mastra-ai/mastra/commit/8b7361d35de68b80d05d30a74e0c69e7218fd612), [`1d39058`](https://github.com/mastra-ai/mastra/commit/1d39058e548efd691799985d5c8af2737f1c3bd2), [`3927473`](https://github.com/mastra-ai/mastra/commit/392747323ddb10c643d12be7b9ae913159dfaeed), [`dce50dc`](https://github.com/mastra-ai/mastra/commit/dce50dc9a1c1fcd0f427bb5f6250ec74910cb04b), [`fd13f8e`](https://github.com/mastra-ai/mastra/commit/fd13f8e21990f9904c3eedba3a626bb4a929cdb8), [`634caff`](https://github.com/mastra-ai/mastra/commit/634caff29a9200ad058b67d53f96d9e5832fb8a2), [`f703f87`](https://github.com/mastra-ai/mastra/commit/f703f878de072d51fda557f9c50867d8252bef05), [`3e26c87`](https://github.com/mastra-ai/mastra/commit/3e26c87de0c5bc2583b795ce6ca5889b6b161acb), [`33f2b88`](https://github.com/mastra-ai/mastra/commit/33f2b88842c09a567f906fac4cb61cd5277ced59), [`177010f`](https://github.com/mastra-ai/mastra/commit/177010ff096d2e4b28d89803be5b1a4cad2a0d6b), [`0ad646f`](https://github.com/mastra-ai/mastra/commit/0ad646f71a530f2454664299e5e01bfd13fa12e5), [`b486abf`](https://github.com/mastra-ai/mastra/commit/b486abfa2a7528c6f527e4015c819ea9fa54aaad), [`54a51e0`](https://github.com/mastra-ai/mastra/commit/54a51e0a484fe1ebad3fb1f7ef5282a075709eb7), [`c43f3a9`](https://github.com/mastra-ai/mastra/commit/c43f3a9d1efde99b38789364ba4d0ba670f430e3), [`a5008f2`](https://github.com/mastra-ai/mastra/commit/a5008f22ae710ad9402ea9f2547d8c02f74d384b), [`e2d5f37`](https://github.com/mastra-ai/mastra/commit/e2d5f373bd289be534d5f8694d34465010533df6), [`4ce0163`](https://github.com/mastra-ai/mastra/commit/4ce0163dc86e675a86809685c8ce6c49f1aeb87e), [`4378341`](https://github.com/mastra-ai/mastra/commit/43783412df5ea3dd35f5b1f6e4851e79c346fc89)]:
+  - @mastra/core@1.51.0
+
+## 0.6.0-alpha.0
+
+### Minor Changes
+
+- Add `E2BCodeModeTransport` for running Code Mode in an `E2BSandbox`. ([#19372](https://github.com/mastra-ai/mastra/pull/19372))
+
+  The default `StdioCodeModeTransport` writes the runner/program to the host tmpdir and spawns `node <hostPath>`, which only works when the sandbox shares the host filesystem (e.g. `LocalSandbox`). E2B runs the program in a remote micro-VM, so those host paths don't exist there. `E2BCodeModeTransport` writes the program and runner into the sandbox via the E2B files API, strips TypeScript on the host with esbuild (no `--experimental-strip-types`, no Node-version dependency), auto-starts the sandbox when needed, surfaces captured stderr in timeout/no-result errors, and cleans up the sandbox directory afterwards.
+
+  ```ts
+  import { createCodeMode } from '@mastra/core/tools';
+  import { E2BSandbox, E2BCodeModeTransport } from '@mastra/e2b';
+
+  const { tool, instructions } = createCodeMode({ tools, sandbox: new E2BSandbox() }, new E2BCodeModeTransport());
+  ```
+
+### Patch Changes
+
+- Updated dependencies [[`edce8d2`](https://github.com/mastra-ai/mastra/commit/edce8d2769f19e27a05737c627af2d765472a4f8)]:
+  - @mastra/core@1.51.0-alpha.9
+
 ## 0.5.1
 
 ### Patch Changes

package/dist/code-mode/transport.d.ts

@@ -0,0 +1,42 @@
+/**
+ * Code Mode — E2B transport
+ *
+ * The default {@link StdioCodeModeTransport} in `@mastra/core` writes the
+ * runner/program files to the *host* tmpdir and spawns `node <hostPath>`. That
+ * only works when the sandbox shares the host filesystem (e.g. `LocalSandbox`).
+ * E2B runs the program in a remote micro-VM with its own filesystem, so the
+ * host paths don't exist there and `node` exits immediately.
+ *
+ * `E2BCodeModeTransport` writes the runner/program *into* the sandbox via the
+ * E2B files API and runs plain `node <runnerPath>` inside the VM. TypeScript is
+ * stripped on the host with esbuild before upload, so it doesn't depend on the
+ * sandbox's Node version (the core transport relies on
+ * `node --experimental-strip-types`, which needs Node >= 22.6).
+ *
+ * The RPC frame protocol (host <-> runner) is unchanged: it reuses
+ * `buildProgramModule`, `buildRunner`, and `FRAME_PREFIX` from
+ * `@mastra/core/tools`.
+ */
+import type { CodeModeToolResult, CodeModeTransport } from '@mastra/core/tools';
+/**
+ * Code Mode transport for {@link E2BSandbox}.
+ *
+ * Writes the generated program and runner into the sandbox filesystem, runs
+ * `node` there, and bridges `external_*` RPC calls back to the host over the
+ * process's stdout/stdin — the same frame protocol as the core stdio transport.
+ *
+ * @example
+ * ```typescript
+ * import { createCodeMode } from '@mastra/core/tools';
+ * import { E2BSandbox, E2BCodeModeTransport } from '@mastra/e2b';
+ *
+ * const { tool, instructions } = createCodeMode(
+ *   { tools: { getWeather, getForecast }, sandbox: new E2BSandbox() },
+ *   new E2BCodeModeTransport(),
+ * );
+ * ```
+ */
+export declare class E2BCodeModeTransport implements CodeModeTransport {
+    run(opts: Parameters<CodeModeTransport['run']>[0]): Promise<CodeModeToolResult>;
+}
+//# sourceMappingURL=transport.d.ts.map

package/dist/code-mode/transport.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"transport.d.ts","sourceRoot":"","sources":["../../src/code-mode/transport.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AAIH,OAAO,KAAK,EAAuB,kBAAkB,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAcrG;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,oBAAqB,YAAW,iBAAiB;IACtD,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,kBAAkB,CAAC;CAkMtF"}

package/dist/index.cjs

@@ -3,6 +3,8 @@
 var workspace = require('@mastra/core/workspace');
 var e2b = require('e2b');
 var crypto = require('crypto');
+var tools = require('@mastra/core/tools');
+var esbuild = require('esbuild');
 
 // src/sandbox/index.ts
 var MOUNTABLE_TEMPLATE_VERSION = "v1";
@@ -1214,7 +1216,169 @@ var e2bSandboxProvider = {
   },
   createSandbox: (config) => new E2BSandbox(config)
 };
+var SANDBOX_TMP = "/home/user/mastra-code-mode";
+function sanitize(id) {
+  const cleaned = id.replace(/[^A-Za-z0-9_$]/g, "_");
+  return /^[A-Za-z_$]/.test(cleaned) ? cleaned : `_${cleaned}`;
+}
+var E2BCodeModeTransport = class {
+  async run(opts) {
+    const { sandbox, program, toolIds, dispatch, timeout, abortSignal, onExternalCall, onExternalResult } = opts;
+    if (!(sandbox instanceof E2BSandbox)) {
+      throw new Error("E2BCodeModeTransport requires an E2BSandbox");
+    }
+    if (!sandbox.processes) {
+      throw new Error("Sandbox has no process manager");
+    }
+    if (sandbox.status !== "running") {
+      await sandbox.start();
+    }
+    const e2b = sandbox.e2b;
+    const externals = toolIds.map((toolId) => ({ toolId, externalName: sanitize(toolId) }));
+    const allowList = new Set(toolIds);
+    const suffix = crypto.randomBytes(4).toString("hex");
+    const dir = `${SANDBOX_TMP}/${suffix}`;
+    const programPath = `${dir}/program-${suffix}.mjs`;
+    const runnerPath = `${dir}/runner-${suffix}.mjs`;
+    const programSource = esbuild.transformSync(tools.buildProgramModule(program), { loader: "ts", target: "es2022" }).code;
+    const runnerSource = tools.buildRunner({ programModule: `file://${programPath}`, externals });
+    const logs = [];
+    let stderr = "";
+    let done;
+    let stdoutBuffer = "";
+    let resolveDone;
+    const donePromise = new Promise((resolve) => {
+      resolveDone = resolve;
+    });
+    const notifyCall = (tool, args) => {
+      try {
+        onExternalCall?.(tool, args);
+      } catch {
+      }
+    };
+    const notifyResult = (tool, durationMs, error) => {
+      try {
+        onExternalResult?.(tool, durationMs, error);
+      } catch {
+      }
+    };
+    try {
+      await e2b.files.makeDir(dir);
+      await e2b.files.write(programPath, programSource);
+      await e2b.files.write(runnerPath, runnerSource);
+      let handle;
+      const respond = async (id, ok, result, error) => {
+        await handle.sendStdin(JSON.stringify({ type: "rpc-result", id, ok, result, error }) + "\n");
+      };
+      const serveRpc = async (id, tool, args) => {
+        const started = Date.now();
+        notifyCall(tool, args);
+        if (!allowList.has(tool)) {
+          notifyResult(tool, Date.now() - started, new Error("not allowed"));
+          await respond(id, false, void 0, {
+            message: `Tool "${tool}" is not available in Code Mode`,
+            name: "NotAllowedError"
+          });
+          return;
+        }
+        try {
+          const result = await dispatch(tool, args);
+          notifyResult(tool, Date.now() - started);
+          await respond(id, true, result);
+        } catch (error) {
+          const err = error;
+          notifyResult(tool, Date.now() - started, error instanceof Error ? error : new Error(String(error)));
+          await respond(id, false, void 0, {
+            message: err?.message ?? String(error),
+            name: err?.name
+          });
+        }
+      };
+      const handleFrame = (frame) => {
+        switch (frame.type) {
+          case "log":
+            logs.push(frame.message);
+            return;
+          case "done":
+            done = frame.ok ? { success: true, result: frame.result, logs } : { success: false, error: frame.error, logs };
+            resolveDone();
+            return;
+          case "rpc":
+            void serveRpc(frame.id, frame.tool, frame.args).catch(() => {
+            });
+            return;
+        }
+      };
+      handle = await sandbox.processes.spawn(`node ${runnerPath}`, {
+        cwd: dir,
+        abortSignal,
+        // E2B failures are otherwise silent, which makes them painful to debug.
+        // Capture stderr and surface it in Timeout/NoResult errors below.
+        onStderr: (chunk) => {
+          stderr += chunk;
+        },
+        onStdout: (chunk) => {
+          stdoutBuffer += chunk;
+          let idx;
+          while ((idx = stdoutBuffer.indexOf("\n")) >= 0) {
+            const line = stdoutBuffer.slice(0, idx);
+            stdoutBuffer = stdoutBuffer.slice(idx + 1);
+            if (!line.startsWith(tools.FRAME_PREFIX)) continue;
+            let frame;
+            try {
+              frame = JSON.parse(line.slice(tools.FRAME_PREFIX.length));
+            } catch {
+              continue;
+            }
+            handleFrame(frame);
+          }
+        }
+      });
+      let timer;
+      const timeoutPromise = new Promise((resolve) => {
+        timer = setTimeout(() => resolve("timeout"), timeout);
+      });
+      const exitPromise = handle.wait().then(() => "exited");
+      const outcome = await Promise.race([
+        donePromise.then(() => "done"),
+        exitPromise.catch(() => "exited"),
+        timeoutPromise
+      ]);
+      if (timer) clearTimeout(timer);
+      if (outcome === "timeout") {
+        await handle.kill().catch(() => {
+        });
+        return {
+          success: false,
+          logs,
+          error: {
+            message: `Code Mode execution timed out after ${timeout}ms${stderr ? `
+stderr: ${stderr}` : ""}`,
+            name: "TimeoutError"
+          }
+        };
+      }
+      if (!done) {
+        await exitPromise.catch(() => {
+        });
+      }
+      return done ?? {
+        success: false,
+        logs,
+        error: {
+          message: `Program exited without returning a result${stderr ? `
+stderr: ${stderr}` : ""}`,
+          name: "NoResultError"
+        }
+      };
+    } finally {
+      await e2b.files.remove(dir).catch(() => {
+      });
+    }
+  }
+};
 
+exports.E2BCodeModeTransport = E2BCodeModeTransport;
 exports.E2BProcessManager = E2BProcessManager;
 exports.E2BSandbox = E2BSandbox;
 exports.createDefaultMountableTemplate = createDefaultMountableTemplate;