apex-code 0.0.1-alpha.8 → 0.0.4

package/examples/extensions/border-status-editor.ts

@@ -87,7 +87,7 @@ export default function (pi: ExtensionAPI) {
 		activeTui?.requestRender();
 	});
 
-	pi.on("agent_end", () => {
+	pi.on("agent_settled", () => {
 		isWorking = false;
 		stopSpinner();
 		activeTui?.requestRender();

package/examples/extensions/built-in-tool-renderer.ts

@@ -105,10 +105,14 @@ export default function (pi: ExtensionAPI) {
 
 		renderCall(args, theme, _context) {
 			let text = theme.fg("toolTitle", theme.bold("$ "));
-			const cmd = args.command.length > 80 ? `${args.command.slice(0, 77)}...` : args.command;
-			text += theme.fg("accent", cmd);
-			if (args.timeout) {
-				text += theme.fg("dim", ` (timeout: ${args.timeout}s)`);
+			if ("command" in args) {
+				const cmd = args.command.length > 80 ? `${args.command.slice(0, 77)}...` : args.command;
+				text += theme.fg("accent", cmd);
+				if (args.timeout) {
+					text += theme.fg("dim", ` (timeout: ${args.timeout}s)`);
+				}
+			} else {
+				text += theme.fg("accent", `${args.handle}${"kill" in args ? " · kill" : ""}`);
 			}
 			return new Text(text, 0, 0);
 		},

package/examples/extensions/custom-provider-anthropic/package-lock.json

@@ -1,12 +1,12 @@
 {
   "name": "pi-extension-custom-provider",
-  "version": "0.84.2",
+  "version": "0.84.4",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "pi-extension-custom-provider",
-      "version": "0.84.2",
+      "version": "0.84.4",
       "dependencies": {
         "@anthropic-ai/sdk": "^0.52.0"
       }

package/examples/extensions/custom-provider-anthropic/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-extension-custom-provider-anthropic",
   "private": true,
-  "version": "0.84.2",
+  "version": "0.84.4",
   "type": "module",
   "scripts": {
     "clean": "echo 'nothing to clean'",

package/examples/extensions/custom-provider-gitlab-duo/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-extension-custom-provider-gitlab-duo",
   "private": true,
-  "version": "0.84.2",
+  "version": "0.84.4",
   "type": "module",
   "scripts": {
     "clean": "echo 'nothing to clean'",

package/examples/extensions/git-checkpoint.ts

@@ -1,53 +1,41 @@
 /**
  * Git Checkpoint Extension
  *
- * Creates git stash checkpoints at each turn so /fork can restore code state.
- * When forking, offers to restore code to that point in history.
+ * Offers to restore code state when /fork rewinds the conversation to an earlier entry.
+ *
+ * Capture is not here. Setting `"checkpoints": { "enabled": true }` makes the harness
+ * snapshot the worktree at every turn, keyed to the session entry, into
+ * `refs/apex-code/checkpoints/<sessionId>/<entryId>`. This extension only asks the
+ * question, because whether to prompt is a UI decision and the harness stays headless.
+ *
+ * The engine is resolved from the workspace rather than handed over, because the
+ * registry is git itself. That is also why a checkpoint survives a restart.
  */
 
-import type { ExtensionAPI } from "apex-code";
+import { createGitCheckpoints, type ExtensionAPI, type GitCheckpoints } from "apex-code";
 
 export default function (pi: ExtensionAPI) {
-	const checkpoints = new Map<string, string>();
-	let currentEntryId: string | undefined;
-
-	// Track the current entry ID when user messages are saved
-	pi.on("tool_result", async (_event, ctx) => {
-		const leaf = ctx.sessionManager.getLeafEntry();
-		if (leaf) currentEntryId = leaf.id;
-	});
-
-	pi.on("turn_start", async () => {
-		// Create a git stash entry before LLM makes changes
-		const { stdout } = await pi.exec("git", ["stash", "create"]);
-		const ref = stdout.trim();
-		if (ref && currentEntryId) {
-			checkpoints.set(currentEntryId, ref);
-		}
-	});
+	let engine: Promise<GitCheckpoints | undefined> | undefined;
 
 	pi.on("session_before_fork", async (event, ctx) => {
-		const ref = checkpoints.get(event.entryId);
-		if (!ref) return;
+		if (!ctx.hasUI) return;
 
-		if (!ctx.hasUI) {
-			// In non-interactive mode, don't restore automatically
-			return;
-		}
+		engine ??= createGitCheckpoints(ctx.cwd, ctx.sessionManager.getSessionId());
+		const checkpoint = await (await engine)?.lookup(event.entryId);
+		if (!checkpoint) return;
 
 		const choice = await ctx.ui.select("Restore code state?", [
 			"Yes, restore code to that point",
 			"No, keep current code",
 		]);
-
-		if (choice?.startsWith("Yes")) {
-			await pi.exec("git", ["stash", "apply", ref]);
-			ctx.ui.notify("Code restored to checkpoint", "info");
-		}
-	});
-
-	pi.on("agent_end", async () => {
-		// Clear checkpoints after agent completes
-		checkpoints.clear();
+		if (!choice?.startsWith("Yes")) return;
+
+		const previous = await (await engine)?.restore(checkpoint);
+		ctx.ui.notify(
+			previous
+				? `Code restored. Previous state kept at ${previous.commit.slice(0, 8)}.`
+				: "Checkpoint restore failed; code is unchanged.",
+			previous ? "info" : "error",
+		);
 	});
 }

package/examples/extensions/gondolin/package-lock.json

@@ -1,12 +1,12 @@
 {
 	"name": "pi-extension-gondolin",
-	"version": "0.84.2",
+	"version": "0.84.4",
 	"lockfileVersion": 3,
 	"requires": true,
 	"packages": {
 		"": {
 			"name": "pi-extension-gondolin",
-			"version": "0.84.2",
+			"version": "0.84.4",
 			"dependencies": {
 				"@earendil-works/gondolin": "0.12.0"
 			}

package/examples/extensions/gondolin/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "pi-extension-gondolin",
 	"private": true,
-	"version": "0.84.2",
+	"version": "0.84.4",
 	"type": "module",
 	"scripts": {
 		"clean": "echo 'nothing to clean'",

package/examples/extensions/minimal-mode.ts

@@ -128,8 +128,8 @@ export default function (pi: ExtensionAPI) {
 		},
 
 		renderCall(args, theme, _context) {
-			const command = args.command || "...";
-			const timeout = args.timeout as number | undefined;
+			const command = "command" in args ? args.command : "…";
+			const timeout = "command" in args ? (args.timeout as number | undefined) : undefined;
 			const timeoutSuffix = timeout ? theme.fg("muted", ` (timeout ${timeout}s)`) : "";
 
 			return new Text(theme.fg("toolTitle", theme.bold(`$ ${command}`)) + timeoutSuffix, 0, 0);

package/examples/extensions/notify.ts

@@ -49,7 +49,9 @@ function notify(title: string, body: string): void {
 }
 
 export default function (pi: ExtensionAPI) {
-	pi.on("agent_end", async () => {
+	// `agent_end` fires after each low-level run; Pi may still retry, compact,
+	// or continue with queued follow-ups. Notify only after the full run settles.
+	pi.on("agent_settled", async () => {
 		notify("Pi", "Ready for input");
 	});
 }

package/examples/extensions/permission-gate.ts

@@ -13,7 +13,7 @@ export default function (pi: ExtensionAPI) {
 	pi.on("tool_call", async (event, ctx) => {
 		if (event.toolName !== "bash") return undefined;
 
-		const command = event.input.command as string;
+		const command = (event.input as Record<string, unknown>).command as string;
 		const isDangerous = dangerousPatterns.some((p) => p.test(command));
 
 		if (isDangerous) {

package/examples/extensions/plan-mode/index.ts

@@ -164,7 +164,7 @@ export default function planModeExtension(pi: ExtensionAPI): void {
 	pi.on("tool_call", async (event) => {
 		if (!planModeEnabled || event.toolName !== "bash") return;
 
-		const command = event.input.command as string;
+		const command = (event.input as Record<string, unknown>).command as string;
 		if (!isSafeCommand(command)) {
 			return {
 				block: true,

package/examples/extensions/rpc-demo.ts

@@ -46,7 +46,7 @@ export default function (pi: ExtensionAPI) {
 	pi.on("tool_call", async (event, ctx) => {
 		if (event.toolName !== "bash") return undefined;
 
-		const command = event.input.command as string;
+		const command = (event.input as Record<string, unknown>).command as string;
 		const isDangerous = /\brm\s+(-rf?|--recursive)/i.test(command) || /\bsudo\b/i.test(command);
 
 		if (isDangerous) {

package/examples/extensions/sandbox/package-lock.json

@@ -1,12 +1,12 @@
 {
 	"name": "pi-extension-sandbox",
-	"version": "1.14.2",
+	"version": "1.14.4",
 	"lockfileVersion": 3,
 	"requires": true,
 	"packages": {
 		"": {
 			"name": "pi-extension-sandbox",
-			"version": "1.14.2",
+			"version": "1.14.4",
 			"dependencies": {
 				"@anthropic-ai/sandbox-runtime": "^0.0.26"
 			}

package/examples/extensions/sandbox/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "pi-extension-sandbox",
 	"private": true,
-	"version": "1.14.2",
+	"version": "1.14.4",
 	"type": "module",
 	"scripts": {
 		"clean": "echo 'nothing to clean'",

package/examples/extensions/subagent/README.md

@@ -62,7 +62,7 @@ This tool executes a separate `pi` subprocess with a delegated system prompt and
 
 To enable project-local agents, pass `agentScope: "both"` (or `"project"`). Only do this for repositories you trust.
 
-When running interactively, the tool prompts for confirmation before running project-local agents. Set `confirmProjectAgents: false` to disable.
+When running interactively, the tool prompts for confirmation before running project-local agents in untrusted projects. Trusted projects skip the additional prompt. Set `confirmProjectAgents: false` to disable confirmation.
 
 ## Usage
 

package/examples/extensions/subagent/index.ts

@@ -511,7 +511,12 @@ export default function (pi: ExtensionAPI) {
 				};
 			}
 
-			if ((agentScope === "project" || agentScope === "both") && confirmProjectAgents && ctx.hasUI) {
+			if (
+				(agentScope === "project" || agentScope === "both") &&
+				confirmProjectAgents &&
+				ctx.hasUI &&
+				!ctx.isProjectTrusted()
+			) {
 				const requestedAgentNames = new Set<string>();
 				if (params.chain) for (const step of params.chain) requestedAgentNames.add(step.agent);
 				if (params.tasks) for (const t of params.tasks) requestedAgentNames.add(t.agent);

package/examples/extensions/titlebar-spinner.ts

@@ -48,7 +48,7 @@ export default function (pi: ExtensionAPI) {
 		startAnimation(ctx);
 	});
 
-	pi.on("agent_end", async (_event, ctx) => {
+	pi.on("agent_settled", async (_event, ctx) => {
 		stopAnimation(ctx);
 	});
 

package/examples/extensions/with-deps/package-lock.json

@@ -1,12 +1,12 @@
 {
   "name": "pi-extension-with-deps",
-  "version": "0.84.2",
+  "version": "0.84.4",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "pi-extension-with-deps",
-      "version": "0.84.2",
+      "version": "0.84.4",
       "dependencies": {
         "ms": "^2.1.3"
       },

package/examples/extensions/with-deps/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-extension-with-deps",
   "private": true,
-  "version": "0.84.2",
+  "version": "0.84.4",
   "type": "module",
   "scripts": {
     "clean": "echo 'nothing to clean'",

package/examples/rpc-extension-ui.ts

@@ -576,7 +576,7 @@ async function main() {
 			return;
 		}
 
-		if (data.type === "agent_end") {
+		if (data.type === "agent_settled") {
 			isStreaming = false;
 			hideLoading();
 			outputLog.append("");

package/examples/sdk/06-extensions.ts

@@ -66,7 +66,7 @@ export default function (pi: ExtensionAPI) {
 	});
 
 	pi.on("agent_end", async (event) => {
-		console.log(\`[Extension] Done, \${event.messages.length} messages\`);
+		console.log(\`[Extension] Low-level run ended, \${event.messages.length} messages\`);
 	});
 
 	// Register a custom tool

package/examples/sdk/README.md

@@ -132,7 +132,7 @@ session.subscribe((event) => {
     case "tool_execution_end":
       console.log(`Result: ${event.result}`);
       break;
-    case "agent_end":
+    case "agent_settled":
       console.log("Done");
       break;
   }