self-bench 0.3.5 → 0.3.8

package/dist/sandbox/providers/e2b/command.js

@@ -0,0 +1,160 @@
+import { InactivityTimeoutError } from "../../../process.js";
+import { readOutputWithRetry } from "../../output-retry.js";
+import { raceWithTermination } from "./lifecycle.js";
+export const E2B_WORK_DIRECTORY = "/work";
+export async function executeE2BCommand(input) {
+    const { sandbox, request, options, signal, terminate, stdout, stderr, termination, setCommand } = input;
+    let inactivityTimer;
+    const clearInactivityTimer = () => {
+        if (inactivityTimer) {
+            clearTimeout(inactivityTimer);
+            inactivityTimer = undefined;
+        }
+    };
+    const armInactivityTimer = () => {
+        clearInactivityTimer();
+        if (!request.inactivityTimeoutMs) {
+            return;
+        }
+        inactivityTimer = setTimeout(() => {
+            terminate(new InactivityTimeoutError(`E2B sandbox ${sandbox.sandboxId} stage ${request.stage}`, request.inactivityTimeoutMs ?? 0));
+        }, request.inactivityTimeoutMs);
+        inactivityTimer.unref();
+    };
+    const capture = (stream, data) => {
+        const chunk = Buffer.from(data);
+        if (chunk.byteLength === 0) {
+            return;
+        }
+        (stream === "stdout" ? stdout : stderr).push(chunk);
+        options.onProgress?.({ stream, bytes: chunk.byteLength });
+        options.onOutput?.(stream, chunk);
+        armInactivityTimer();
+    };
+    try {
+        const command = await raceWithTermination(sandbox.commands
+            .run(shellCommand(request.command), {
+            background: true,
+            cwd: E2B_WORK_DIRECTORY,
+            envs: workloadEnvironment(request),
+            onStderr: (data) => capture("stderr", data),
+            onStdout: (data) => capture("stdout", data),
+            signal,
+            timeoutMs: request.timeoutMs,
+        })
+            .then((handle) => {
+            setCommand(handle);
+            return handle;
+        }), termination);
+        armInactivityTimer();
+        const supervision = input.startSupervision?.(sandbox);
+        let completed;
+        try {
+            completed = await raceWithTermination(command.wait(), termination);
+        }
+        catch (error) {
+            if (!isCommandResult(error)) {
+                await supervision?.finish().catch(() => undefined);
+                throw error;
+            }
+            completed = error;
+        }
+        clearInactivityTimer();
+        if (supervision) {
+            await raceWithTermination(supervision.finish(), termination);
+        }
+        captureUnstreamedCommandOutput(completed, stdout, stderr);
+        const outputs = await raceWithTermination(collectOutputs(sandbox, request, completed.exitCode, signal, input.sleep), termination);
+        return {
+            sandboxId: sandbox.sandboxId,
+            exitCode: completed.exitCode,
+            stdout: stdout.text(),
+            stderr: stderr.text(),
+            outputs,
+        };
+    }
+    finally {
+        clearInactivityTimer();
+    }
+}
+/** execute/readFile/writeFile over a running E2B sandbox. */
+export function e2bBacking(sandbox) {
+    return {
+        execute: async (command) => {
+            try {
+                const result = await sandbox.commands.run(shellCommand(command), {
+                    cwd: E2B_WORK_DIRECTORY,
+                });
+                return { exitCode: result.exitCode, stdout: result.stdout, stderr: result.stderr };
+            }
+            catch (error) {
+                if (isCommandResult(error)) {
+                    return { exitCode: error.exitCode, stdout: error.stdout, stderr: error.stderr };
+                }
+                throw error;
+            }
+        },
+        readFile: async (path) => {
+            try {
+                return await sandbox.files.read(path, { format: "bytes" });
+            }
+            catch {
+                return undefined;
+            }
+        },
+        writeFile: async (path, contents) => {
+            await sandbox.files.writeFiles([
+                {
+                    path,
+                    data: typeof contents === "string" ? contents : Uint8Array.from(contents).buffer,
+                },
+            ]);
+        },
+    };
+}
+function shellCommand(command) {
+    return command.map(shellQuote).join(" ");
+}
+function shellQuote(value) {
+    return `'${value.replaceAll("'", `'"'"'`)}'`;
+}
+function workloadEnvironment(request) {
+    const environment = { ...request.environment, ...request.secrets };
+    for (const key of Object.keys(environment)) {
+        if (key.startsWith("E2B_")) {
+            delete environment[key];
+        }
+    }
+    return environment;
+}
+async function collectOutputs(sandbox, request, exitCode, signal, sleep) {
+    const outputs = {};
+    for (const path of request.outputPaths ?? []) {
+        const { value, lastError } = await readOutputWithRetry(() => sandbox.files.read(path, { format: "bytes", signal }), { signal, ...(sleep ? { sleep: (ms) => sleep(ms, signal) } : {}) });
+        if (value !== undefined) {
+            outputs[path] = value;
+        }
+        else if (exitCode === 0) {
+            throw new Error(`sandbox ${sandbox.sandboxId} exited successfully without output ${path}`, {
+                cause: lastError,
+            });
+        }
+    }
+    return outputs;
+}
+function captureUnstreamedCommandOutput(completed, stdout, stderr) {
+    if (!stdout.text() && completed.stdout) {
+        stdout.push(Buffer.from(completed.stdout));
+    }
+    if (!stderr.text() && completed.stderr) {
+        stderr.push(Buffer.from(completed.stderr));
+    }
+}
+function isCommandResult(error) {
+    return (typeof error === "object" &&
+        error !== null &&
+        typeof error.exitCode === "number" &&
+        typeof error.stdout === "string" &&
+        typeof error.stderr === "string");
+}
+//# sourceMappingURL=command.js.map

package/dist/sandbox/providers/e2b/command.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"command.js","sourceRoot":"","sources":["../../../../src/sandbox/providers/e2b/command.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sBAAsB,EAAsB,MAAM,qBAAqB,CAAC;AAGjF,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAGrD,MAAM,CAAC,MAAM,kBAAkB,GAAG,OAAO,CAAC;AAE1C,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,KAavC;IACC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,GAC7F,KAAK,CAAC;IACR,IAAI,eAA0D,CAAC;IAC/D,MAAM,oBAAoB,GAAG,GAAS,EAAE;QACtC,IAAI,eAAe,EAAE,CAAC;YACpB,YAAY,CAAC,eAAe,CAAC,CAAC;YAC9B,eAAe,GAAG,SAAS,CAAC;QAC9B,CAAC;IACH,CAAC,CAAC;IACF,MAAM,kBAAkB,GAAG,GAAS,EAAE;QACpC,oBAAoB,EAAE,CAAC;QACvB,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QACD,eAAe,GAAG,UAAU,CAAC,GAAG,EAAE;YAChC,SAAS,CACP,IAAI,sBAAsB,CACxB,eAAe,OAAO,CAAC,SAAS,UAAU,OAAO,CAAC,KAAK,EAAE,EACzD,OAAO,CAAC,mBAAmB,IAAI,CAAC,CACjC,CACF,CAAC;QACJ,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAChC,eAAe,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC,CAAC;IACF,MAAM,OAAO,GAAG,CAAC,MAA2B,EAAE,IAAY,EAAQ,EAAE;QAClE,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChC,IAAI,KAAK,CAAC,UAAU,KAAK,CAAC,EAAE,CAAC;YAC3B,OAAO;QACT,CAAC;QACD,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,OAAO,CAAC,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;QAC1D,OAAO,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAClC,kBAAkB,EAAE,CAAC;IACvB,CAAC,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,mBAAmB,CACvC,OAAO,CAAC,QAAQ;aACb,GAAG,CAAC,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAClC,UAAU,EAAE,IAAI;YAChB,GAAG,EAAE,kBAAkB;YACvB,IAAI,EAAE,mBAAmB,CAAC,OAAO,CAAC;YAClC,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC;YAC3C,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC;YAC3C,MAAM;YACN,SAAS,EAAE,OAAO,CAAC,SAAS;SAC7B,CAAC;aACD,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACf,UAAU,CAAC,MAAM,CAAC,CAAC;YACnB,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EACJ,WAAW,CACZ,CAAC;QACF,kBAAkB,EAAE,CAAC;QACrB,MAAM,WAAW,GAAG,KAAK,CAAC,gBAAgB,EAAE,CAAC,OAAO,CAAC,CAAC;QAEtD,IAAI,SAAwB,CAAC;QAC7B,IAAI,CAAC;YACH,SAAS,GAAG,MAAM,mBAAmB,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,WAAW,CAAC,CAAC;QACrE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5B,MAAM,WAAW,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBACnD,MAAM,KAAK,CAAC;YACd,CAAC;YACD,SAAS,GAAG,KAAK,CAAC;QACpB,CAAC;QACD,oBAAoB,EAAE,CAAC;QACvB,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,mBAAmB,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,WAAW,CAAC,CAAC;QAC/D,CAAC;QACD,8BAA8B,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAE1D,MAAM,OAAO,GAAG,MAAM,mBAAmB,CACvC,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,EACzE,WAAW,CACZ,CAAC;QACF,OAAO;YACL,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,QAAQ,EAAE,SAAS,CAAC,QAAQ;YAC5B,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;YACrB,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;YACrB,OAAO;SACR,CAAC;IACJ,CAAC;YAAS,CAAC;QACT,oBAAoB,EAAE,CAAC;IACzB,CAAC;AACH,CAAC;AAED,6DAA6D;AAC7D,MAAM,UAAU,UAAU,CAAC,OAAyB;IAClD,OAAO;QACL,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACzB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE;oBAC/D,GAAG,EAAE,kBAAkB;iBACxB,CAAC,CAAC;gBACH,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;YACrF,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC3B,OAAO,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;gBAClF,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QACD,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACvB,IAAI,CAAC;gBACH,OAAO,MAAM,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7D,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;QACD,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YAClC,MAAM,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC;gBAC7B;oBACE,IAAI;oBACJ,IAAI,EAAE,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM;iBACjF;aACF,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,OAA0B;IAC9C,OAAO,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC3C,CAAC;AAED,SAAS,UAAU,CAAC,KAAa;IAC/B,OAAO,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC,GAAG,CAAC;AAC/C,CAAC;AAED,SAAS,mBAAmB,CAAC,OAAuB;IAClD,MAAM,WAAW,GAAG,EAAE,GAAG,OAAO,CAAC,WAAW,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IACnE,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;QAC3C,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,cAAc,CAC3B,OAAyB,EACzB,OAAuB,EACvB,QAAgB,EAChB,MAAmB,EACnB,KAA0D;IAE1D,MAAM,OAAO,GAA+B,EAAE,CAAC;IAC/C,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,WAAW,IAAI,EAAE,EAAE,CAAC;QAC7C,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,MAAM,mBAAmB,CACpD,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EAC3D,EAAE,MAAM,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CACnE,CAAC;QACF,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,OAAO,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC;QACxB,CAAC;aAAM,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,WAAW,OAAO,CAAC,SAAS,uCAAuC,IAAI,EAAE,EAAE;gBACzF,KAAK,EAAE,SAAS;aACjB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,8BAA8B,CACrC,SAAwB,EACxB,MAAqB,EACrB,MAAqB;IAErB,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7C,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7C,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,OAAQ,KAAgC,CAAC,QAAQ,KAAK,QAAQ;QAC9D,OAAQ,KAAgC,CAAC,MAAM,KAAK,QAAQ;QAC5D,OAAQ,KAAgC,CAAC,MAAM,KAAK,QAAQ,CAC7D,CAAC;AACJ,CAAC"}

package/dist/sandbox/providers/e2b/config.d.ts

@@ -0,0 +1,13 @@
+import type { SelfBenchWorkerConfig } from "../../../config.js";
+export type E2BExecutionConfig = Extract<SelfBenchWorkerConfig["execution"], {
+    readonly kind: "e2b";
+}>;
+export interface E2BLifecycleTimings {
+    readonly cleanupCallTimeoutMs: number;
+    readonly cleanupRecoveryDelaysMs: readonly number[];
+    readonly cleanupTimeoutMs: number;
+    readonly commandKillGraceMs: number;
+    readonly diagnosticTimeoutMs: number;
+}
+export type E2BSleep = (delayMs: number, signal: AbortSignal) => Promise<void>;
+//# sourceMappingURL=config.d.ts.map

package/dist/sandbox/providers/e2b/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../../../src/sandbox/providers/e2b/config.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAEhE,MAAM,MAAM,kBAAkB,GAAG,OAAO,CACtC,qBAAqB,CAAC,WAAW,CAAC,EAClC;IAAE,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAA;CAAE,CACzB,CAAC;AAEF,MAAM,WAAW,mBAAmB;IAClC,QAAQ,CAAC,oBAAoB,EAAE,MAAM,CAAC;IACtC,QAAQ,CAAC,uBAAuB,EAAE,SAAS,MAAM,EAAE,CAAC;IACpD,QAAQ,CAAC,gBAAgB,EAAE,MAAM,CAAC;IAClC,QAAQ,CAAC,kBAAkB,EAAE,MAAM,CAAC;IACpC,QAAQ,CAAC,mBAAmB,EAAE,MAAM,CAAC;CACtC;AAED,MAAM,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC"}

package/dist/sandbox/providers/e2b/config.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=config.js.map

package/dist/sandbox/providers/e2b/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../../../src/sandbox/providers/e2b/config.ts"],"names":[],"mappings":""}

package/dist/sandbox/providers/e2b/executor.d.ts

@@ -0,0 +1,15 @@
+import type { SandboxExecResult, SandboxExecutor, SandboxRequest, SandboxResult, SandboxRunOptions } from "../../contracts.js";
+import type { E2BExecutionConfig, E2BLifecycleTimings, E2BSleep } from "./config.js";
+import type { E2BSandboxApi } from "./types.js";
+export type { E2BLifecycleTimings } from "./config.js";
+export type { E2BSandboxApi, E2BSandboxHandle } from "./types.js";
+export declare class E2BSandboxExecutor implements SandboxExecutor {
+    #private;
+    constructor(config: E2BExecutionConfig, api?: E2BSandboxApi, sleep?: E2BSleep, timings?: Partial<E2BLifecycleTimings>);
+    run(request: SandboxRequest, options?: SandboxRunOptions): Promise<SandboxResult>;
+    execute(sandboxId: string, command: readonly string[]): Promise<SandboxExecResult>;
+    readFile(sandboxId: string, path: string): Promise<Uint8Array | undefined>;
+    writeFile(sandboxId: string, path: string, contents: Uint8Array | string): Promise<void>;
+    close(): void;
+}
+//# sourceMappingURL=executor.d.ts.map

package/dist/sandbox/providers/e2b/executor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"executor.d.ts","sourceRoot":"","sources":["../../../../src/sandbox/providers/e2b/executor.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,iBAAiB,EACjB,eAAe,EACf,cAAc,EACd,aAAa,EACb,iBAAiB,EAClB,MAAM,oBAAoB,CAAC;AAI5B,OAAO,KAAK,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAYrF,OAAO,KAAK,EAAE,aAAa,EAAoB,MAAM,YAAY,CAAC;AAQlE,YAAY,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AACvD,YAAY,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AA8BlE,qBAAa,kBAAmB,YAAW,eAAe;;gBAQtD,MAAM,EAAE,kBAAkB,EAC1B,GAAG,CAAC,EAAE,aAAa,EACnB,KAAK,GAAE,QAAyB,EAChC,OAAO,GAAE,OAAO,CAAC,mBAAmB,CAAM;IAQtC,GAAG,CAAC,OAAO,EAAE,cAAc,EAAE,OAAO,GAAE,iBAAsB,GAAG,OAAO,CAAC,aAAa,CAAC;IA+K3F,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,MAAM,EAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAIlF,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;IAI1E,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,GAAG,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIxF,KAAK,IAAI,IAAI;CACd"}

package/dist/sandbox/providers/e2b/executor.js

@@ -0,0 +1,207 @@
+import { E2B } from "e2b";
+import { RollingOutput } from "../../../process.js";
+import { LiveSandboxRegistry } from "../../live.js";
+import { E2BCleanup } from "./cleanup.js";
+import { e2bBacking, executeE2BCommand } from "./command.js";
+import { abortReason, createTerminationGate, raceWithTermination } from "./lifecycle.js";
+import { abortableDelay, attachCleanupError, collectPartialOutputs, createErrorConfirmsNoAllocation, sandboxExecutionError, sanitizeCleanupError, waitForCommandKill, } from "./outcome.js";
+import { stageRequestFiles } from "./stage-files.js";
+import { validateAllocatedSandbox, validateConfig, validateLifecycleTimings, validateRequest, } from "./validation.js";
+const CLEANUP_REQUEST_TIMEOUT_MS = 30_000;
+const CLEANUP_CALL_TIMEOUT_MS = 10_000;
+const CLEANUP_RECOVERY_DELAYS_MS = [0, 250, 750, 1_500, 3_000, 5_000, 7_500, 10_000];
+const COMMAND_KILL_GRACE_MS = 500;
+// Creating a sandbox from a large template regularly exceeds the 30 s cleanup budget; a short
+// client timeout here leaves E2B with a half-created sandbox and the worker with nothing.
+const CREATE_REQUEST_TIMEOUT_MS = 120_000;
+const DIAGNOSTIC_TIMEOUT_MS = 5_000;
+const HARD_TIMEOUT_EXIT_CODE = 124;
+const DEFAULT_LIFECYCLE_TIMINGS = {
+    cleanupCallTimeoutMs: CLEANUP_CALL_TIMEOUT_MS,
+    cleanupRecoveryDelaysMs: CLEANUP_RECOVERY_DELAYS_MS,
+    cleanupTimeoutMs: CLEANUP_REQUEST_TIMEOUT_MS,
+    commandKillGraceMs: COMMAND_KILL_GRACE_MS,
+    diagnosticTimeoutMs: DIAGNOSTIC_TIMEOUT_MS,
+};
+class E2BHardTimeoutError extends Error {
+    constructor(sandboxId, stage, timeoutMs) {
+        super(`E2B sandbox ${sandboxId} stage ${stage} exceeded ${timeoutMs}ms`);
+        this.name = "TimeoutError";
+    }
+}
+export class E2BSandboxExecutor {
+    #config;
+    #api;
+    #sleep;
+    #timings;
+    #live = new LiveSandboxRegistry();
+    constructor(config, api, sleep = abortableDelay, timings = {}) {
+        this.#config = validateConfig(config);
+        this.#api = api ?? new E2B(this.#config.credentials).Sandbox;
+        this.#sleep = sleep;
+        this.#timings = validateLifecycleTimings({ ...DEFAULT_LIFECYCLE_TIMINGS, ...timings });
+    }
+    async run(request, options = {}) {
+        options.signal?.throwIfAborted();
+        const resources = validateRequest(request);
+        const allocationId = crypto.randomUUID();
+        const expectedSandboxId = `allocation ${allocationId}`;
+        const metadata = {
+            selfbench_allocation: allocationId,
+            selfbench_run: request.runId.slice(0, 256),
+            selfbench_stage: request.stage.slice(0, 256),
+        };
+        const controller = new AbortController();
+        const stdout = new RollingOutput();
+        const stderr = new RollingOutput();
+        let terminationError;
+        let sandbox;
+        let command;
+        let allocationMayExist = false;
+        let commandKillPromise;
+        const termination = createTerminationGate();
+        const killCommand = () => {
+            if (!command || commandKillPromise) {
+                return;
+            }
+            try {
+                commandKillPromise = command.kill();
+            }
+            catch (error) {
+                commandKillPromise = Promise.reject(error);
+            }
+            void commandKillPromise.catch(() => undefined);
+        };
+        const terminate = (error) => {
+            if (terminationError !== undefined) {
+                return;
+            }
+            terminationError = error;
+            controller.abort(error);
+            termination.reject(error);
+            killCommand();
+        };
+        const abort = () => terminate(abortReason(options.signal));
+        options.signal?.addEventListener("abort", abort, { once: true });
+        if (options.signal?.aborted) {
+            abort();
+        }
+        const hardTimeout = setTimeout(() => {
+            terminate(new E2BHardTimeoutError(sandbox?.sandboxId ?? expectedSandboxId, request.stage, request.timeoutMs));
+        }, request.timeoutMs);
+        hardTimeout.unref();
+        let outcome;
+        try {
+            allocationMayExist = true;
+            const create = this.#api
+                .create(this.#config.image, {
+                lifecycle: { onTimeout: "kill" },
+                metadata,
+                requestTimeoutMs: Math.min(CREATE_REQUEST_TIMEOUT_MS, request.timeoutMs),
+                signal: controller.signal,
+                timeoutMs: request.timeoutMs,
+            })
+                .then((created) => {
+                sandbox = created;
+                allocationMayExist = false;
+                return created;
+            }, (error) => {
+                if (createErrorConfirmsNoAllocation(error)) {
+                    allocationMayExist = false;
+                }
+                throw error;
+            });
+            sandbox = await raceWithTermination(create, termination.promise);
+            allocationMayExist = false;
+            throwIfTerminated(terminationError);
+            const info = await raceWithTermination(sandbox.getInfo({ signal: controller.signal }), termination.promise);
+            validateAllocatedSandbox(sandbox.sandboxId, info, resources, metadata);
+            await stageRequestFiles(sandbox, request, controller.signal, termination.promise);
+            throwIfTerminated(terminationError);
+            const execution = await executeE2BCommand({
+                sandbox,
+                request,
+                options,
+                signal: controller.signal,
+                terminate,
+                stdout,
+                stderr,
+                termination: termination.promise,
+                setCommand: (handle) => {
+                    command = handle;
+                    if (terminationError !== undefined) {
+                        killCommand();
+                    }
+                },
+                startSupervision: (live) => this.#live.start(live.sandboxId, e2bBacking(live), options),
+                sleep: this.#sleep,
+            });
+            throwIfTerminated(terminationError);
+            outcome = { ok: true, result: execution };
+        }
+        catch (error) {
+            killCommand();
+            const diagnosticSignal = AbortSignal.timeout(this.#timings.diagnosticTimeoutMs);
+            await waitForCommandKill(commandKillPromise, diagnosticSignal, this.#timings.commandKillGraceMs);
+            const partialOutputs = sandbox
+                ? await collectPartialOutputs(sandbox, request, diagnosticSignal, this.#timings.diagnosticTimeoutMs)
+                : {};
+            const failure = terminationError ?? error;
+            if (failure instanceof E2BHardTimeoutError) {
+                outcome = {
+                    ok: true,
+                    result: {
+                        sandboxId: sandbox?.sandboxId ?? expectedSandboxId,
+                        exitCode: HARD_TIMEOUT_EXIT_CODE,
+                        stdout: stdout.text(),
+                        stderr: stderr.text(),
+                        outputs: {},
+                    },
+                };
+            }
+            else if (terminationError !== undefined && options.signal?.aborted) {
+                outcome = { ok: false, error: failure };
+            }
+            else if (sandbox) {
+                outcome = {
+                    ok: false,
+                    error: sandboxExecutionError(failure, sandbox, stdout, stderr, partialOutputs),
+                };
+            }
+            else {
+                outcome = { ok: false, error: failure };
+            }
+        }
+        clearTimeout(hardTimeout);
+        options.signal?.removeEventListener("abort", abort);
+        try {
+            await new E2BCleanup(this.#api, this.#sleep, this.#timings).cleanup(metadata, () => sandbox, () => allocationMayExist);
+        }
+        catch (cleanupError) {
+            const publicCleanupError = sanitizeCleanupError(cleanupError, this.#config.credentials.apiKey);
+            outcome = outcome.ok
+                ? { ok: false, error: publicCleanupError }
+                : { ok: false, error: attachCleanupError(outcome.error, publicCleanupError) };
+        }
+        if (!outcome.ok) {
+            throw outcome.error;
+        }
+        return outcome.result;
+    }
+    execute(sandboxId, command) {
+        return this.#live.execute(sandboxId, command);
+    }
+    readFile(sandboxId, path) {
+        return this.#live.readFile(sandboxId, path);
+    }
+    writeFile(sandboxId, path, contents) {
+        return this.#live.writeFile(sandboxId, path, contents);
+    }
+    close() { }
+}
+function throwIfTerminated(error) {
+    if (error !== undefined) {
+        throw error;
+    }
+}
+//# sourceMappingURL=executor.js.map

package/dist/sandbox/providers/e2b/executor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../../../src/sandbox/providers/e2b/executor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,GAAG,EAAE,MAAM,KAAK,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAQpD,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAE7D,OAAO,EAAE,WAAW,EAAE,qBAAqB,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AACzF,OAAO,EACL,cAAc,EACd,kBAAkB,EAClB,qBAAqB,EACrB,+BAA+B,EAC/B,qBAAqB,EACrB,oBAAoB,EACpB,kBAAkB,GACnB,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,OAAO,EACL,wBAAwB,EACxB,cAAc,EACd,wBAAwB,EACxB,eAAe,GAChB,MAAM,iBAAiB,CAAC;AAKzB,MAAM,0BAA0B,GAAG,MAAM,CAAC;AAC1C,MAAM,uBAAuB,GAAG,MAAM,CAAC;AACvC,MAAM,0BAA0B,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAU,CAAC;AAC9F,MAAM,qBAAqB,GAAG,GAAG,CAAC;AAClC,8FAA8F;AAC9F,0FAA0F;AAC1F,MAAM,yBAAyB,GAAG,OAAO,CAAC;AAC1C,MAAM,qBAAqB,GAAG,KAAK,CAAC;AACpC,MAAM,sBAAsB,GAAG,GAAG,CAAC;AACnC,MAAM,yBAAyB,GAAwB;IACrD,oBAAoB,EAAE,uBAAuB;IAC7C,uBAAuB,EAAE,0BAA0B;IACnD,gBAAgB,EAAE,0BAA0B;IAC5C,kBAAkB,EAAE,qBAAqB;IACzC,mBAAmB,EAAE,qBAAqB;CAC3C,CAAC;AAMF,MAAM,mBAAoB,SAAQ,KAAK;IACrC,YAAY,SAAiB,EAAE,KAAa,EAAE,SAAiB;QAC7D,KAAK,CAAC,eAAe,SAAS,UAAU,KAAK,aAAa,SAAS,IAAI,CAAC,CAAC;QACzE,IAAI,CAAC,IAAI,GAAG,cAAc,CAAC;IAC7B,CAAC;CACF;AAED,MAAM,OAAO,kBAAkB;IACpB,OAAO,CAAqB;IAC5B,IAAI,CAAgB;IACpB,MAAM,CAAW;IACjB,QAAQ,CAAsB;IAC9B,KAAK,GAAG,IAAI,mBAAmB,EAAE,CAAC;IAE3C,YACE,MAA0B,EAC1B,GAAmB,EACnB,QAAkB,cAAc,EAChC,UAAwC,EAAE;QAE1C,IAAI,CAAC,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,GAAG,GAAG,IAAI,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC;QAC7D,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,QAAQ,GAAG,wBAAwB,CAAC,EAAE,GAAG,yBAAyB,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACzF,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAuB,EAAE,UAA6B,EAAE;QAChE,OAAO,CAAC,MAAM,EAAE,cAAc,EAAE,CAAC;QACjC,MAAM,SAAS,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;QAC3C,MAAM,YAAY,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QACzC,MAAM,iBAAiB,GAAG,cAAc,YAAY,EAAE,CAAC;QACvD,MAAM,QAAQ,GAAG;YACf,oBAAoB,EAAE,YAAY;YAClC,aAAa,EAAE,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;YAC1C,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;SAC7C,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;QACnC,IAAI,gBAAyB,CAAC;QAC9B,IAAI,OAAqC,CAAC;QAC1C,IAAI,OAAkC,CAAC;QACvC,IAAI,kBAAkB,GAAG,KAAK,CAAC;QAC/B,IAAI,kBAAgD,CAAC;QACrD,MAAM,WAAW,GAAG,qBAAqB,EAAE,CAAC;QAC5C,MAAM,WAAW,GAAG,GAAS,EAAE;YAC7B,IAAI,CAAC,OAAO,IAAI,kBAAkB,EAAE,CAAC;gBACnC,OAAO;YACT,CAAC;YACD,IAAI,CAAC;gBACH,kBAAkB,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;YACtC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,kBAAkB,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC7C,CAAC;YACD,KAAK,kBAAkB,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACjD,CAAC,CAAC;QACF,MAAM,SAAS,GAAG,CAAC,KAAc,EAAQ,EAAE;YACzC,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;gBACnC,OAAO;YACT,CAAC;YACD,gBAAgB,GAAG,KAAK,CAAC;YACzB,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACxB,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC1B,WAAW,EAAE,CAAC;QAChB,CAAC,CAAC;QACF,MAAM,KAAK,GAAG,GAAS,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QACjE,OAAO,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACjE,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;YAC5B,KAAK,EAAE,CAAC;QACV,CAAC;QACD,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,EAAE;YAClC,SAAS,CACP,IAAI,mBAAmB,CACrB,OAAO,EAAE,SAAS,IAAI,iBAAiB,EACvC,OAAO,CAAC,KAAK,EACb,OAAO,CAAC,SAAS,CAClB,CACF,CAAC;QACJ,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;QACtB,WAAW,CAAC,KAAK,EAAE,CAAC;QAEpB,IAAI,OAAmB,CAAC;QACxB,IAAI,CAAC;YACH,kBAAkB,GAAG,IAAI,CAAC;YAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI;iBACrB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE;gBAC1B,SAAS,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;gBAChC,QAAQ;gBACR,gBAAgB,EAAE,IAAI,CAAC,GAAG,CAAC,yBAAyB,EAAE,OAAO,CAAC,SAAS,CAAC;gBACxE,MAAM,EAAE,UAAU,CAAC,MAAM;gBACzB,SAAS,EAAE,OAAO,CAAC,SAAS;aAC7B,CAAC;iBACD,IAAI,CACH,CAAC,OAAO,EAAE,EAAE;gBACV,OAAO,GAAG,OAAO,CAAC;gBAClB,kBAAkB,GAAG,KAAK,CAAC;gBAC3B,OAAO,OAAO,CAAC;YACjB,CAAC,EACD,CAAC,KAAc,EAAE,EAAE;gBACjB,IAAI,+BAA+B,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC3C,kBAAkB,GAAG,KAAK,CAAC;gBAC7B,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC,CACF,CAAC;YACJ,OAAO,GAAG,MAAM,mBAAmB,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,CAAC;YACjE,kBAAkB,GAAG,KAAK,CAAC;YAC3B,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;YAEpC,MAAM,IAAI,GAAG,MAAM,mBAAmB,CACpC,OAAO,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC,EAC9C,WAAW,CAAC,OAAO,CACpB,CAAC;YACF,wBAAwB,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;YAEvE,MAAM,iBAAiB,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,CAAC;YAClF,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;YAEpC,MAAM,SAAS,GAAG,MAAM,iBAAiB,CAAC;gBACxC,OAAO;gBACP,OAAO;gBACP,OAAO;gBACP,MAAM,EAAE,UAAU,CAAC,MAAM;gBACzB,SAAS;gBACT,MAAM;gBACN,MAAM;gBACN,WAAW,EAAE,WAAW,CAAC,OAAO;gBAChC,UAAU,EAAE,CAAC,MAAM,EAAE,EAAE;oBACrB,OAAO,GAAG,MAAM,CAAC;oBACjB,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;wBACnC,WAAW,EAAE,CAAC;oBAChB,CAAC;gBACH,CAAC;gBACD,gBAAgB,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC;gBACvF,KAAK,EAAE,IAAI,CAAC,MAAM;aACnB,CAAC,CAAC;YACH,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;YACpC,OAAO,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QAC5C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,WAAW,EAAE,CAAC;YACd,MAAM,gBAAgB,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;YAChF,MAAM,kBAAkB,CACtB,kBAAkB,EAClB,gBAAgB,EAChB,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CACjC,CAAC;YACF,MAAM,cAAc,GAAG,OAAO;gBAC5B,CAAC,CAAC,MAAM,qBAAqB,CACzB,OAAO,EACP,OAAO,EACP,gBAAgB,EAChB,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAClC;gBACH,CAAC,CAAC,EAAE,CAAC;YACP,MAAM,OAAO,GAAG,gBAAgB,IAAI,KAAK,CAAC;YAC1C,IAAI,OAAO,YAAY,mBAAmB,EAAE,CAAC;gBAC3C,OAAO,GAAG;oBACR,EAAE,EAAE,IAAI;oBACR,MAAM,EAAE;wBACN,SAAS,EAAE,OAAO,EAAE,SAAS,IAAI,iBAAiB;wBAClD,QAAQ,EAAE,sBAAsB;wBAChC,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;wBACrB,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;wBACrB,OAAO,EAAE,EAAE;qBACZ;iBACF,CAAC;YACJ,CAAC;iBAAM,IAAI,gBAAgB,KAAK,SAAS,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;gBACrE,OAAO,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;YAC1C,CAAC;iBAAM,IAAI,OAAO,EAAE,CAAC;gBACnB,OAAO,GAAG;oBACR,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,qBAAqB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,CAAC;iBAC/E,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;YAC1C,CAAC;QACH,CAAC;QAED,YAAY,CAAC,WAAW,CAAC,CAAC;QAC1B,OAAO,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACpD,IAAI,CAAC;YACH,MAAM,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,CACjE,QAAQ,EACR,GAAG,EAAE,CAAC,OAAO,EACb,GAAG,EAAE,CAAC,kBAAkB,CACzB,CAAC;QACJ,CAAC;QAAC,OAAO,YAAY,EAAE,CAAC;YACtB,MAAM,kBAAkB,GAAG,oBAAoB,CAC7C,YAAY,EACZ,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAChC,CAAC;YACF,OAAO,GAAG,OAAO,CAAC,EAAE;gBAClB,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,kBAAkB,EAAE;gBAC1C,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,kBAAkB,CAAC,OAAO,CAAC,KAAK,EAAE,kBAAkB,CAAC,EAAE,CAAC;QAClF,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC;YAChB,MAAM,OAAO,CAAC,KAAK,CAAC;QACtB,CAAC;QACD,OAAO,OAAO,CAAC,MAAM,CAAC;IACxB,CAAC;IAED,OAAO,CAAC,SAAiB,EAAE,OAA0B;QACnD,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC;IAED,QAAQ,CAAC,SAAiB,EAAE,IAAY;QACtC,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC9C,CAAC;IAED,SAAS,CAAC,SAAiB,EAAE,IAAY,EAAE,QAA6B;QACtE,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;IACzD,CAAC;IAED,KAAK,KAAU,CAAC;CACjB;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/sandbox/providers/e2b/lifecycle.d.ts

@@ -0,0 +1,8 @@
+export declare function createTerminationGate(): {
+    readonly promise: Promise<never>;
+    readonly reject: (error: unknown) => void;
+};
+export declare function raceWithTermination<T>(operation: Promise<T>, termination: Promise<never>): Promise<T>;
+export declare function raceWithSignal<T>(operation: Promise<T>, signal: AbortSignal): Promise<T>;
+export declare function abortReason(signal: AbortSignal | undefined): unknown;
+//# sourceMappingURL=lifecycle.d.ts.map

package/dist/sandbox/providers/e2b/lifecycle.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"lifecycle.d.ts","sourceRoot":"","sources":["../../../../src/sandbox/providers/e2b/lifecycle.ts"],"names":[],"mappings":"AAAA,wBAAgB,qBAAqB,IAAI;IACvC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IACjC,QAAQ,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CAC3C,CAOA;AAED,wBAAsB,mBAAmB,CAAC,CAAC,EACzC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,EACrB,WAAW,EAAE,OAAO,CAAC,KAAK,CAAC,GAC1B,OAAO,CAAC,CAAC,CAAC,CAGZ;AAED,wBAAsB,cAAc,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,CAmB9F;AAED,wBAAgB,WAAW,CAAC,MAAM,EAAE,WAAW,GAAG,SAAS,GAAG,OAAO,CAEpE"}

package/dist/sandbox/providers/e2b/lifecycle.js

@@ -0,0 +1,37 @@
+export function createTerminationGate() {
+    let reject = (_error) => { };
+    const promise = new Promise((_resolve, rejectPromise) => {
+        reject = rejectPromise;
+    });
+    void promise.catch(() => undefined);
+    return { promise, reject };
+}
+export async function raceWithTermination(operation, termination) {
+    void operation.catch(() => undefined);
+    return await Promise.race([operation, termination]);
+}
+export async function raceWithSignal(operation, signal) {
+    if (signal.aborted) {
+        void operation.catch(() => undefined);
+        throw abortReason(signal);
+    }
+    let removeAbortListener = () => { };
+    const aborted = new Promise((_resolve, reject) => {
+        const abort = () => {
+            void operation.catch(() => undefined);
+            reject(abortReason(signal));
+        };
+        signal.addEventListener("abort", abort, { once: true });
+        removeAbortListener = () => signal.removeEventListener("abort", abort);
+    });
+    try {
+        return await Promise.race([operation, aborted]);
+    }
+    finally {
+        removeAbortListener();
+    }
+}
+export function abortReason(signal) {
+    return signal?.reason ?? new DOMException("The operation was aborted", "AbortError");
+}
+//# sourceMappingURL=lifecycle.js.map

package/dist/sandbox/providers/e2b/lifecycle.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lifecycle.js","sourceRoot":"","sources":["../../../../src/sandbox/providers/e2b/lifecycle.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,qBAAqB;IAInC,IAAI,MAAM,GAAG,CAAC,MAAe,EAAQ,EAAE,GAAE,CAAC,CAAC;IAC3C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAQ,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE;QAC7D,MAAM,GAAG,aAAa,CAAC;IACzB,CAAC,CAAC,CAAC;IACH,KAAK,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACpC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AAC7B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,SAAqB,EACrB,WAA2B;IAE3B,KAAK,SAAS,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACtC,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,CAAC;AACtD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAI,SAAqB,EAAE,MAAmB;IAChF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,KAAK,SAAS,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACtC,MAAM,WAAW,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC;IACD,IAAI,mBAAmB,GAAG,GAAS,EAAE,GAAE,CAAC,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE;QACtD,MAAM,KAAK,GAAG,GAAS,EAAE;YACvB,KAAK,SAAS,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACtC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAC9B,CAAC,CAAC;QACF,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,mBAAmB,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IACH,IAAI,CAAC;QACH,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC;IAClD,CAAC;YAAS,CAAC;QACT,mBAAmB,EAAE,CAAC;IACxB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,MAA+B;IACzD,OAAO,MAAM,EAAE,MAAM,IAAI,IAAI,YAAY,CAAC,2BAA2B,EAAE,YAAY,CAAC,CAAC;AACvF,CAAC"}

package/dist/sandbox/providers/e2b/outcome.d.ts

@@ -0,0 +1,11 @@
+import type { RollingOutput } from "../../../process.js";
+import { SandboxExecutionError, type SandboxRequest } from "../../contracts.js";
+import type { E2BSandboxHandle } from "./types.js";
+export declare function collectPartialOutputs(sandbox: E2BSandboxHandle, request: SandboxRequest, signal: AbortSignal, requestTimeoutMs: number): Promise<Record<string, Uint8Array>>;
+export declare function sandboxExecutionError(failure: unknown, sandbox: E2BSandboxHandle, stdout: RollingOutput, stderr: RollingOutput, outputs: Record<string, Uint8Array>): SandboxExecutionError;
+export declare function createErrorConfirmsNoAllocation(error: unknown): boolean;
+export declare function waitForCommandKill(commandKill: Promise<boolean> | undefined, diagnosticSignal: AbortSignal, graceMs: number): Promise<void>;
+export declare function abortableDelay(delayMs: number, signal: AbortSignal): Promise<void>;
+export declare function attachCleanupError(primaryError: unknown, cleanupError: unknown): unknown;
+export declare function sanitizeCleanupError(error: unknown, redactedValue: string): Error;
+//# sourceMappingURL=outcome.d.ts.map

package/dist/sandbox/providers/e2b/outcome.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"outcome.d.ts","sourceRoot":"","sources":["../../../../src/sandbox/providers/e2b/outcome.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,qBAAqB,EAAE,KAAK,cAAc,EAAsB,MAAM,oBAAoB,CAAC;AAEpG,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,gBAAgB,EACzB,OAAO,EAAE,cAAc,EACvB,MAAM,EAAE,WAAW,EACnB,gBAAgB,EAAE,MAAM,GACvB,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAgBrC;AAED,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,aAAa,EACrB,MAAM,EAAE,aAAa,EACrB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAClC,qBAAqB,CAavB;AAED,wBAAgB,+BAA+B,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAEvE;AAED,wBAAsB,kBAAkB,CACtC,WAAW,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,SAAS,EACzC,gBAAgB,EAAE,WAAW,EAC7B,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,IAAI,CAAC,CAQf;AAED,wBAAsB,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAExF;AAED,wBAAgB,kBAAkB,CAAC,YAAY,EAAE,OAAO,EAAE,YAAY,EAAE,OAAO,GAAG,OAAO,CAoBxF;AAED,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,GAAG,KAAK,CAMjF"}

package/dist/sandbox/providers/e2b/outcome.js

@@ -0,0 +1,72 @@
+import { setTimeout as delay } from "node:timers/promises";
+import { AuthenticationError, InvalidArgumentError } from "e2b";
+import { SandboxExecutionError } from "../../contracts.js";
+import { raceWithSignal } from "./lifecycle.js";
+export async function collectPartialOutputs(sandbox, request, signal, requestTimeoutMs) {
+    const outputs = {};
+    await Promise.all((request.outputPaths ?? []).map(async (path) => {
+        try {
+            signal.throwIfAborted();
+            outputs[path] = await raceWithSignal(sandbox.files.read(path, { format: "bytes", requestTimeoutMs, signal }), signal);
+        }
+        catch {
+            // Partial outputs are best effort after timeout or infrastructure failure.
+        }
+    }));
+    return outputs;
+}
+export function sandboxExecutionError(failure, sandbox, stdout, stderr, outputs) {
+    const result = {
+        sandboxId: sandbox.sandboxId,
+        exitCode: 1,
+        stdout: stdout.text(),
+        stderr: stderr.text(),
+        outputs,
+    };
+    return new SandboxExecutionError(`${errorMessage(failure)}; E2B sandbox ${sandbox.sandboxId}`, result, { cause: failure });
+}
+export function createErrorConfirmsNoAllocation(error) {
+    return error instanceof AuthenticationError || error instanceof InvalidArgumentError;
+}
+export async function waitForCommandKill(commandKill, diagnosticSignal, graceMs) {
+    if (!commandKill)
+        return;
+    const signal = AbortSignal.any([diagnosticSignal, AbortSignal.timeout(graceMs)]);
+    try {
+        await raceWithSignal(commandKill, signal);
+    }
+    catch {
+        // Sandbox cleanup remains the authoritative process/resource stop.
+    }
+}
+export async function abortableDelay(delayMs, signal) {
+    await delay(delayMs, undefined, { signal });
+}
+export function attachCleanupError(primaryError, cleanupError) {
+    if (primaryError instanceof Error) {
+        const message = `${primaryError.message}; E2B sandbox cleanup also failed: ${errorMessage(cleanupError).slice(0, 500)}`;
+        try {
+            Object.defineProperties(primaryError, {
+                cleanupError: { configurable: true, value: cleanupError },
+                message: { configurable: true, value: message, writable: true },
+            });
+            return primaryError;
+        }
+        catch {
+            const wrapped = new Error(message, { cause: primaryError });
+            wrapped.name = primaryError.name;
+            Object.defineProperty(wrapped, "cleanupError", { configurable: true, value: cleanupError });
+            return wrapped;
+        }
+    }
+    return new AggregateError([primaryError, cleanupError], "E2B sandbox execution and cleanup both failed");
+}
+export function sanitizeCleanupError(error, redactedValue) {
+    const sanitized = new Error(errorMessage(error).replaceAll(redactedValue, "[redacted]").slice(0, 1_000));
+    sanitized.name = "E2BSandboxCleanupError";
+    return sanitized;
+}
+function errorMessage(error) {
+    return error instanceof Error ? error.message : String(error);
+}
+//# sourceMappingURL=outcome.js.map

package/dist/sandbox/providers/e2b/outcome.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"outcome.js","sourceRoot":"","sources":["../../../../src/sandbox/providers/e2b/outcome.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,IAAI,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,MAAM,KAAK,CAAC;AAEhE,OAAO,EAAE,qBAAqB,EAA2C,MAAM,oBAAoB,CAAC;AACpG,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAGhD,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,OAAyB,EACzB,OAAuB,EACvB,MAAmB,EACnB,gBAAwB;IAExB,MAAM,OAAO,GAA+B,EAAE,CAAC;IAC/C,MAAM,OAAO,CAAC,GAAG,CACf,CAAC,OAAO,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QAC7C,IAAI,CAAC;YACH,MAAM,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,cAAc,CAClC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,CAAC,EACvE,MAAM,CACP,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,2EAA2E;QAC7E,CAAC;IACH,CAAC,CAAC,CACH,CAAC;IACF,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,OAAgB,EAChB,OAAyB,EACzB,MAAqB,EACrB,MAAqB,EACrB,OAAmC;IAEnC,MAAM,MAAM,GAAkB;QAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,QAAQ,EAAE,CAAC;QACX,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;QACrB,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;QACrB,OAAO;KACR,CAAC;IACF,OAAO,IAAI,qBAAqB,CAC9B,GAAG,YAAY,CAAC,OAAO,CAAC,iBAAiB,OAAO,CAAC,SAAS,EAAE,EAC5D,MAAM,EACN,EAAE,KAAK,EAAE,OAAO,EAAE,CACnB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,+BAA+B,CAAC,KAAc;IAC5D,OAAO,KAAK,YAAY,mBAAmB,IAAI,KAAK,YAAY,oBAAoB,CAAC;AACvF,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,WAAyC,EACzC,gBAA6B,EAC7B,OAAe;IAEf,IAAI,CAAC,WAAW;QAAE,OAAO;IACzB,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACjF,IAAI,CAAC;QACH,MAAM,cAAc,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IAC5C,CAAC;IAAC,MAAM,CAAC;QACP,mEAAmE;IACrE,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,OAAe,EAAE,MAAmB;IACvE,MAAM,KAAK,CAAC,OAAO,EAAE,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,YAAqB,EAAE,YAAqB;IAC7E,IAAI,YAAY,YAAY,KAAK,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,GAAG,YAAY,CAAC,OAAO,sCAAsC,YAAY,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;QACxH,IAAI,CAAC;YACH,MAAM,CAAC,gBAAgB,CAAC,YAAY,EAAE;gBACpC,YAAY,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE;gBACzD,OAAO,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE;aAChE,CAAC,CAAC;YACH,OAAO,YAAY,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YAC5D,OAAO,CAAC,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC;YACjC,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YAC5F,OAAO,OAAO,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,IAAI,cAAc,CACvB,CAAC,YAAY,EAAE,YAAY,CAAC,EAC5B,+CAA+C,CAChD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,KAAc,EAAE,aAAqB;IACxE,MAAM,SAAS,GAAG,IAAI,KAAK,CACzB,YAAY,CAAC,KAAK,CAAC,CAAC,UAAU,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAC5E,CAAC;IACF,SAAS,CAAC,IAAI,GAAG,wBAAwB,CAAC;IAC1C,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAChE,CAAC"}

package/dist/sandbox/providers/e2b/stage-files.d.ts

@@ -0,0 +1,10 @@
+import { type SandboxRequest } from "../../contracts.js";
+import type { E2BSandboxHandle } from "./types.js";
+/**
+ * Puts the request's files in place before the command runs. Inline files are uploaded in one
+ * `writeFiles` call and their bytes dropped afterwards (the request lives as long as the command,
+ * hours). Remote files are pulled by the sandbox itself from their URL and digest-checked there,
+ * so the worker never buffers or uploads large bundles; E2B recommends this over pushing them.
+ */
+export declare function stageRequestFiles(sandbox: E2BSandboxHandle, request: SandboxRequest, signal: AbortSignal, termination: Promise<never>): Promise<void>;
+//# sourceMappingURL=stage-files.d.ts.map

package/dist/sandbox/providers/e2b/stage-files.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stage-files.d.ts","sourceRoot":"","sources":["../../../../src/sandbox/providers/e2b/stage-files.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,KAAK,cAAc,EACpB,MAAM,oBAAoB,CAAC;AAI5B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD;;;;;GAKG;AACH,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,gBAAgB,EACzB,OAAO,EAAE,cAAc,EACvB,MAAM,EAAE,WAAW,EACnB,WAAW,EAAE,OAAO,CAAC,KAAK,CAAC,GAC1B,OAAO,CAAC,IAAI,CAAC,CA2Cf"}