pi-agent-browser-native 0.2.45 → 0.2.46

package/CHANGELOG.md

@@ -4,6 +4,22 @@
 
 No changes yet.
 
+## 0.2.46 - 2026-06-07
+
+### Changed
+
+- Reduced native extension startup cost by replacing heavy top-level Pi and TypeBox runtime imports with lightweight local schema and event helpers while preserving the public tool schemas.
+- Kept custom browser tool rendering compact without importing Pi's full coding-agent runtime during extension load.
+
+### Fixed
+
+- Fixed issue #84 by cutting local cold extension import plus factory registration from roughly 1.1 seconds to roughly 76 milliseconds in checkout measurements.
+- Stabilized timeout partial-progress diagnostics so post-navigation timeouts are recognized from later completed-step evidence even when live URL recovery is unavailable under load.
+
+### Validation
+
+- Added a cold-start budget test for the real extension entrypoint, schema parity coverage against the canonical TypeBox/StringEnum builders, rendering coverage for JSON highlighting plus collapsed-output expand hints, and stabilized deterministic dogfood smoke by avoiding a rapid Windows browser close/relaunch race.
+
 ## 0.2.45 - 2026-06-06
 
 ### Added

package/extensions/agent-browser/index.ts

@@ -10,11 +10,10 @@ import type { ChildProcess } from "node:child_process";
 import { dirname, join, resolve } from "node:path";
 import { fileURLToPath } from "node:url";
 
-import {
-	isToolCallEventType,
-	type AgentToolResult,
-	type ExtensionAPI,
-	type ExtensionContext,
+import type {
+	AgentToolResult,
+	ExtensionAPI,
+	ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
 import { Text } from "@earendil-works/pi-tui";
 import {
@@ -91,7 +90,7 @@ import {
 	restoreElectronLaunchRecordsFromBranch,
 	type ElectronLaunchRecord,
 } from "./lib/orchestration/electron-host/index.js";
-import { buildValidationFailureResult, resolveAgentBrowserInput } from "./lib/orchestration/input-plan.js";
+import { buildValidationFailureResult, resolveAgentBrowserInput, type AgentBrowserExecuteParams } from "./lib/orchestration/input-plan.js";
 import { applyAgentBrowserOutputPath } from "./lib/orchestration/output-file.js";
 import type { NetworkRouteRecord } from "./lib/results/contracts.js";
 import type { SessionArtifactManifest } from "./lib/results/contracts.js";
@@ -131,6 +130,16 @@ import {
 
 const DEFAULT_SESSION_MODE = "auto" as const;
 
+type BashToolCallLike = {
+	input: { command: string };
+	toolName: "bash";
+};
+
+function isBashToolCallEvent(event: unknown): event is BashToolCallLike {
+	if (!isRecord(event) || event.toolName !== "bash" || !isRecord(event.input)) return false;
+	return typeof event.input.command === "string";
+}
+
 type OwnedManagedSession = {
 	branchOwned: boolean;
 	cwd: string;
@@ -753,7 +762,7 @@ export default function agentBrowserExtension(pi: ExtensionAPI) {
 	pi.on("tool_call", async (event, ctx) => {
 		const promptPolicy = buildPromptPolicy(getLatestUserPrompt(ctx.sessionManager.getBranch()));
 		if (
-			isToolCallEventType("bash", event) &&
+			isBashToolCallEvent(event) &&
 			!promptPolicy.allowLegacyAgentBrowserBash &&
 			looksLikeDirectAgentBrowserBash(event.input.command) &&
 			!isHarmlessAgentBrowserInspectionCommand(event.input.command) &&
@@ -789,7 +798,7 @@ export default function agentBrowserExtension(pi: ExtensionAPI) {
 			component.setState(formatAgentBrowserRenderResult(result, options, theme, context.isError), options.expanded, theme);
 			return component;
 		},
-		async execute(_toolCallId, params, signal, onUpdate, ctx) {
+		async execute(_toolCallId, params: AgentBrowserExecuteParams, signal, onUpdate, ctx) {
 			const promptPolicy = buildPromptPolicy(getLatestUserPrompt(ctx.sessionManager.getBranch()));
 			const outputPath = isRecord(params) && typeof params.outputPath === "string" ? params.outputPath : undefined;
 			const resolvedInput = resolveAgentBrowserInput({

package/extensions/agent-browser/lib/input-modes/params.ts

@@ -4,8 +4,8 @@
  * Scope: Schema-only; behavioral validation lives in the mode compilers.
  */
 
-import { StringEnum } from "@earendil-works/pi-ai";
-import { Type } from "typebox";
+import { JsonSchema, type JsonSchemaBuilder } from "../json-schema.js";
+import { StringEnum as localStringEnum, type StringEnumBuilder } from "../string-enum-schema.js";
 
 import {
 	ELECTRON_DISCOVERY_DEFAULT_MAX_RESULTS,
@@ -23,7 +23,11 @@ import {
 	SOURCE_LOOKUP_MAX_WORKSPACE_FILES,
 } from "./types.js";
 
-export const AGENT_BROWSER_PARAMS = Type.Object({
+export function createAgentBrowserParamsSchema(
+	Type: JsonSchemaBuilder = JsonSchema,
+	StringEnum: StringEnumBuilder = localStringEnum,
+) {
+	return Type.Object({
 
 	args: Type.Optional(
 		Type.Array(Type.String({ description: "Exact agent-browser CLI arguments, excluding the binary name. Do not pass --json; the wrapper injects it. First-call recipe: open → snapshot -i → click/fill @eN → snapshot -i." }), {
@@ -195,4 +199,7 @@ export const AGENT_BROWSER_PARAMS = Type.Object({
 			default: DEFAULT_SESSION_MODE,
 		}),
 	),
-}, { additionalProperties: false });
+	}, { additionalProperties: false });
+}
+
+export const AGENT_BROWSER_PARAMS = createAgentBrowserParamsSchema();

package/extensions/agent-browser/lib/json-schema.ts

@@ -0,0 +1,73 @@
+/**
+ * Purpose: Build the small JSON Schema subset used by Pi tool schemas without importing TypeBox at runtime.
+ * Responsibilities: Preserve plain JSON Schema objects Pi consumes while keeping extension startup cheap.
+ * Scope: Schema construction only; runtime validation still belongs to Pi and the tool input compilers.
+ */
+
+import type { TSchema, TSchemaOptions, TUnsafe } from "typebox";
+
+const OPTIONAL_SCHEMA = Symbol("pi-agent-browser-optional-schema");
+
+type SchemaObject = TSchema & { [OPTIONAL_SCHEMA]?: true };
+type SchemaProperties = Record<string, TSchema>;
+
+function withOptions(schema: Record<string, unknown>, options?: TSchemaOptions): TSchema {
+	return { ...schema, ...(options ?? {}) } as TSchema;
+}
+
+function literalType(value: unknown): "boolean" | "number" | "string" | undefined {
+	const valueType = typeof value;
+	return valueType === "string" || valueType === "number" || valueType === "boolean" ? valueType : undefined;
+}
+
+function propertySchema(schema: TSchema): TSchema {
+	const clone = { ...(schema as SchemaObject & Record<PropertyKey, unknown>) };
+	delete clone[OPTIONAL_SCHEMA];
+	return clone as TSchema;
+}
+
+export const JsonSchema = {
+	Array(items: TSchema, options?: TSchemaOptions): TSchema {
+		return withOptions({ type: "array", items }, options);
+	},
+	Boolean(options?: TSchemaOptions): TSchema {
+		return withOptions({ type: "boolean" }, options);
+	},
+	Integer(options?: TSchemaOptions): TSchema {
+		return withOptions({ type: "integer" }, options);
+	},
+	Literal(value: unknown, options?: TSchemaOptions): TSchema {
+		const type = literalType(value);
+		return withOptions(type ? { type, const: value } : { const: value }, options);
+	},
+	Number(options?: TSchemaOptions): TSchema {
+		return withOptions({ type: "number" }, options);
+	},
+	Object(properties: SchemaProperties, options?: TSchemaOptions): TSchema {
+		const required = globalThis.Object.entries(properties)
+			.filter(([, schema]) => (schema as SchemaObject)[OPTIONAL_SCHEMA] !== true)
+			.map(([key]) => key);
+		return withOptions({
+			type: "object",
+			properties: globalThis.Object.fromEntries(
+				globalThis.Object.entries(properties).map(([key, schema]) => [key, propertySchema(schema)]),
+			),
+			...(required.length > 0 ? { required } : {}),
+		}, options);
+	},
+	Optional(schema: TSchema): TSchema {
+		return { ...(schema as SchemaObject), [OPTIONAL_SCHEMA]: true } as TSchema;
+	},
+	String(options?: TSchemaOptions): TSchema {
+		return withOptions({ type: "string" }, options);
+	},
+	Union(types: TSchema[], options?: TSchemaOptions): TSchema {
+		return withOptions({ anyOf: types }, options);
+	},
+	Unsafe<Value>(schema: TSchema): TUnsafe<Value> {
+		return schema as TUnsafe<Value>;
+	},
+};
+
+export type JsonSchemaBuilder = typeof JsonSchema;
+export type { TSchema, TSchemaOptions, TUnsafe };

package/extensions/agent-browser/lib/orchestration/browser-run/diagnostics.ts

@@ -824,6 +824,12 @@ function buildTimeoutProgressSteps(options: {
 			step.reason = "Later step completion evidence indicates the batch advanced past this step before timeout.";
 		}
 	}
+	for (const step of progressSteps) {
+		const command = step.args[0];
+		if (step.status === "completed" && (isOpenNavigationCommand(command) || command === "pushstate")) {
+			lastCompletedNavigationIndex = Math.max(lastCompletedNavigationIndex ?? 0, step.index);
+		}
+	}
 	for (const step of progressSteps) {
 		if (step.status === "completed") continue;
 		if (!retryStep) {

package/extensions/agent-browser/lib/pi-tool-rendering.ts

@@ -1,6 +1,5 @@
 import type { AgentToolResult, Theme, ToolResultEvent } from "@earendil-works/pi-coding-agent";
-import { highlightCode, keyHint } from "@earendil-works/pi-coding-agent";
-import { Text, truncateToWidth } from "@earendil-works/pi-tui";
+import { getKeybindings, Text, truncateToWidth } from "@earendil-works/pi-tui";
 
 import {
 	compileAgentBrowserElectron,
@@ -16,6 +15,7 @@ import { redactInvocationArgs } from "./runtime.js";
 const TUI_INVOCATION_PREVIEW_MAX_CHARS = 160;
 const TUI_COLLAPSED_OUTPUT_MAX_LINES = 12;
 const ANSI_CONTROL_SEQUENCE_PATTERN = /\x1B(?:\][^\x07\x1B]*(?:\x07|\x1B\\)|\[[0-?]*[ -/]*[@-~]|P[^\x1B]*(?:\x1B\\)|_[^\x1B]*(?:\x1B\\)|\^[^\x1B]*(?:\x1B\\)|[@-Z\\-_])/g;
+const JSON_TOKEN_PATTERN = /"(?:\\.|[^"\\])*"(?=\s*:)|"(?:\\.|[^"\\])*"|-?\d+(?:\.\d+)?(?:[eE][+-]?\d+)?|true|false|null|[{}\[\],:]/g;
 const UNSAFE_DISPLAY_CONTROL_PATTERN = /[\x00-\x08\x0B\x0C\x0E-\x1F\x7F\x80-\x9F]/g;
 
 function sanitizeDisplayText(value: string): string {
@@ -48,6 +48,26 @@ function isJsonDocumentText(value: string): boolean {
 	}
 }
 
+function colorizeJsonLine(line: string, theme: Theme): string {
+	let output = "";
+	let cursor = 0;
+	for (const match of line.matchAll(JSON_TOKEN_PATTERN)) {
+		const token = match[0];
+		const index = match.index ?? 0;
+		output += line.slice(cursor, index);
+		const color = token.startsWith('"')
+			? /"\s*$/.test(token) && line.slice(index + token.length).trimStart().startsWith(":")
+				? "syntaxVariable"
+				: "syntaxString"
+			: /^[{}\[\],:]$/.test(token)
+				? "syntaxPunctuation"
+				: "syntaxType";
+		output += theme.fg(color, token);
+		cursor = index + token.length;
+	}
+	return output + line.slice(cursor);
+}
+
 function getPrimaryTextContent(result: AgentToolResult<unknown>): string {
 	const textContent = result.content.find((item) => item.type === "text");
 	return textContent?.type === "text" ? textContent.text : "";
@@ -57,23 +77,24 @@ function colorizeToolOutputLines(outputText: string, theme: Theme, isError: bool
 	const normalizedLines = trimTrailingBlankLines(replaceTabsForDisplay(sanitizeDisplayText(outputText)).split("\n"));
 	const normalizedText = normalizedLines.join("\n");
 	if (normalizedText.length === 0) return [];
-	if (isJsonDocumentText(normalizedText)) {
-		return highlightCode(normalizedText, "json");
-	}
+	const isJsonDocument = !isError && isJsonDocumentText(normalizedText);
 	return normalizedLines.map((line) => {
 		if (line.length === 0) {
 			return "";
 		}
+		if (isJsonDocument) return colorizeJsonLine(line, theme);
 		return isError ? theme.fg("error", line) : theme.fg("toolOutput", line);
 	});
 }
 
 function formatExpandHint(theme: Theme): string {
 	try {
-		return keyHint("app.tools.expand", "to expand");
+		const [key] = getKeybindings().getKeys("app.tools.expand" as never);
+		if (key) return `${theme.fg("dim", key)} ${theme.fg("muted", "to expand")}`;
 	} catch {
-		return `${theme.fg("dim", "ctrl+o")} ${theme.fg("muted", "to expand")}`;
+		// Fall through to the built-in default key when coding-agent keybindings are unavailable.
 	}
+	return `${theme.fg("dim", "ctrl+o")} ${theme.fg("muted", "to expand")}`;
 }
 
 function formatVisualTruncationNotice(remainingLines: number, totalLines: number, theme: Theme, width: number): string {

package/extensions/agent-browser/lib/string-enum-schema.ts

@@ -0,0 +1,20 @@
+/**
+ * Purpose: Build compact JSON-schema string enums without importing pi runtime helpers.
+ * Responsibilities: Mirror pi-ai StringEnum's `{ type: "string", enum: [...] }` shape while keeping extension startup imports light.
+ * Scope: Schema construction only.
+ */
+
+import { JsonSchema, type TSchemaOptions, type TUnsafe } from "./json-schema.js";
+
+export type StringEnumBuilder = typeof StringEnum;
+
+export function StringEnum<const Values extends readonly string[]>(
+	values: Values,
+	options?: TSchemaOptions,
+): TUnsafe<Values[number]> {
+	return JsonSchema.Unsafe<Values[number]>({
+		type: "string",
+		enum: [...values],
+		...options,
+	});
+}

package/extensions/agent-browser/lib/web-search.ts

@@ -4,9 +4,8 @@
  * Scope: Live web search only; browser automation remains in the `agent_browser` tool.
  */
 
-import { StringEnum } from "@earendil-works/pi-ai";
-import { defineTool } from "@earendil-works/pi-coding-agent";
-import { Type } from "typebox";
+import { JsonSchema, type JsonSchemaBuilder } from "./json-schema.js";
+import { StringEnum as localStringEnum, type StringEnumBuilder } from "./string-enum-schema.js";
 import {
 	DEFAULT_WEB_SEARCH_PROVIDER,
 	WEB_SEARCH_PROVIDERS,
@@ -119,8 +118,12 @@ export interface WebSearchProviderAdapter<Request = unknown, Response = unknown>
 	provider: WebSearchProvider;
 }
 
-export const AgentBrowserWebSearchParams = Type.Object(
-	{
+export function createAgentBrowserWebSearchParamsSchema(
+	Type: JsonSchemaBuilder = JsonSchema,
+	StringEnum: StringEnumBuilder = localStringEnum,
+) {
+	return Type.Object(
+		{
 		query: Type.String({
 			minLength: 1,
 			description: "Search query to run with the configured Exa or Brave web search provider.",
@@ -172,9 +175,12 @@ export const AgentBrowserWebSearchParams = Type.Object(
 				description: "Optional freshness window: pd=past day, pw=past week, pm=past month, py=past year.",
 			}),
 		),
-	},
-	{ additionalProperties: false },
-);
+		},
+		{ additionalProperties: false },
+	);
+}
+
+export const AgentBrowserWebSearchParams = createAgentBrowserWebSearchParamsSchema();
 
 const HTML_ENTITY_REPLACEMENTS: Readonly<Record<string, string>> = {
 	amp: "&",
@@ -644,9 +650,21 @@ function buildMissingCredentialError(provider: WebSearchProviderParam): string {
 	return "No Exa or Brave web search credential resolved. Configure webSearch.exaApiKey or webSearch.braveApiKey, or load EXA_API_KEY/BRAVE_API_KEY in the runtime environment.";
 }
 
+type AgentBrowserWebSearchParamsInput = {
+	country?: string;
+	count?: number;
+	freshness?: SearchFreshness;
+	offset?: number;
+	provider?: WebSearchProviderParam;
+	query: string;
+	safesearch?: "off" | "moderate" | "strict";
+	searchLang?: string;
+	searchType?: ExaSearchType;
+};
+
 export function createAgentBrowserWebSearchTool(configState: AgentBrowserConfigState) {
 	const requestGate = new WebSearchRequestGate();
-	return defineTool({
+	return {
 		name: AGENT_BROWSER_WEB_SEARCH_TOOL_NAME,
 		label: "Agent Browser Web Search",
 		description: `Search the web with Exa or Brave when configured. Returns up to ${MAX_SEARCH_RESULT_COUNT} concise web results.`,
@@ -659,7 +677,7 @@ export function createAgentBrowserWebSearchTool(configState: AgentBrowserConfigS
 			"After using agent_browser_web_search, cite result URLs in the final answer when web evidence informed the answer.",
 		],
 		parameters: AgentBrowserWebSearchParams,
-		async execute(_toolCallId, params, signal) {
+		async execute(_toolCallId: string, params: AgentBrowserWebSearchParamsInput, signal?: AbortSignal) {
 			if (!configState.webSearchEnabled) {
 				throw new Error("agent_browser_web_search is disabled by pi-agent-browser-native config.");
 			}
@@ -695,9 +713,9 @@ export function createAgentBrowserWebSearchTool(configState: AgentBrowserConfigS
 				results: normalized.results,
 			};
 			return {
-				content: [{ type: "text", text: formatSearchResults(adapter.provider, normalized.returnedQuery, normalized.results) }],
+				content: [{ type: "text" as const, text: formatSearchResults(adapter.provider, normalized.returnedQuery, normalized.results) }],
 				details,
 			};
 		},
-	});
+	};
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-agent-browser-native",
-  "version": "0.2.45",
+  "version": "0.2.46",
   "description": "pi extension that exposes agent-browser as a native tool for browser automation",
   "type": "module",
   "author": "Mitch Fultz (https://github.com/fitchmultz)",