@mastra/e2b 0.12.1-alpha.2 → 0.12.2-alpha.0

package/dist/index.js

@@ -1546,7 +1546,8 @@ var E2BSandbox = class E2BSandbox extends MastraSandbox {
 	* When the E2B sandbox times out or crashes mid-operation, this method
 	* resets sandbox state, restarts it, and retries the operation once.
 	*
-	* @internal Used by E2BProcessManager to handle dead sandboxes during spawn.
+	* @internal Used by E2BProcessManager (spawn) and E2BCodeModeTransport (file setup)
+	* to handle dead sandboxes.
 	*/
 	async retryOnDead(fn) {
 		try {
@@ -2046,7 +2047,7 @@ var E2BCodeModeTransport = class {
 		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;
+		let e2b = sandbox.e2b;
 		const externals = toolIds.map((toolId) => ({
 			toolId,
 			externalName: sanitizeToolId(toolId)
@@ -2083,9 +2084,12 @@ var E2BCodeModeTransport = class {
 			} catch {}
 		};
 		try {
-			await e2b.files.makeDir(dir);
-			await e2b.files.write(programPath, programSource);
-			await e2b.files.write(runnerPath, runnerSource);
+			await sandbox.retryOnDead(async () => {
+				e2b = sandbox.e2b;
+				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({