@themoltnet/pi-extension 0.38.12 → 0.38.14

package/dist/index.d.ts

@@ -1,8 +1,10 @@
 import { createPiOtelExtension } from '@themoltnet/pi-runtime';
 import { ExtensionAPI } from '@earendil-works/pi-coding-agent';
 import { ExtensionContext } from '@earendil-works/pi-coding-agent';
+import { GondolinVmRetirement } from '@themoltnet/pi-runtime';
 import { PiOtelOptions } from '@themoltnet/pi-runtime';
 import { ProviderErrorRetryUi } from '@themoltnet/pi-runtime';
+import { VM } from '@earendil-works/gondolin';
 
 export { createPiOtelExtension }
 
@@ -13,4 +15,10 @@ export default moltnetExtension;
 
 export { PiOtelOptions }
 
+export declare function retirePiExtensionVm(config: {
+    activeVm: Pick<VM, 'close'>;
+    retirement: GondolinVmRetirement;
+    release: () => void;
+}): Promise<void>;
+
 export { }

package/dist/index.js

@@ -2,7 +2,7 @@ import { execFileSync } from "node:child_process";
 import { existsSync, readFileSync } from "node:fs";
 import path from "node:path";
 import { createBashTool, createEditTool, createFindTool, createGrepTool, createLsTool, createReadTool, createWriteTool } from "@earendil-works/pi-coding-agent";
-import { HOST_EXEC_DEFAULT_BASE_ENV, activateAgentEnv, buildWorkspaceMountInstructions, createGondolinBashOps, createGondolinEditOps, createGondolinFindOps, createGondolinLsOps, createGondolinReadOps, createGondolinWriteOps, createMoltNetTools, createPiOtelExtension, ensureSnapshot, executeGondolinGrep, findMainWorktree, resumeVm } from "@themoltnet/pi-runtime";
+import { HOST_EXEC_DEFAULT_BASE_ENV, activateAgentEnv, buildWorkspaceMountInstructions, createGondolinBashOps, createGondolinEditOps, createGondolinFindOps, createGondolinLsOps, createGondolinReadOps, createGondolinToolLifecycle, createGondolinWriteOps, createMoltNetTools, createPiOtelExtension, ensureSnapshot, executeGondolinGrep, findMainWorktree, resumeVm } from "@themoltnet/pi-runtime";
 import { connect } from "@themoltnet/sdk/node";
 //#region src/commands/moltnet-reflect.ts
 var registerMoltnetReflectCommand = (pi, state) => {
@@ -157,6 +157,13 @@ function createPiProviderErrorRetryUi(ctx) {
 		}
 	};
 }
+async function retirePiExtensionVm(config) {
+	try {
+		if (!config.retirement.backendRetired) await config.activeVm.close();
+	} finally {
+		config.release();
+	}
+}
 function moltnetExtension(pi) {
 	pi.registerFlag("agent", {
 		description: "MoltNet agent name (required — pass --agent <name>)",
@@ -255,6 +262,31 @@ function moltnetExtension(pi) {
 		})();
 		return vmStarting;
 	}
+	async function releaseRetiredVm(activeVm, retirement, ctx) {
+		try {
+			await retirePiExtensionVm({
+				activeVm,
+				retirement,
+				release: () => {
+					if (vm === activeVm) {
+						vm = null;
+						vmStarting = null;
+						guestWorkspace = initialGuestWorkspace;
+					}
+				}
+			});
+		} finally {
+			ctx?.ui.setStatus("sandbox", ctx.ui.theme.fg("warning", "Sandbox: retired; restart pending"));
+			ctx?.ui.notify(`Sandbox retired after ${retirement.trigger}; the next tool call will start a fresh VM.`, "warning");
+		}
+	}
+	function createSandboxBashOperations(activeVm, ctx) {
+		const lifecycle = createGondolinToolLifecycle();
+		return createGondolinBashOps(activeVm, localCwd, guestWorkspace, {
+			lifecycle,
+			retireVm: (retirement) => releaseRetiredVm(activeVm, retirement, ctx)
+		});
+	}
 	pi.on("session_start", async (_event, ctx) => {
 		await ensureVm(ctx);
 	});
@@ -297,7 +329,7 @@ function moltnetExtension(pi) {
 		...localBash,
 		label: "bash (sandbox)",
 		async execute(id, params, signal, onUpdate, ctx) {
-			return createBashTool(localCwd, { operations: createGondolinBashOps(await ensureVm(ctx), localCwd, guestWorkspace) }).execute(id, params, signal, onUpdate);
+			return createBashTool(localCwd, { operations: createSandboxBashOperations(await ensureVm(ctx), ctx) }).execute(id, params, signal, onUpdate);
 		}
 	});
 	pi.registerTool({
@@ -318,9 +350,8 @@ function moltnetExtension(pi) {
 			return executeGondolinGrep(await ensureVm(ctx), localCwd, guestWorkspace, params, signal);
 		}
 	});
-	pi.on("user_bash", (_event, _ctx) => {
-		if (!vm) return;
-		return { operations: createGondolinBashOps(vm, localCwd, guestWorkspace) };
+	pi.on("user_bash", async (_event, ctx) => {
+		return { operations: createSandboxBashOperations(await ensureVm(ctx), ctx) };
 	});
 	const sessionErrors = [];
 	const moltnetTools = createMoltNetTools({
@@ -447,4 +478,4 @@ function moltnetExtension(pi) {
 	registerMoltnetReflectCommand(pi, state);
 }
 //#endregion
-export { createPiOtelExtension, createPiProviderErrorRetryUi, moltnetExtension as default };
+export { createPiOtelExtension, createPiProviderErrorRetryUi, moltnetExtension as default, retirePiExtensionVm };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@themoltnet/pi-extension",
-  "version": "0.38.12",
+  "version": "0.38.14",
   "type": "module",
   "description": "MoltNet pi extension — sandboxed tool execution in Gondolin VMs with MoltNet identity and persistent memory",
   "keywords": [
@@ -33,10 +33,10 @@
     ]
   },
   "dependencies": {
-    "@earendil-works/gondolin": "^0.9.1",
-    "@themoltnet/pi-runtime": "0.11.0",
-    "@themoltnet/sdk": "0.136.0",
-    "@themoltnet/os-keyring": "0.2.0"
+    "@earendil-works/gondolin": "^0.12.0",
+    "@themoltnet/os-keyring": "0.2.0",
+    "@themoltnet/pi-runtime": "0.12.1",
+    "@themoltnet/sdk": "0.137.0"
   },
   "peerDependencies": {
     "@earendil-works/pi-ai": "0.79.4",