@oh-my-pi/pi-coding-agent 15.1.2 → 15.1.4

package/src/edit/modes/patch.ts

@@ -1578,16 +1578,16 @@ export async function computePatchDiff(
 
 export const patchEditEntrySchema = z
 	.object({
-		op: z.enum(["create", "delete", "update"]).optional().describe("Operation (default: update)"),
-		rename: z.string().describe("New path for move").optional(),
-		diff: z.string().describe("Diff hunks (update) or full content (create)").optional(),
+		op: z.enum(["create", "delete", "update"]).optional().describe("operation (default update)"),
+		rename: z.string().describe("new path for move").optional(),
+		diff: z.string().describe("diff hunks or full content for create").optional(),
 	})
 	.strict();
 
 export const patchEditSchema = z
 	.object({
-		path: z.string().describe("file path for edits"),
-		edits: z.array(patchEditEntrySchema).min(1).describe("Patch operations"),
+		path: z.string().describe("file path"),
+		edits: z.array(patchEditEntrySchema).min(1).describe("patch operations"),
 	})
 	.strict();
 

package/src/edit/modes/replace.ts

@@ -978,16 +978,16 @@ export function findContextLine(
 
 export const replaceEditEntrySchema = z
 	.object({
-		old_text: z.string().describe("Text to find (fuzzy whitespace matching enabled)"),
-		new_text: z.string().describe("Replacement text"),
-		all: z.boolean().describe("Replace all occurrences (default: unique match required)").optional(),
+		old_text: z.string().describe("text to find"),
+		new_text: z.string().describe("replacement text"),
+		all: z.boolean().describe("replace all occurrences").optional(),
 	})
 	.strict();
 
 export const replaceEditSchema = z
 	.object({
-		path: z.string().describe("file path for edits"),
-		edits: z.array(replaceEditEntrySchema).min(1).describe("Replacements"),
+		path: z.string().describe("file path"),
+		edits: z.array(replaceEditEntrySchema).min(1).describe("replacements"),
 	})
 	.strict();
 

package/src/edit/renderer.ts

@@ -1,9 +1,10 @@
 /**
  * Edit tool renderer and LSP batching helpers.
  */
-import { sanitizeText } from "@oh-my-pi/pi-natives";
+
 import type { Component } from "@oh-my-pi/pi-tui";
 import { Text, visibleWidth, wrapTextWithAnsi } from "@oh-my-pi/pi-tui";
+import { sanitizeText } from "@oh-my-pi/pi-utils";
 import type { RenderResultOptions } from "../extensibility/custom-tools/types";
 import type { FileDiagnosticsResult } from "../lsp";
 import { renderDiff as renderDiffColored } from "../modes/components/diff";

package/src/edit/streaming.ts

@@ -13,7 +13,7 @@
  * the injected `editMode` rather than probing argument shape.
  */
 
-import { sanitizeText } from "@oh-my-pi/pi-natives";
+import { sanitizeText } from "@oh-my-pi/pi-utils";
 import {
 	ABORT_MARKER,
 	BEGIN_PATCH_MARKER,

package/src/eval/index.ts

@@ -1,6 +1,4 @@
 export * from "./backend";
 export { default as jsBackend } from "./js";
-export * from "./parse";
 export { default as pythonBackend } from "./py";
-export * from "./sniff";
 export * from "./types";

package/src/eval/js/shared/runtime.ts

@@ -38,6 +38,75 @@ export interface RuntimeOptions {
 	extraGlobals?: Record<string, unknown>;
 }
 
+// Strict base64: characters from the standard alphabet plus optional `=` padding, and a
+// length that is a multiple of four. URL-safe base64 and embedded whitespace are not
+// accepted — the Anthropic API only honors strict base64 in image sources.
+const BASE64_STRICT_RE = /^[A-Za-z0-9+/]+={0,2}$/;
+const DECIMAL_CSV_RE = /^\d{1,3}(?:,\d{1,3})*$/;
+
+function isStrictBase64(s: string): boolean {
+	if (s.length === 0 || s.length % 4 !== 0) return false;
+	return BASE64_STRICT_RE.test(s);
+}
+
+/**
+ * Normalize the `data` field of an `{ type: "image", data, mimeType }` display payload
+ * into strict base64. Accepts:
+ *   - already-valid base64 strings (passed through verbatim)
+ *   - `Uint8Array` / `Buffer` / `ArrayBuffer` / typed array views
+ *   - `{ type: "Buffer", data: number[] }` (the shape Node serializes Buffers to via
+ *     `JSON.stringify`)
+ *   - decimal-CSV byte strings (the output of `uint8array.toString("base64")`, which
+ *     silently ignores the encoding argument and returns `Array.prototype.toString` —
+ *     a footgun for callers expecting `Buffer.toString` semantics)
+ * Returns `null` if no recovery is possible.
+ */
+function coerceImageBase64(data: unknown): string | null {
+	if (typeof data === "string") {
+		if (isStrictBase64(data)) return data;
+		if (DECIMAL_CSV_RE.test(data)) {
+			const parts = data.split(",");
+			const bytes = new Uint8Array(parts.length);
+			for (let i = 0; i < parts.length; i++) {
+				const n = Number(parts[i]);
+				if (!Number.isInteger(n) || n < 0 || n > 255) return null;
+				bytes[i] = n;
+			}
+			return Buffer.from(bytes).toString("base64");
+		}
+		return null;
+	}
+	if (data instanceof Uint8Array) return Buffer.from(data).toString("base64");
+	if (data instanceof ArrayBuffer) return Buffer.from(data).toString("base64");
+	if (ArrayBuffer.isView(data)) {
+		const view = data as ArrayBufferView;
+		return Buffer.from(view.buffer, view.byteOffset, view.byteLength).toString("base64");
+	}
+	if (data && typeof data === "object") {
+		const obj = data as { type?: unknown; data?: unknown };
+		if (obj.type === "Buffer" && Array.isArray(obj.data)) {
+			const arr = obj.data as unknown[];
+			const bytes = new Uint8Array(arr.length);
+			for (let i = 0; i < arr.length; i++) {
+				const n = arr[i];
+				if (typeof n !== "number" || !Number.isInteger(n) || n < 0 || n > 255) return null;
+				bytes[i] = n;
+			}
+			return Buffer.from(bytes).toString("base64");
+		}
+	}
+	return null;
+}
+
+function describeDataType(data: unknown): string {
+	if (data === null) return "null";
+	if (data instanceof Uint8Array) return "Uint8Array";
+	if (data instanceof ArrayBuffer) return "ArrayBuffer";
+	if (ArrayBuffer.isView(data)) return data.constructor.name;
+	if (typeof data === "string") return `string(${data.length})`;
+	return typeof data;
+}
+
 /**
  * Shared JS runtime for the eval worker and the browser tab worker. Owns the prelude,
  * helper bag, console bridge, and indirect-eval execution. Emits text/display/tool-call
@@ -109,8 +178,19 @@ export class JsRuntime {
 		if (!hooks) return;
 		if (value && typeof value === "object") {
 			const record = value as Record<string, unknown>;
-			if (record.type === "image" && typeof record.data === "string" && typeof record.mimeType === "string") {
-				hooks.onDisplay({ type: "image", data: record.data, mimeType: record.mimeType });
+			if (record.type === "image" && typeof record.mimeType === "string") {
+				const data = coerceImageBase64(record.data);
+				if (data !== null) {
+					hooks.onDisplay({ type: "image", data, mimeType: record.mimeType });
+					return;
+				}
+				logger.warn("js displayValue: dropping image with unrecognized data shape", {
+					mimeType: record.mimeType,
+					dataType: describeDataType(record.data),
+				});
+				hooks.onText(
+					`[display: image dropped — \`data\` must be a base64 string, Uint8Array/Buffer, or ArrayBuffer; got ${describeDataType(record.data)}]\n`,
+				);
 				return;
 			}
 			try {
@@ -137,6 +217,13 @@ export class JsRuntime {
 			},
 			__omp_import__: async (source: string, options?: ImportCallOptions) => {
 				const target = resolveImportSpecifier(this.#cwd, source);
+				// Always invalidate cached module records for user-owned source files so edits
+				// between cells are picked up. Bun ignores query-string busting on `file:` URLs
+				// but honors `delete require.cache[absPath]`; bare specifiers and URL schemes are
+				// left alone to keep package identity stable across cells.
+				if (isLocalPathSpecifier(source) && path.isAbsolute(target)) {
+					delete require.cache[target];
+				}
 				return options !== undefined ? await import(target, options) : await import(target);
 			},
 			__omp_emit_status__: (op: string, data: Record<string, unknown> = {}) => {
@@ -193,3 +280,21 @@ function resolveImportSpecifier(cwd: string, source: string): string {
 		return source;
 	}
 }
+
+/**
+ * Returns true when the original specifier is a relative or absolute filesystem path
+ * (i.e. user-owned source the agent is iterating on). Bare specifiers and URL schemes
+ * are excluded — `node:` built-ins cannot be reloaded, and busting bare packages would
+ * defeat module identity for every cell while bringing no editing benefit.
+ */
+function isLocalPathSpecifier(source: string): boolean {
+	return (
+		source.startsWith("./") ||
+		source.startsWith("../") ||
+		source === "." ||
+		source === ".." ||
+		source.startsWith("/") ||
+		source.startsWith("~/") ||
+		/^[a-zA-Z]:[\\/]/.test(source)
+	);
+}

package/src/eval/py/kernel.ts

@@ -626,7 +626,7 @@ export class PythonKernel {
 		const exitedPromise = this.#exitedPromise;
 		const timeout = new Promise<null>(resolve => {
 			const timer = setTimeout(() => resolve(null), Math.max(0, timeoutMs));
-			(timer as { unref?: () => void }).unref?.();
+			timer.unref?.();
 		});
 		return Promise.race([exitedPromise.then(code => code as number | null), timeout]);
 	}

package/src/exa/researcher.ts

@@ -14,9 +14,9 @@ const researcherStartTool = createExaTool(
 	"Start Deep Research",
 	"Start an asynchronous deep research task using Exa's researcher. Returns a task_id for polling completion.",
 	z.object({
-		query: z.string().describe("Research query to investigate"),
-		depth: z.number().int().min(1).max(5).describe("Research depth (1-5, default: 3)").optional(),
-		breadth: z.number().int().min(1).max(5).describe("Research breadth (1-5, default: 3)").optional(),
+		query: z.string().describe("research query"),
+		depth: z.number().int().min(1).max(5).describe("research depth (1-5)").optional(),
+		breadth: z.number().int().min(1).max(5).describe("research breadth (1-5)").optional(),
 	}),
 	"deep_researcher_start",
 	{ formatResponse: false },
@@ -27,7 +27,7 @@ const researcherPollTool = createExaTool(
 	"Poll Research Status",
 	"Poll the status of an asynchronous research task. Returns status (pending|running|completed|failed) and result if completed.",
 	z.object({
-		task_id: z.string().describe("Task ID returned from exa_researcher_start"),
+		task_id: z.string().describe("task id"),
 	}),
 	"deep_researcher_check",
 	{ formatResponse: false },

package/src/exa/search.ts

@@ -30,28 +30,16 @@ Parameters:
 - num_results: Maximum number of results to return (default: 10, max: 100)`,
 
 	z.object({
-		query: z.string().describe("Search query"),
-		type: z
-			.enum(["keyword", "neural", "auto"])
-			.describe("Search type - neural (semantic), keyword (exact), or auto")
-			.optional(),
-		include_domains: z.array(z.string()).describe("Only include results from these domains").optional(),
-		exclude_domains: z.array(z.string()).describe("Exclude results from these domains").optional(),
-		start_published_date: z
-			.string()
-			.describe("Filter results published after this date (ISO 8601 format)")
-			.optional(),
-		end_published_date: z.string().describe("Filter results published before this date (ISO 8601 format)").optional(),
-		use_autoprompt: z.boolean().describe("Let Exa optimize your query automatically (default: true)").optional(),
-		text: z.boolean().describe("Include page text content in results (costs more, default: false)").optional(),
-		highlights: z.boolean().describe("Include highlighted relevant snippets (default: false)").optional(),
-		num_results: z
-			.number()
-			.int()
-			.min(1)
-			.max(100)
-			.describe("Maximum number of results to return (default: 10, max: 100)")
-			.optional(),
+		query: z.string().describe("search query"),
+		type: z.enum(["keyword", "neural", "auto"]).describe("search type").optional(),
+		include_domains: z.array(z.string()).describe("include domains").optional(),
+		exclude_domains: z.array(z.string()).describe("exclude domains").optional(),
+		start_published_date: z.string().describe("published after (iso 8601)").optional(),
+		end_published_date: z.string().describe("published before (iso 8601)").optional(),
+		use_autoprompt: z.boolean().describe("autoprompt").optional(),
+		text: z.boolean().describe("include page text").optional(),
+		highlights: z.boolean().describe("include highlights").optional(),
+		num_results: z.number().int().min(1).max(100).describe("max results (1-100)").optional(),
 	}),
 	"web_search_exa",
 );

package/src/exa/websets.ts

@@ -53,8 +53,8 @@ const websetCreateTool = createWebsetTool(
 	"Create Webset",
 	"Create a new webset collection for organizing web content.",
 	z.object({
-		name: z.string().describe("Name of the webset"),
-		description: z.string().describe("Optional description").optional(),
+		name: z.string().describe("webset name"),
+		description: z.string().describe("description").optional(),
 	}),
 	"create_webset",
 );
@@ -72,7 +72,7 @@ const websetGetTool = createWebsetTool(
 	"Get Webset",
 	"Get details of a specific webset by ID.",
 	z.object({
-		id: z.string().describe("Webset ID"),
+		id: z.string().describe("webset id"),
 	}),
 	"get_webset",
 );
@@ -82,9 +82,9 @@ const websetUpdateTool = createWebsetTool(
 	"Update Webset",
 	"Update a webset's name or description.",
 	z.object({
-		id: z.string().describe("Webset ID"),
-		name: z.string().describe("New name").optional(),
-		description: z.string().describe("New description").optional(),
+		id: z.string().describe("webset id"),
+		name: z.string().describe("new name").optional(),
+		description: z.string().describe("new description").optional(),
 	}),
 	"update_webset",
 );
@@ -94,7 +94,7 @@ const websetDeleteTool = createWebsetTool(
 	"Delete Webset",
 	"Delete a webset and all its contents.",
 	z.object({
-		id: z.string().describe("Webset ID"),
+		id: z.string().describe("webset id"),
 	}),
 	"delete_webset",
 );
@@ -105,9 +105,9 @@ const websetItemsListTool = createWebsetTool(
 	"List Webset Items",
 	"List items in a webset with optional pagination.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		limit: z.number().describe("Number of items to return").optional(),
-		offset: z.number().describe("Pagination offset").optional(),
+		webset_id: z.string().describe("webset id"),
+		limit: z.number().describe("max items").optional(),
+		offset: z.number().describe("offset").optional(),
 	}),
 	"list_webset_items",
 );
@@ -117,8 +117,8 @@ const websetItemGetTool = createWebsetTool(
 	"Get Webset Item",
 	"Get a specific item from a webset.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		item_id: z.string().describe("Item ID"),
+		webset_id: z.string().describe("webset id"),
+		item_id: z.string().describe("item id"),
 	}),
 	"get_item",
 );
@@ -129,8 +129,8 @@ const websetSearchCreateTool = createWebsetTool(
 	"Create Webset Search",
 	"Create a new search within a webset.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		query: z.string().describe("Search query"),
+		webset_id: z.string().describe("webset id"),
+		query: z.string().describe("search query"),
 	}),
 	"create_search",
 );
@@ -140,8 +140,8 @@ const websetSearchGetTool = createWebsetTool(
 	"Get Webset Search",
 	"Get the status and results of a webset search.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		search_id: z.string().describe("Search ID"),
+		webset_id: z.string().describe("webset id"),
+		search_id: z.string().describe("search id"),
 	}),
 	"get_search",
 );
@@ -151,8 +151,8 @@ const websetSearchCancelTool = createWebsetTool(
 	"Cancel Webset Search",
 	"Cancel a running webset search.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		search_id: z.string().describe("Search ID"),
+		webset_id: z.string().describe("webset id"),
+		search_id: z.string().describe("search id"),
 	}),
 	"cancel_search",
 );
@@ -163,9 +163,9 @@ const websetEnrichmentCreateTool = createWebsetTool(
 	"Create Enrichment",
 	"Create a new enrichment task for a webset.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		name: z.string().describe("Enrichment name"),
-		prompt: z.string().describe("Enrichment prompt"),
+		webset_id: z.string().describe("webset id"),
+		name: z.string().describe("enrichment name"),
+		prompt: z.string().describe("enrichment prompt"),
 	}),
 	"create_enrichment",
 );
@@ -175,8 +175,8 @@ const websetEnrichmentGetTool = createWebsetTool(
 	"Get Enrichment",
 	"Get the status and results of an enrichment task.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		enrichment_id: z.string().describe("Enrichment ID"),
+		webset_id: z.string().describe("webset id"),
+		enrichment_id: z.string().describe("enrichment id"),
 	}),
 	"get_enrichment",
 );
@@ -186,10 +186,10 @@ const websetEnrichmentUpdateTool = createWebsetTool(
 	"Update Enrichment",
 	"Update an enrichment's name or prompt.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		enrichment_id: z.string().describe("Enrichment ID"),
-		name: z.string().describe("New name").optional(),
-		prompt: z.string().describe("New prompt").optional(),
+		webset_id: z.string().describe("webset id"),
+		enrichment_id: z.string().describe("enrichment id"),
+		name: z.string().describe("new name").optional(),
+		prompt: z.string().describe("new prompt").optional(),
 	}),
 	"update_enrichment",
 );
@@ -199,8 +199,8 @@ const websetEnrichmentDeleteTool = createWebsetTool(
 	"Delete Enrichment",
 	"Delete an enrichment task.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		enrichment_id: z.string().describe("Enrichment ID"),
+		webset_id: z.string().describe("webset id"),
+		enrichment_id: z.string().describe("enrichment id"),
 	}),
 	"delete_enrichment",
 );
@@ -210,8 +210,8 @@ const websetEnrichmentCancelTool = createWebsetTool(
 	"Cancel Enrichment",
 	"Cancel a running enrichment task.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		enrichment_id: z.string().describe("Enrichment ID"),
+		webset_id: z.string().describe("webset id"),
+		enrichment_id: z.string().describe("enrichment id"),
 	}),
 	"cancel_enrichment",
 );
@@ -222,8 +222,8 @@ const websetMonitorCreateTool = createWebsetTool(
 	"Create Monitor",
 	"Create a monitoring task for a webset with optional webhook notifications.",
 	z.object({
-		webset_id: z.string().describe("Webset ID"),
-		webhook_url: z.string().describe("Webhook URL for notifications").optional(),
+		webset_id: z.string().describe("webset id"),
+		webhook_url: z.string().describe("webhook url").optional(),
 	}),
 	"create_monitor",
 );

package/src/extensibility/typebox.ts

@@ -21,7 +21,7 @@
  * `@sinclair/typebox` directly in their own package.
  */
 
-import { areJsonValuesEqual } from "@oh-my-pi/pi-ai/utils/schema";
+import { areJsonValuesEqual, zodToWireSchema } from "@oh-my-pi/pi-ai/utils/schema";
 import {
 	type ZodArray,
 	type ZodEnum,
@@ -104,19 +104,46 @@ interface ObjectOpts extends Meta {
 // Helpers
 // ---------------------------------------------------------------------------
 
+/**
+ * Stamp a non-enumerable `toJSON()` on a schema so `JSON.stringify(schema)`
+ * yields a clean draft 2020-12 JSON Schema — matching real TypeBox semantics
+ * where the schema object IS already a JSON Schema. Without this, an extension
+ * author who serialises the schema across any JSON boundary (worker
+ * postMessage, MCP transport, config persistence, network hop, structuredClone
+ * fallback) ships the raw Zod internals (`def`, `_zod`, object-shaped `enum`,
+ * `"type":"enum"`) — neither valid JSON Schema nor parseable Zod. See
+ * issue #1101 for the symptoms when this leaks into a tool's `input_schema`.
+ *
+ * Idempotent: re-stamping the same instance is a no-op.
+ */
+function wire<T extends ZodType>(schema: T): T {
+	if (!Object.hasOwn(schema as object, "toJSON")) {
+		Object.defineProperty(schema as object, "toJSON", {
+			value: function toJSON(this: ZodType) {
+				return zodToWireSchema(this);
+			},
+			enumerable: false,
+			writable: true,
+			configurable: true,
+		});
+	}
+	return schema;
+}
+
 function withMeta<T extends ZodType>(schema: T, opts: Meta | undefined): T {
-	if (!opts) return schema;
 	let out: ZodType = schema;
-	if (typeof opts.description === "string") out = out.describe(opts.description);
-	if ("default" in opts) out = out.default(opts.default as never) as unknown as ZodType;
+	if (opts) {
+		if (typeof opts.description === "string") out = out.describe(opts.description);
+		if ("default" in opts) out = out.default(opts.default as never) as unknown as ZodType;
 
-	const metadata: Record<string, unknown> = {};
-	for (const [key, value] of Object.entries(opts)) {
-		if (key === "description" || key === "default" || key === "additionalProperties") continue;
-		metadata[key] = value;
+		const metadata: Record<string, unknown> = {};
+		for (const key in opts) {
+			if (key === "description" || key === "default" || key === "additionalProperties") continue;
+			metadata[key] = opts[key];
+		}
+		if (Object.keys(metadata).length > 0) out = out.meta(metadata);
 	}
-	if (Object.keys(metadata).length > 0) out = out.meta(metadata);
-	return out as T;
+	return wire(out as T);
 }
 
 // ---------------------------------------------------------------------------
@@ -313,7 +340,8 @@ function tRecord<V extends ZodType>(key: ZodType, value: V, opts?: Meta): ZodTyp
 }
 
 function tOptional<E extends ZodType>(schema: E, _opts?: Meta): ZodOptional<E> {
-	return isOptional(schema) ? (schema as unknown as ZodOptional<E>) : (schema.optional() as ZodOptional<E>);
+	if (isOptional(schema)) return wire(schema as unknown as ZodOptional<E>);
+	return wire(schema.optional() as ZodOptional<E>);
 }
 
 function tNullable<E extends ZodType>(schema: E, opts?: Meta): ZodType {
@@ -322,27 +350,26 @@ function tNullable<E extends ZodType>(schema: E, opts?: Meta): ZodType {
 
 function tReadonly<E extends ZodType>(schema: E): E {
 	// TypeBox's `Type.Readonly` is purely a marker; runtime parsing is identical.
-	return schema;
+	return wire(schema);
 }
 
 function tPartial<P extends ZodRawShape>(obj: ZodObject<P>): ZodObject<P> {
-	return obj.partial() as unknown as ZodObject<P>;
+	return wire(obj.partial() as unknown as ZodObject<P>);
 }
 
 function tRequired<P extends ZodRawShape>(obj: ZodObject<P>): ZodObject<P> {
-	return obj.required() as unknown as ZodObject<P>;
+	return wire(obj.required() as unknown as ZodObject<P>);
 }
 
 function tPick<P extends ZodRawShape, K extends keyof P>(obj: ZodObject<P>, keys: readonly K[]): ZodObject<Pick<P, K>> {
 	const mask = Object.fromEntries(keys.map(k => [k as string, true]));
-	return obj.pick(mask as never) as unknown as ZodObject<Pick<P, K>>;
+	return wire(obj.pick(mask as never) as unknown as ZodObject<Pick<P, K>>);
 }
 
 function tOmit<P extends ZodRawShape, K extends keyof P>(obj: ZodObject<P>, keys: readonly K[]): ZodObject<Omit<P, K>> {
 	const mask = Object.fromEntries(keys.map(k => [k as string, true]));
-	return obj.omit(mask as never) as unknown as ZodObject<Omit<P, K>>;
+	return wire(obj.omit(mask as never) as unknown as ZodObject<Omit<P, K>>);
 }
-
 function tComposite(objects: readonly ZodObject<ZodRawShape>[], opts?: Meta): ZodObject<ZodRawShape> {
 	// `Type.Composite([...])` flattens every object schema into one object schema
 	// rather than producing an intersection. Mirror that via repeated `extend`.

package/src/goals/tools/goal-tool.ts

@@ -15,9 +15,9 @@ import { completionBudgetReport, remainingTokens } from "../runtime";
 import type { Goal, GoalStatus, GoalToolDetails } from "../state";
 
 const goalSchema = z.object({
-	op: z.union([z.literal("create"), z.literal("get"), z.literal("complete")]).describe("Goal operation."),
-	objective: z.string().describe("Goal objective. Required when op=create.").optional(),
-	token_budget: z.number().int().describe("Optional positive token budget. Only honored when op=create.").optional(),
+	op: z.enum(["create", "get", "complete"]).describe("goal operation"),
+	objective: z.string().describe("goal objective").optional(),
+	token_budget: z.number().int().describe("token budget").optional(),
 });
 
 export type GoalToolInput = z.infer<typeof goalSchema>;

package/src/index.ts

@@ -2,9 +2,6 @@ import { HookEditorComponent, HookInputComponent, HookSelectorComponent } from "
 
 // Core session management
 
-// TypeBox helper for string enums (convenience for custom tools)
-// Re-export from pi-ai which uses the correct enum-based schema format
-export { StringEnum } from "@oh-my-pi/pi-ai";
 // Re-export TUI components for custom tool rendering
 export { Container, Markdown, Spacer, Text } from "@oh-my-pi/pi-tui";
 // Logging