@eminent337/aery 0.1.114 → 0.1.116

package/examples/extensions/handoff.ts

@@ -15,6 +15,7 @@
 import type { ExtensionAPI, SessionEntry } from "@eminent337/aery";
 import { BorderedLoader, convertToLlm, serializeConversation } from "@eminent337/aery";
 import { complete, type Message } from "@eminent337/aery-ai";
+import type { AgentMessage } from "@eminent337/aery-core";
 
 const SYSTEM_PROMPT = `You are a context transfer assistant. Given a conversation history and the user's goal for a new thread, generate a focused prompt that:
 
@@ -38,6 +39,44 @@ Files involved:
 ## Task
 [Clear description of what to do next based on user's goal]`;
 
+function entryToMessage(entry: SessionEntry): AgentMessage | undefined {
+	if (entry.type === "message") {
+		return entry.message;
+	}
+	if (entry.type === "compaction") {
+		return {
+			role: "compactionSummary",
+			summary: entry.summary,
+			tokensBefore: entry.tokensBefore,
+			timestamp: new Date(entry.timestamp).getTime(),
+		};
+	}
+	return undefined;
+}
+
+function getHandoffMessages(branch: SessionEntry[]): AgentMessage[] {
+	let compactionIndex = -1;
+	for (let i = branch.length - 1; i >= 0; i--) {
+		if (branch[i].type === "compaction") {
+			compactionIndex = i;
+			break;
+		}
+	}
+	if (compactionIndex < 0) {
+		return branch.map(entryToMessage).filter((message) => message !== undefined);
+	}
+
+	const compaction = branch[compactionIndex];
+	const firstKeptIndex =
+		compaction.type === "compaction" ? branch.findIndex((entry) => entry.id === compaction.firstKeptEntryId) : -1;
+	const compactedBranch = [
+		compaction,
+		...(firstKeptIndex >= 0 ? branch.slice(firstKeptIndex, compactionIndex) : []),
+		...branch.slice(compactionIndex + 1),
+	];
+	return compactedBranch.map(entryToMessage).filter((message) => message !== undefined);
+}
+
 export default function (pi: ExtensionAPI) {
 	pi.registerCommand("handoff", {
 		description: "Transfer context to a new focused session",
@@ -58,11 +97,9 @@ export default function (pi: ExtensionAPI) {
 				return;
 			}
 
-			// Gather conversation context from current branch
-			const branch = ctx.sessionManager.getBranch();
-			const messages = branch
-				.filter((entry): entry is SessionEntry & { type: "message" } => entry.type === "message")
-				.map((entry) => entry.message);
+			// Gather conversation context from current branch. If the branch was compacted,
+			// include the compaction summary plus entries from firstKeptEntryId onward.
+			const messages = getHandoffMessages(ctx.sessionManager.getBranch());
 
 			if (messages.length === 0) {
 				ctx.ui.notify("No conversation to hand off", "error");

package/examples/extensions/hidden-thinking-label.ts

@@ -5,7 +5,7 @@
  * when thinking blocks are hidden.
  *
  * Usage:
- *   aery --extension examples/extensions/hidden-thinking-label.ts
+ *   pi --extension examples/extensions/hidden-thinking-label.ts
  *
  * Test:
  *   1. Load this extension

package/examples/extensions/inline-bash.ts

@@ -1,8 +1,8 @@
 /**
  * Inline Bash Extension - expands inline bash commands in user prompts.
  *
- * Start aery with this extension:
- *   aery -e ./examples/extensions/inline-bash.ts
+ * Start pi with this extension:
+ *   pi -e ./examples/extensions/inline-bash.ts
  *
  * Then type prompts with inline bash:
  *   What's in !{pwd}?

package/examples/extensions/input-transform.ts

@@ -1,10 +1,10 @@
 /**
  * Input Transform Example - demonstrates the `input` event for intercepting user input.
  *
- * Start aery with this extension:
- *   aery -e ./examples/extensions/input-transform.ts
+ * Start pi with this extension:
+ *   pi -e ./examples/extensions/input-transform.ts
  *
- * Then type these inside aery:
+ * Then type these inside pi:
  *   ?quick What is TypeScript?  → "Respond briefly: What is TypeScript?"
  *   ping                        → "pong" (instant, no LLM)
  *   time                        → current time (instant, no LLM)

package/examples/extensions/interactive-shell.ts

@@ -5,7 +5,7 @@
  * with full terminal access. The TUI suspends while they run.
  *
  * Usage:
- *   aery -e examples/extensions/interactive-shell.ts
+ *   pi -e examples/extensions/interactive-shell.ts
  *
  *   !vim file.txt        # Auto-detected as interactive
  *   !i any-command       # Force interactive mode with !i prefix

package/examples/extensions/mac-system-theme.ts

@@ -1,8 +1,8 @@
 /**
- * Syncs aery theme with macOS system appearance (dark/light mode).
+ * Syncs pi theme with macOS system appearance (dark/light mode).
  *
  * Usage:
- *   aery -e examples/extensions/mac-system-theme.ts
+ *   pi -e examples/extensions/mac-system-theme.ts
  */
 
 import { exec } from "node:child_process";

package/examples/extensions/minimal-mode.ts

@@ -11,7 +11,7 @@
  * - Minimal: Shows only tool call, no output (this extension's collapsed mode)
  *
  * Usage:
- *   aery -e ./minimal-mode.ts
+ *   pi -e ./minimal-mode.ts
  *
  * Then use ctrl+o to toggle between minimal (collapsed) and full (expanded) views.
  */

package/examples/extensions/modal-editor.ts

@@ -1,7 +1,7 @@
 /**
  * Modal Editor - vim-like modal editing example
  *
- * Usage: aery --extension ./examples/extensions/modal-editor.ts
+ * Usage: pi --extension ./examples/extensions/modal-editor.ts
  *
  * - Escape: insert → normal mode (in normal mode, aborts agent)
  * - i: normal → insert mode

package/examples/extensions/model-status.ts

@@ -4,7 +4,7 @@
  * Demonstrates the `model_select` hook which fires when the model changes
  * via /model command, Ctrl+P cycling, or session restore.
  *
- * Usage: aery -e ./model-status.ts
+ * Usage: pi -e ./model-status.ts
  */
 
 import type { ExtensionAPI } from "@eminent337/aery";

package/examples/extensions/overlay-qa-tests.ts

@@ -1,7 +1,7 @@
 /**
  * Overlay QA Tests - comprehensive overlay positioning and edge case tests
  *
- * Usage: aery --extension ./examples/extensions/overlay-qa-tests.ts
+ * Usage: pi --extension ./examples/extensions/overlay-qa-tests.ts
  *
  * Commands:
  *   /overlay-animation  - Real-time animation demo (~30 FPS, proves DOOM-like rendering works)
@@ -28,7 +28,7 @@ import { spawn } from "child_process";
 let globalToggleHandle: OverlayHandle | null = null;
 
 export default function (pi: ExtensionAPI) {
-	// Animation demo - proves overlays can handle real-time updates (like aery-doom would need)
+	// Animation demo - proves overlays can handle real-time updates (like pi-doom would need)
 	pi.registerCommand("overlay-animation", {
 		description: "Test real-time animation in overlay (~30 FPS)",
 		handler: async (_args: string, ctx: ExtensionCommandContext) => {
@@ -473,13 +473,13 @@ class StreamingOverflowComponent extends BaseOverlay {
 			echo ""
 			for i in $(seq 1 100); do
 				# Simulate long file paths with OSC 8 hyperlinks (clickable) - tests width overflow
-				DIR="/Users/nicobailon/Documents/development/aery/packages/coding-agent/src/modes/interactive"
+				DIR="/Users/nicobailon/Documents/development/pi-mono/packages/coding-agent/src/modes/interactive"
 				FILE="\${DIR}/components/very-long-component-name-that-exceeds-width-\${i}.ts"
 				echo -e "\\033]8;;file://\${FILE}\\007▶ read: \${FILE}\\033]8;;\\007"
 
 				# Add some colored status messages with long text
 				if [ $((i % 5)) -eq 0 ]; then
-					echo -e "  \\033[32m✓ Successfully processed \${i} files in /Users/nicobailon/Documents/development/aery\\033[0m"
+					echo -e "  \\033[32m✓ Successfully processed \${i} files in /Users/nicobailon/Documents/development/pi-mono\\033[0m"
 				fi
 				if [ $((i % 7)) -eq 0 ]; then
 					echo -e "  \\033[33m⚠ Warning: potential issue detected at line \${i} in very-long-component-name-that-exceeds-width.ts\\033[0m"
@@ -743,7 +743,7 @@ class SidepanelComponent extends BaseOverlay {
 	}
 }
 
-// Animation demo - proves overlays can handle real-time updates like aery-doom
+// Animation demo - proves overlays can handle real-time updates like pi-doom
 class AnimationDemoComponent extends BaseOverlay {
 	private frame = 0;
 	private interval: ReturnType<typeof setInterval> | null = null;
@@ -819,7 +819,7 @@ class AnimationDemoComponent extends BaseOverlay {
 		lines.push(border("│") + padLine(``) + border("│"));
 		lines.push(border("│") + padLine(th.fg("dim", " This proves overlays can handle")) + border("│"));
 		lines.push(border("│") + padLine(th.fg("dim", " real-time game-like rendering.")) + border("│"));
-		lines.push(border("│") + padLine(th.fg("dim", " (aery-doom uses same approach)")) + border("│"));
+		lines.push(border("│") + padLine(th.fg("dim", " (pi-doom uses same approach)")) + border("│"));
 		lines.push(border("│") + padLine(``) + border("│"));
 		lines.push(border("│") + padLine(th.fg("dim", " Press Esc to close")) + border("│"));
 		lines.push(border(`╰${"─".repeat(innerW)}╯`));

package/examples/extensions/overlay-test.ts

@@ -1,7 +1,7 @@
 /**
  * Overlay Test - validates overlay compositing with inline text inputs
  *
- * Usage: aery --extension ./examples/extensions/overlay-test.ts
+ * Usage: pi --extension ./examples/extensions/overlay-test.ts
  *
  * Run /overlay-test to show a floating overlay with:
  * - Inline text inputs within menu items

package/examples/extensions/preset.ts

@@ -30,7 +30,7 @@
  * ```
  *
  * Usage:
- * - `aery --preset plan` - start with plan preset
+ * - `pi --preset plan` - start with plan preset
  * - `/preset` - show selector to switch presets mid-session
  * - `/preset implement` - switch to implement preset directly
  * - `Ctrl+Shift+U` - cycle through presets
@@ -69,7 +69,7 @@ interface PresetsConfig {
  */
 function loadPresets(cwd: string): PresetsConfig {
 	const globalPath = join(getAgentDir(), "presets.json");
-	const projectPath = join(cwd, ".aery", "presets.json");
+	const projectPath = join(cwd, ".pi", "presets.json");
 
 	let globalPresets: PresetsConfig = {};
 	let projectPresets: PresetsConfig = {};

package/examples/extensions/provider-payload.ts

@@ -3,7 +3,7 @@ import { join } from "node:path";
 import type { ExtensionAPI } from "@eminent337/aery";
 
 export default function (pi: ExtensionAPI) {
-	const logFile = join(process.cwd(), ".aery", "provider-payload.log");
+	const logFile = join(process.cwd(), ".pi", "provider-payload.log");
 
 	pi.on("before_provider_request", (event) => {
 		appendFileSync(logFile, `${JSON.stringify(event.payload, null, 2)}\n\n`, "utf8");

package/examples/extensions/rainbow-editor.ts

@@ -1,7 +1,7 @@
 /**
  * Rainbow Editor - highlights "ultrathink" with animated shine effect
  *
- * Usage: aery --extension ./examples/extensions/rainbow-editor.ts
+ * Usage: pi --extension ./examples/extensions/rainbow-editor.ts
  */
 
 import { CustomEditor, type ExtensionAPI } from "@eminent337/aery";

package/examples/extensions/rpc-demo.ts

@@ -25,7 +25,7 @@ export default function (pi: ExtensionAPI) {
 	// -- setTitle, setWidget, setStatus on session lifecycle --
 
 	pi.on("session_start", async (event, ctx) => {
-		ctx.ui.setTitle(event.reason === "new" ? "aery RPC Demo (new session)" : "aery RPC Demo");
+		ctx.ui.setTitle(event.reason === "new" ? "pi RPC Demo (new session)" : "pi RPC Demo");
 		ctx.ui.setWidget("rpc-demo", ["--- RPC Extension UI Demo ---", "Loaded and ready."]);
 		ctx.ui.setStatus("rpc-demo", `Turns: ${turnCount}`);
 	});

package/examples/extensions/sandbox/index.ts

@@ -30,8 +30,8 @@
  * ```
  *
  * Usage:
- * - `aery -e ./sandbox` - sandbox enabled with default/config settings
- * - `aery -e ./sandbox --no-sandbox` - disable sandboxing
+ * - `pi -e ./sandbox` - sandbox enabled with default/config settings
+ * - `pi -e ./sandbox --no-sandbox` - disable sandboxing
  * - `/sandbox` - show current sandbox configuration
  *
  * Setup:
@@ -77,7 +77,7 @@ const DEFAULT_CONFIG: SandboxConfig = {
 };
 
 function loadConfig(cwd: string): SandboxConfig {
-	const projectConfigPath = join(cwd, ".aery", "sandbox.json");
+	const projectConfigPath = join(cwd, ".pi", "sandbox.json");
 	const globalConfigPath = join(getAgentDir(), "extensions", "sandbox.json");
 
 	let globalConfig: Partial<SandboxConfig> = {};

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

@@ -1,12 +1,12 @@
 {
-	"name": "pi-extension-sandbox",
-	"version": "1.0.0",
+	"name": "aery-extension-sandbox",
+	"version": "1.4.0",
 	"lockfileVersion": 3,
 	"requires": true,
 	"packages": {
 		"": {
-			"name": "pi-extension-sandbox",
-			"version": "1.0.0",
+			"name": "aery-extension-sandbox",
+			"version": "1.4.0",
 			"dependencies": {
 				"@anthropic-ai/sandbox-runtime": "^0.0.26"
 			}
@@ -62,9 +62,9 @@
 			}
 		},
 		"node_modules/lodash-es": {
-			"version": "4.17.22",
-			"resolved": "https://registry.npmjs.org/lodash-es/-/lodash-es-4.17.22.tgz",
-			"integrity": "sha512-XEawp1t0gxSi9x01glktRZ5HDy0HXqrM0x5pXQM98EaI0NxO6jVM7omDOxsuEo5UIASAnm2bRp1Jt/e0a2XU8Q==",
+			"version": "4.18.1",
+			"resolved": "https://registry.npmjs.org/lodash-es/-/lodash-es-4.18.1.tgz",
+			"integrity": "sha512-J8xewKD/Gk22OZbhpOVSwcs60zhd95ESDwezOFuA3/099925PdHJ7OFHNTGtajL3AlZkykD32HykiMo+BIBI8A==",
 			"license": "MIT"
 		},
 		"node_modules/shell-quote": {

package/examples/extensions/sandbox/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "aery-extension-sandbox",
 	"private": true,
-	"version": "1.0.0",
+	"version": "1.4.0",
 	"type": "module",
 	"scripts": {
 		"clean": "echo 'nothing to clean'",

package/examples/extensions/shutdown-command.ts

@@ -2,16 +2,16 @@
  * Shutdown Command Extension
  *
  * Adds a /quit command that allows extensions to trigger clean shutdown.
- * Demonstrates how extensions can use ctx.shutdown() to exit aery cleanly.
+ * Demonstrates how extensions can use ctx.shutdown() to exit pi cleanly.
  */
 
 import type { ExtensionAPI } from "@eminent337/aery";
 import { Type } from "typebox";
 
 export default function (pi: ExtensionAPI) {
-	// Register a /quit command that cleanly exits aery
+	// Register a /quit command that cleanly exits pi
 	pi.registerCommand("quit", {
-		description: "Exit aery cleanly",
+		description: "Exit pi cleanly",
 		handler: async (_args, ctx) => {
 			ctx.shutdown();
 		},
@@ -21,7 +21,7 @@ export default function (pi: ExtensionAPI) {
 	pi.registerTool({
 		name: "finish_and_exit",
 		label: "Finish and Exit",
-		description: "Complete a task and exit aery",
+		description: "Complete a task and exit pi",
 		parameters: Type.Object({}),
 		async execute(_toolCallId, _params, _signal, _onUpdate, ctx) {
 			// Do any final work here...
@@ -40,7 +40,7 @@ export default function (pi: ExtensionAPI) {
 	pi.registerTool({
 		name: "deploy_and_exit",
 		label: "Deploy and Exit",
-		description: "Deploy the application and exit aery",
+		description: "Deploy the application and exit pi",
 		parameters: Type.Object({
 			environment: Type.String({ description: "Target environment (e.g., production, staging)" }),
 		}),

package/examples/extensions/ssh.ts

@@ -5,8 +5,8 @@
  * When --ssh is provided, read/write/edit/bash run on the remote.
  *
  * Usage:
- *   aery -e ./ssh.ts --ssh user@host
- *   aery -e ./ssh.ts --ssh user@host:/remote/path
+ *   pi -e ./ssh.ts --ssh user@host
+ *   pi -e ./ssh.ts --ssh user@host:/remote/path
  *
  * Requirements:
  *   - SSH key-based auth (no password prompts)

package/examples/extensions/subagent/README.md

@@ -4,7 +4,7 @@ Delegate tasks to specialized subagents with isolated context windows.
 
 ## Features
 
-- **Isolated context**: Each subagent runs in a separate `aery` process
+- **Isolated context**: Each subagent runs in a separate `pi` process
 - **Streaming output**: See tool calls and progress as they happen
 - **Parallel streaming**: All parallel tasks stream updates simultaneously
 - **Markdown rendering**: Final output rendered with proper formatting (expanded view)
@@ -54,7 +54,7 @@ done
 
 ## Security Model
 
-This tool executes a separate `aery` subprocess with a delegated system prompt and tool/model configuration.
+This tool executes a separate `pi` subprocess with a delegated system prompt and tool/model configuration.
 
 **Project-local agents** (`.aery/agents/*.md`) are repo-controlled prompts that can instruct the model to read files, run bash commands, etc.
 

package/examples/extensions/subagent/agents/aery-pods.md

@@ -15,7 +15,7 @@ Pods handles:
 - OpenAI-compatible API endpoints per model
 - Interactive agent with file system tools for testing
 
-Key env vars: HF_TOKEN (HuggingFace), PI_API_KEY (API auth).
+Key env vars: HF_TOKEN (HuggingFace), AERY_API_KEY (API auth).
 Primary provider: DataCrunch (NFS volumes for shared model storage).
 
 Rules:

package/examples/extensions/subagent/agents.ts

@@ -85,7 +85,7 @@ function isDirectory(p: string): boolean {
 function findNearestProjectAgentsDir(cwd: string): string | null {
 	let currentDir = cwd;
 	while (true) {
-		const candidate = path.join(currentDir, ".aery", "agents");
+		const candidate = path.join(currentDir, ".pi", "agents");
 		if (isDirectory(candidate)) return candidate;
 
 		const parentDir = path.dirname(currentDir);

package/examples/extensions/subagent/index.ts

@@ -1,7 +1,7 @@
 /**
  * Subagent Tool - Delegate tasks to specialized agents
  *
- * Spawns a separate `aery` process for each subagent invocation,
+ * Spawns a separate `pi` process for each subagent invocation,
  * giving it an isolated context window.
  *
  * Supports three modes:
@@ -208,7 +208,7 @@ async function mapWithConcurrencyLimit<TIn, TOut>(
 }
 
 async function writePromptToTempFile(agentName: string, prompt: string): Promise<{ dir: string; filePath: string }> {
-	const tmpDir = await fs.promises.mkdtemp(path.join(os.tmpdir(), "aery-subagent-"));
+	const tmpDir = await fs.promises.mkdtemp(path.join(os.tmpdir(), "pi-subagent-"));
 	const safeName = agentName.replace(/[^\w.-]+/g, "_");
 	const filePath = path.join(tmpDir, `prompt-${safeName}.md`);
 	await withFileMutationQueue(filePath, async () => {

package/examples/extensions/titlebar-spinner.ts

@@ -5,7 +5,7 @@
  * Uses `ctx.ui.setTitle()` to update the terminal title via the extension API.
  *
  * Usage:
- *   aery --extension examples/extensions/titlebar-spinner.ts
+ *   pi --extension examples/extensions/titlebar-spinner.ts
  */
 
 import path from "node:path";

package/examples/extensions/tool-override.ts

@@ -5,7 +5,7 @@
  * This is useful for:
  * - Adding logging or auditing to tool calls
  * - Implementing access control or sandboxing
- * - Routing tool calls to remote systems (e.g., aery-ssh-remote)
+ * - Routing tool calls to remote systems (e.g., pi-ssh-remote)
  * - Modifying tool behavior for specific workflows
  *
  * This example overrides the `read` tool to:
@@ -17,7 +17,7 @@
  * is used automatically (syntax highlighting, line numbers, truncation warnings).
  *
  * Usage:
- *   aery -e ./tool-override.ts
+ *   pi -e ./tool-override.ts
  */
 
 import { type ExtensionAPI, getAgentDir, withFileMutationQueue } from "@eminent337/aery";

package/examples/extensions/truncated-tool.ts

@@ -109,7 +109,7 @@ export default function (pi: ExtensionAPI) {
 
 			if (truncation.truncated) {
 				// Save full output to a temp file so LLM can access it if needed
-				const tempDir = await mkdtemp(join(tmpdir(), "aery-rg-"));
+				const tempDir = await mkdtemp(join(tmpdir(), "pi-rg-"));
 				const tempFile = join(tempDir, "output.txt");
 				await withFileMutationQueue(tempFile, async () => {
 					await writeFile(tempFile, output, "utf8");

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

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

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

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

package/examples/extensions/working-indicator.ts

@@ -2,10 +2,10 @@
  * Working Indicator Extension
  *
  * Demonstrates `ctx.ui.setWorkingIndicator()` for customizing the inline
- * working indicator shown while aery is streaming a response.
+ * working indicator shown while pi is streaming a response.
  *
  * Usage:
- *   aery --extension examples/extensions/working-indicator.ts
+ *   pi --extension examples/extensions/working-indicator.ts
  *
  * Commands:
  *   /working-indicator           Show current mode
@@ -13,7 +13,7 @@
  *   /working-indicator pulse     Use a custom animated indicator
  *   /working-indicator none      Hide the indicator entirely
  *   /working-indicator spinner   Restore an animated spinner
- *   /working-indicator reset     Restore aery's default spinner
+ *   /working-indicator reset     Restore pi's default spinner
  */
 
 import type { ExtensionAPI, ExtensionContext, WorkingIndicatorOptions } from "@eminent337/aery";
@@ -79,7 +79,7 @@ function describeMode(mode: WorkingIndicatorMode): string {
 		case "spinner":
 			return "custom spinner";
 		case "default":
-			return "aery default spinner";
+			return "pi default spinner";
 	}
 }
 

package/examples/extensions/working-message-test.ts

@@ -5,7 +5,7 @@
  * verify they survive across loader recreations (e.g. between agent turns).
  *
  * Usage:
- *   aery --extension examples/extensions/working-message-test.ts
+ *   pi --extension examples/extensions/working-message-test.ts
  *
  * Then send a few messages in interactive mode. The working message should
  * stay "Working... (custom)" with a brown dot indicator every time the

package/examples/sdk/01-minimal.ts

@@ -9,14 +9,18 @@ import { createAgentSession } from "@eminent337/aery";
 
 const { session } = await createAgentSession();
 
-session.subscribe((event) => {
-	if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
-		process.stdout.write(event.assistantMessageEvent.delta);
-	}
-});
+try {
+	session.subscribe((event) => {
+		if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
+			process.stdout.write(event.assistantMessageEvent.delta);
+		}
+	});
 
-await session.prompt("What files are in the current directory?");
-session.state.messages.forEach((msg) => {
-	console.log(msg);
-});
-console.log();
+	await session.prompt("What files are in the current directory?");
+	session.state.messages.forEach((msg) => {
+		console.log(msg);
+	});
+	console.log();
+} finally {
+	session.dispose();
+}

package/examples/sdk/02-custom-model.ts

@@ -38,12 +38,16 @@ if (available.length > 0) {
 		modelRegistry,
 	});
 
-	session.subscribe((event) => {
-		if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
-			process.stdout.write(event.assistantMessageEvent.delta);
-		}
-	});
-
-	await session.prompt("Say hello in one sentence.");
-	console.log();
+	try {
+		session.subscribe((event) => {
+			if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
+				process.stdout.write(event.assistantMessageEvent.delta);
+			}
+		});
+
+		await session.prompt("Say hello in one sentence.");
+		console.log();
+	} finally {
+		session.dispose();
+	}
 }