@mariozechner/pi-coding-agent 0.34.1 → 0.35.0

package/examples/{custom-tools → extensions}/subagent/index.ts

@@ -19,19 +19,13 @@ import * as path from "node:path";
 import type { AgentToolResult } from "@mariozechner/pi-agent-core";
 import type { Message } from "@mariozechner/pi-ai";
 import { StringEnum } from "@mariozechner/pi-ai";
-import {
-	type CustomTool,
-	type CustomToolAPI,
-	type CustomToolFactory,
-	getMarkdownTheme,
-} from "@mariozechner/pi-coding-agent";
+import { type ExtensionAPI, getMarkdownTheme } from "@mariozechner/pi-coding-agent";
 import { Container, Markdown, Spacer, Text } from "@mariozechner/pi-tui";
 import { Type } from "@sinclair/typebox";
-import { type AgentConfig, type AgentScope, discoverAgents, formatAgentList } from "./agents.js";
+import { type AgentConfig, type AgentScope, discoverAgents } from "./agents.js";
 
 const MAX_PARALLEL_TASKS = 8;
 const MAX_CONCURRENCY = 4;
-const MAX_AGENTS_IN_DESCRIPTION = 10;
 const COLLAPSED_ITEM_COUNT = 10;
 
 function formatTokens(count: number): string {
@@ -224,7 +218,7 @@ function writePromptToTempFile(agentName: string, prompt: string): { dir: string
 type OnUpdateCallback = (partial: AgentToolResult<SubagentDetails>) => void;
 
 async function runSingleAgent(
-	pi: CustomToolAPI,
+	defaultCwd: string,
 	agents: AgentConfig[],
 	agentName: string,
 	task: string,
@@ -289,7 +283,7 @@ async function runSingleAgent(
 		let wasAborted = false;
 
 		const exitCode = await new Promise<number>((resolve) => {
-			const proc = spawn("pi", args, { cwd: cwd ?? pi.cwd, shell: false, stdio: ["ignore", "pipe", "pipe"] });
+			const proc = spawn("pi", args, { cwd: cwd ?? defaultCwd, shell: false, stdio: ["ignore", "pipe", "pipe"] });
 			let buffer = "";
 
 			const processLine = (line: string) => {
@@ -410,32 +404,21 @@ const SubagentParams = Type.Object({
 	cwd: Type.Optional(Type.String({ description: "Working directory for the agent process (single mode)" })),
 });
 
-const factory: CustomToolFactory = (pi) => {
-	const tool: CustomTool<typeof SubagentParams, SubagentDetails> = {
+export default function (pi: ExtensionAPI) {
+	pi.registerTool({
 		name: "subagent",
 		label: "Subagent",
-		get description() {
-			const user = discoverAgents(pi.cwd, "user");
-			const project = discoverAgents(pi.cwd, "project");
-			const userList = formatAgentList(user.agents, MAX_AGENTS_IN_DESCRIPTION);
-			const projectList = formatAgentList(project.agents, MAX_AGENTS_IN_DESCRIPTION);
-			const userSuffix = userList.remaining > 0 ? `; ... and ${userList.remaining} more` : "";
-			const projectSuffix = projectList.remaining > 0 ? `; ... and ${projectList.remaining} more` : "";
-			const projectDirNote = project.projectAgentsDir ? ` (from ${project.projectAgentsDir})` : "";
-			return [
-				"Delegate tasks to specialized subagents with isolated context.",
-				"Modes: single (agent + task), parallel (tasks array), chain (sequential with {previous} placeholder).",
-				'Default agent scope is "user" (from ~/.pi/agent/agents).',
-				'To enable project-local agents in .pi/agents, set agentScope: "both" (or "project").',
-				`User agents: ${userList.text}${userSuffix}.`,
-				`Project agents${projectDirNote}: ${projectList.text}${projectSuffix}.`,
-			].join(" ");
-		},
+		description: [
+			"Delegate tasks to specialized subagents with isolated context.",
+			"Modes: single (agent + task), parallel (tasks array), chain (sequential with {previous} placeholder).",
+			'Default agent scope is "user" (from ~/.pi/agent/agents).',
+			'To enable project-local agents in .pi/agents, set agentScope: "both" (or "project").',
+		].join(" "),
 		parameters: SubagentParams,
 
-		async execute(_toolCallId, params, onUpdate, _ctx, signal) {
+		async execute(_toolCallId, params, onUpdate, ctx, signal) {
 			const agentScope: AgentScope = params.agentScope ?? "user";
-			const discovery = discoverAgents(pi.cwd, agentScope);
+			const discovery = discoverAgents(ctx.cwd, agentScope);
 			const agents = discovery.agents;
 			const confirmProjectAgents = params.confirmProjectAgents ?? true;
 
@@ -466,7 +449,7 @@ const factory: CustomToolFactory = (pi) => {
 				};
 			}
 
-			if ((agentScope === "project" || agentScope === "both") && confirmProjectAgents && pi.hasUI) {
+			if ((agentScope === "project" || agentScope === "both") && confirmProjectAgents && ctx.hasUI) {
 				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);
@@ -479,7 +462,7 @@ const factory: CustomToolFactory = (pi) => {
 				if (projectAgentsRequested.length > 0) {
 					const names = projectAgentsRequested.map((a) => a.name).join(", ");
 					const dir = discovery.projectAgentsDir ?? "(unknown)";
-					const ok = await pi.ui.confirm(
+					const ok = await ctx.ui.confirm(
 						"Run project-local agents?",
 						`Agents: ${names}\nSource: ${dir}\n\nProject agents are repo-controlled. Only continue for trusted repositories.`,
 					);
@@ -515,7 +498,7 @@ const factory: CustomToolFactory = (pi) => {
 						: undefined;
 
 					const result = await runSingleAgent(
-						pi,
+						ctx.cwd,
 						agents,
 						step.agent,
 						taskWithContext,
@@ -589,7 +572,7 @@ const factory: CustomToolFactory = (pi) => {
 
 				const results = await mapWithConcurrencyLimit(params.tasks, MAX_CONCURRENCY, async (t, index) => {
 					const result = await runSingleAgent(
-						pi,
+						ctx.cwd,
 						agents,
 						t.agent,
 						t.task,
@@ -629,7 +612,7 @@ const factory: CustomToolFactory = (pi) => {
 
 			if (params.agent && params.task) {
 				const result = await runSingleAgent(
-					pi,
+					ctx.cwd,
 					agents,
 					params.agent,
 					params.task,
@@ -707,7 +690,7 @@ const factory: CustomToolFactory = (pi) => {
 		},
 
 		renderResult(result, { expanded }, theme) {
-			const { details } = result;
+			const details = result.details as SubagentDetails | undefined;
 			if (!details || details.results.length === 0) {
 				const text = result.content[0];
 				return new Text(text?.type === "text" ? text.text : "(no output)", 0, 0);
@@ -976,9 +959,5 @@ const factory: CustomToolFactory = (pi) => {
 			const text = result.content[0];
 			return new Text(text?.type === "text" ? text.text : "(no output)", 0, 0);
 		},
-	};
-
-	return tool;
-};
-
-export default factory;
+	});
+}

package/examples/{custom-tools/todo/index.ts → extensions/todo.ts}

@@ -1,21 +1,18 @@
 /**
- * Todo Tool - Demonstrates state management via session entries
+ * Todo Extension - Demonstrates state management via session entries
  *
- * This tool stores state in tool result details (not external files),
- * which allows proper branching - when you branch, the todo state
- * is automatically correct for that point in history.
+ * This extension:
+ * - Registers a `todo` tool for the LLM to manage todos
+ * - Registers a `/todos` command for users to view the list
  *
- * The onSession callback reconstructs state by scanning past tool results.
+ * State is stored in tool result details (not external files), which allows
+ * proper branching - when you branch, the todo state is automatically
+ * correct for that point in history.
  */
 
 import { StringEnum } from "@mariozechner/pi-ai";
-import type {
-	CustomTool,
-	CustomToolContext,
-	CustomToolFactory,
-	CustomToolSessionEvent,
-} from "@mariozechner/pi-coding-agent";
-import { Text } from "@mariozechner/pi-tui";
+import type { ExtensionAPI, ExtensionContext, Theme } from "@mariozechner/pi-coding-agent";
+import { matchesKey, Text, truncateToWidth } from "@mariozechner/pi-tui";
 import { Type } from "@sinclair/typebox";
 
 interface Todo {
@@ -24,7 +21,6 @@ interface Todo {
 	done: boolean;
 }
 
-// State stored in tool result details
 interface TodoDetails {
 	action: "list" | "add" | "toggle" | "clear";
 	todos: Todo[];
@@ -32,14 +28,81 @@ interface TodoDetails {
 	error?: string;
 }
 
-// Define schema separately for proper type inference
 const TodoParams = Type.Object({
 	action: StringEnum(["list", "add", "toggle", "clear"] as const),
 	text: Type.Optional(Type.String({ description: "Todo text (for add)" })),
 	id: Type.Optional(Type.Number({ description: "Todo ID (for toggle)" })),
 });
 
-const factory: CustomToolFactory = (_pi) => {
+/**
+ * UI component for the /todos command
+ */
+class TodoListComponent {
+	private todos: Todo[];
+	private theme: Theme;
+	private onClose: () => void;
+	private cachedWidth?: number;
+	private cachedLines?: string[];
+
+	constructor(todos: Todo[], theme: Theme, onClose: () => void) {
+		this.todos = todos;
+		this.theme = theme;
+		this.onClose = onClose;
+	}
+
+	handleInput(data: string): void {
+		if (matchesKey(data, "escape") || matchesKey(data, "ctrl+c")) {
+			this.onClose();
+		}
+	}
+
+	render(width: number): string[] {
+		if (this.cachedLines && this.cachedWidth === width) {
+			return this.cachedLines;
+		}
+
+		const lines: string[] = [];
+		const th = this.theme;
+
+		lines.push("");
+		const title = th.fg("accent", " Todos ");
+		const headerLine =
+			th.fg("borderMuted", "─".repeat(3)) + title + th.fg("borderMuted", "─".repeat(Math.max(0, width - 10)));
+		lines.push(truncateToWidth(headerLine, width));
+		lines.push("");
+
+		if (this.todos.length === 0) {
+			lines.push(truncateToWidth(`  ${th.fg("dim", "No todos yet. Ask the agent to add some!")}`, width));
+		} else {
+			const done = this.todos.filter((t) => t.done).length;
+			const total = this.todos.length;
+			lines.push(truncateToWidth(`  ${th.fg("muted", `${done}/${total} completed`)}`, width));
+			lines.push("");
+
+			for (const todo of this.todos) {
+				const check = todo.done ? th.fg("success", "✓") : th.fg("dim", "○");
+				const id = th.fg("accent", `#${todo.id}`);
+				const text = todo.done ? th.fg("dim", todo.text) : th.fg("text", todo.text);
+				lines.push(truncateToWidth(`  ${check} ${id} ${text}`, width));
+			}
+		}
+
+		lines.push("");
+		lines.push(truncateToWidth(`  ${th.fg("dim", "Press Escape to close")}`, width));
+		lines.push("");
+
+		this.cachedWidth = width;
+		this.cachedLines = lines;
+		return lines;
+	}
+
+	invalidate(): void {
+		this.cachedWidth = undefined;
+		this.cachedLines = undefined;
+	}
+}
+
+export default function (pi: ExtensionAPI) {
 	// In-memory state (reconstructed from session on load)
 	let todos: Todo[] = [];
 	let nextId = 1;
@@ -48,18 +111,14 @@ const factory: CustomToolFactory = (_pi) => {
 	 * Reconstruct state from session entries.
 	 * Scans tool results for this tool and applies them in order.
 	 */
-	const reconstructState = (_event: CustomToolSessionEvent, ctx: CustomToolContext) => {
+	const reconstructState = (ctx: ExtensionContext) => {
 		todos = [];
 		nextId = 1;
 
-		// Use getBranch() to get entries on the current branch
 		for (const entry of ctx.sessionManager.getBranch()) {
 			if (entry.type !== "message") continue;
 			const msg = entry.message;
-
-			// Tool results have role "toolResult"
-			if (msg.role !== "toolResult") continue;
-			if (msg.toolName !== "todo") continue;
+			if (msg.role !== "toolResult" || msg.toolName !== "todo") continue;
 
 			const details = msg.details as TodoDetails | undefined;
 			if (details) {
@@ -69,15 +128,19 @@ const factory: CustomToolFactory = (_pi) => {
 		}
 	};
 
-	const tool: CustomTool<typeof TodoParams, TodoDetails> = {
+	// Reconstruct state on session events
+	pi.on("session_start", async (_event, ctx) => reconstructState(ctx));
+	pi.on("session_switch", async (_event, ctx) => reconstructState(ctx));
+	pi.on("session_branch", async (_event, ctx) => reconstructState(ctx));
+	pi.on("session_tree", async (_event, ctx) => reconstructState(ctx));
+
+	// Register the todo tool for the LLM
+	pi.registerTool({
 		name: "todo",
 		label: "Todo",
 		description: "Manage a todo list. Actions: list, add (text), toggle (id), clear",
 		parameters: TodoParams,
 
-		// Called on session start/switch/branch/clear
-		onSession: reconstructState,
-
 		async execute(_toolCallId, params, _onUpdate, _ctx, _signal) {
 			switch (params.action) {
 				case "list":
@@ -90,21 +153,21 @@ const factory: CustomToolFactory = (_pi) => {
 									: "No todos",
 							},
 						],
-						details: { action: "list", todos: [...todos], nextId },
+						details: { action: "list", todos: [...todos], nextId } as TodoDetails,
 					};
 
 				case "add": {
 					if (!params.text) {
 						return {
 							content: [{ type: "text", text: "Error: text required for add" }],
-							details: { action: "add", todos: [...todos], nextId, error: "text required" },
+							details: { action: "add", todos: [...todos], nextId, error: "text required" } as TodoDetails,
 						};
 					}
 					const newTodo: Todo = { id: nextId++, text: params.text, done: false };
 					todos.push(newTodo);
 					return {
 						content: [{ type: "text", text: `Added todo #${newTodo.id}: ${newTodo.text}` }],
-						details: { action: "add", todos: [...todos], nextId },
+						details: { action: "add", todos: [...todos], nextId } as TodoDetails,
 					};
 				}
 
@@ -112,20 +175,25 @@ const factory: CustomToolFactory = (_pi) => {
 					if (params.id === undefined) {
 						return {
 							content: [{ type: "text", text: "Error: id required for toggle" }],
-							details: { action: "toggle", todos: [...todos], nextId, error: "id required" },
+							details: { action: "toggle", todos: [...todos], nextId, error: "id required" } as TodoDetails,
 						};
 					}
 					const todo = todos.find((t) => t.id === params.id);
 					if (!todo) {
 						return {
 							content: [{ type: "text", text: `Todo #${params.id} not found` }],
-							details: { action: "toggle", todos: [...todos], nextId, error: `#${params.id} not found` },
+							details: {
+								action: "toggle",
+								todos: [...todos],
+								nextId,
+								error: `#${params.id} not found`,
+							} as TodoDetails,
 						};
 					}
 					todo.done = !todo.done;
 					return {
 						content: [{ type: "text", text: `Todo #${todo.id} ${todo.done ? "completed" : "uncompleted"}` }],
-						details: { action: "toggle", todos: [...todos], nextId },
+						details: { action: "toggle", todos: [...todos], nextId } as TodoDetails,
 					};
 				}
 
@@ -135,14 +203,19 @@ const factory: CustomToolFactory = (_pi) => {
 					nextId = 1;
 					return {
 						content: [{ type: "text", text: `Cleared ${count} todos` }],
-						details: { action: "clear", todos: [], nextId: 1 },
+						details: { action: "clear", todos: [], nextId: 1 } as TodoDetails,
 					};
 				}
 
 				default:
 					return {
 						content: [{ type: "text", text: `Unknown action: ${params.action}` }],
-						details: { action: "list", todos: [...todos], nextId, error: `unknown action: ${params.action}` },
+						details: {
+							action: "list",
+							todos: [...todos],
+							nextId,
+							error: `unknown action: ${params.action}`,
+						} as TodoDetails,
 					};
 			}
 		},
@@ -155,13 +228,12 @@ const factory: CustomToolFactory = (_pi) => {
 		},
 
 		renderResult(result, { expanded }, theme) {
-			const { details } = result;
+			const details = result.details as TodoDetails | undefined;
 			if (!details) {
 				const text = result.content[0];
 				return new Text(text?.type === "text" ? text.text : "", 0, 0);
 			}
 
-			// Error
 			if (details.error) {
 				return new Text(theme.fg("error", `Error: ${details.error}`), 0, 0);
 			}
@@ -208,9 +280,20 @@ const factory: CustomToolFactory = (_pi) => {
 					return new Text(theme.fg("success", "✓ ") + theme.fg("muted", "Cleared all todos"), 0, 0);
 			}
 		},
-	};
+	});
 
-	return tool;
-};
+	// Register the /todos command for users
+	pi.registerCommand("todos", {
+		description: "Show all todos on the current branch",
+		handler: async (_args, ctx) => {
+			if (!ctx.hasUI) {
+				ctx.ui.notify("/todos requires interactive mode", "error");
+				return;
+			}
 
-export default factory;
+			await ctx.ui.custom<void>((_tui, theme, done) => {
+				return new TodoListComponent(todos, theme, () => done());
+			});
+		},
+	});
+}

package/examples/{hooks → extensions}/tools.ts

@@ -1,16 +1,16 @@
 /**
- * Tools Hook
+ * Tools Extension
  *
  * Provides a /tools command to enable/disable tools interactively.
  * Tool selection persists across session reloads and respects branch navigation.
  *
  * Usage:
- * 1. Copy this file to ~/.pi/agent/hooks/ or your project's .pi/hooks/
+ * 1. Copy this file to ~/.pi/agent/extensions/ or your project's .pi/extensions/
  * 2. Use /tools to open the tool selector
  */
 
+import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
 import { getSettingsListTheme } from "@mariozechner/pi-coding-agent";
-import type { HookAPI, HookContext } from "@mariozechner/pi-coding-agent/hooks";
 import { Container, type SettingItem, SettingsList } from "@mariozechner/pi-tui";
 
 // State persisted to session
@@ -18,7 +18,7 @@ interface ToolsState {
 	enabledTools: string[];
 }
 
-export default function toolsHook(pi: HookAPI) {
+export default function toolsExtension(pi: ExtensionAPI) {
 	// Track enabled tools
 	let enabledTools: Set<string> = new Set();
 	let allTools: string[] = [];
@@ -36,7 +36,7 @@ export default function toolsHook(pi: HookAPI) {
 	}
 
 	// Find the last tools-config entry in the current branch
-	function restoreFromBranch(ctx: HookContext) {
+	function restoreFromBranch(ctx: ExtensionContext) {
 		allTools = pi.getAllTools();
 
 		// Get entries in current branch only

package/examples/extensions/with-deps/index.ts

@@ -0,0 +1,40 @@
+/**
+ * Example extension with its own npm dependencies.
+ * Tests that jiti resolves modules from the extension's own node_modules.
+ *
+ * Requires: npm install in this directory
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { Type } from "@sinclair/typebox";
+import ms from "ms";
+
+export default function (pi: ExtensionAPI) {
+	// Use the ms package to prove it loaded
+	const uptime = ms(process.uptime() * 1000, { long: true });
+	console.log(`[with-deps] Extension loaded. Process uptime: ${uptime}`);
+
+	// Register a tool that uses ms
+	pi.registerTool({
+		name: "parse_duration",
+		label: "Parse Duration",
+		description: "Parse a human-readable duration string (e.g., '2 days', '1h', '5m') to milliseconds",
+		parameters: Type.Object({
+			duration: Type.String({ description: "Duration string like '2 days', '1h', '5m'" }),
+		}),
+		execute: async (_toolCallId, params) => {
+			const result = ms(params.duration as ms.StringValue);
+			if (result === undefined) {
+				return {
+					content: [{ type: "text", text: `Invalid duration: "${params.duration}"` }],
+					isError: true,
+					details: {},
+				};
+			}
+			return {
+				content: [{ type: "text", text: `${params.duration} = ${result} milliseconds` }],
+				details: {},
+			};
+		},
+	});
+}

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

@@ -0,0 +1,31 @@
+{
+  "name": "pi-extension-with-deps",
+  "version": "1.0.0",
+  "lockfileVersion": 3,
+  "requires": true,
+  "packages": {
+    "": {
+      "name": "pi-extension-with-deps",
+      "version": "1.0.0",
+      "dependencies": {
+        "ms": "^2.1.3"
+      },
+      "devDependencies": {
+        "@types/ms": "^2.1.0"
+      }
+    },
+    "node_modules/@types/ms": {
+      "version": "2.1.0",
+      "resolved": "https://registry.npmjs.org/@types/ms/-/ms-2.1.0.tgz",
+      "integrity": "sha512-GsCCIZDE/p3i96vtEqx+7dBUGXrc7zeSK3wwPHIaRThS+9OhWIXRqzs4d6k1SVU8g91DrNRWxWUGhp5KXQb2VA==",
+      "dev": true,
+      "license": "MIT"
+    },
+    "node_modules/ms": {
+      "version": "2.1.3",
+      "resolved": "https://registry.npmjs.org/ms/-/ms-2.1.3.tgz",
+      "integrity": "sha512-6FlzubTLZG3J2a/NVCAleEhjzq5oxgHyaCU9yYXvcLsvoVaHJq/s5xXI6/XXP6tz7R9xAOtHnSO/tXtF3WRTlA==",
+      "license": "MIT"
+    }
+  }
+}

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

@@ -0,0 +1,16 @@
+{
+  "name": "pi-extension-with-deps",
+  "version": "1.0.0",
+  "type": "module",
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  },
+  "dependencies": {
+    "ms": "^2.1.3"
+  },
+  "devDependencies": {
+    "@types/ms": "^2.1.0"
+  }
+}

package/examples/sdk/01-minimal.ts

@@ -1,7 +1,7 @@
 /**
  * Minimal SDK Usage
  *
- * Uses all defaults: discovers skills, hooks, tools, context files
+ * Uses all defaults: discovers skills, extensions, tools, context files
  * from cwd and ~/.pi/agent. Model chosen from settings or first available.
  */
 

package/examples/sdk/05-tools.ts

@@ -1,27 +1,28 @@
 /**
  * Tools Configuration
  *
- * Use built-in tool sets, individual tools, or add custom tools.
+ * Use built-in tool sets or individual tools.
  *
  * IMPORTANT: When using a custom `cwd`, you must use the tool factory functions
  * (createCodingTools, createReadOnlyTools, createReadTool, etc.) to ensure
  * tools resolve paths relative to your cwd, not process.cwd().
+ *
+ * For custom tools, see 06-extensions.ts - custom tools are now registered
+ * via the extensions system using pi.registerTool().
  */
 
 import {
-	bashTool, // read, bash, edit, write - uses process.cwd()
-	type CustomTool,
+	bashTool,
 	createAgentSession,
 	createBashTool,
-	createCodingTools, // Factory: creates tools for specific cwd
+	createCodingTools,
 	createGrepTool,
 	createReadTool,
 	grepTool,
-	readOnlyTools, // read, grep, find, ls - uses process.cwd()
+	readOnlyTools,
 	readTool,
 	SessionManager,
 } from "@mariozechner/pi-coding-agent";
-import { Type } from "@sinclair/typebox";
 
 // Read-only mode (no edit/write) - uses process.cwd()
 await createAgentSession({
@@ -53,38 +54,3 @@ await createAgentSession({
 	sessionManager: SessionManager.inMemory(),
 });
 console.log("Specific tools with custom cwd session created");
-
-// Inline custom tool (needs TypeBox schema)
-const weatherTool: CustomTool = {
-	name: "get_weather",
-	label: "Get Weather",
-	description: "Get current weather for a city",
-	parameters: Type.Object({
-		city: Type.String({ description: "City name" }),
-	}),
-	execute: async (_toolCallId, params) => ({
-		content: [{ type: "text", text: `Weather in ${(params as { city: string }).city}: 22°C, sunny` }],
-		details: {},
-	}),
-};
-
-const { session } = await createAgentSession({
-	customTools: [{ tool: weatherTool }],
-	sessionManager: SessionManager.inMemory(),
-});
-
-session.subscribe((event) => {
-	if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
-		process.stdout.write(event.assistantMessageEvent.delta);
-	}
-});
-
-await session.prompt("What's the weather in Tokyo?");
-console.log();
-
-// Merge with discovered tools from cwd/.pi/tools and ~/.pi/agent/tools:
-// const discovered = await discoverCustomTools();
-// customTools: [...discovered, { tool: myTool }]
-
-// Or add paths without replacing discovery:
-// additionalCustomToolPaths: ["/extra/tools"]