@xynogen/pix-9router 0.4.5 → 0.7.0

package/src/render.ts

@@ -1,151 +0,0 @@
-/**
- * render.ts — shared dim renderers for fetch/search tool output.
- *
- * fetch/search return external web content the model already consumed; the
- * full body is noise in the transcript. We render it dim (theme `dim` token)
- * so it reads like faded reasoning — present for inspection, visually quiet.
- * The tool title line stays normal so calls remain scannable.
- *
- * pi-tui's Text component is loaded via require() so the renderers degrade to
- * the default pi renderer (undefined-safe via cast) when pi-tui is absent.
- */
-
-import type {
-	AgentToolResult,
-	Theme,
-	ToolRenderResultOptions,
-} from "@earendil-works/pi-coding-agent";
-import type { Component } from "@earendil-works/pi-tui";
-import {
-	type CollapsedToolStatus,
-	formatCollapsedToolRow,
-	frameToolResult,
-	hideCollapsedToolCall,
-	unframeToolResult,
-} from "@xynogen/pix-pretty/utils";
-import { type CollapseState, tickCollapse } from "@xynogen/pix-runtime/collapse";
-
-interface TextLike extends Component {
-	setText(text: string): void;
-}
-
-// ToolRenderContext is not re-exported by pi-coding-agent; model the fields the
-// renderers actually read. Structurally compatible with the SDK context.
-interface RenderCtx {
-	lastComponent: Component | undefined;
-	isError: boolean;
-	state: Record<string, unknown>;
-	expanded: boolean;
-	invalidate: () => void;
-}
-
-interface CompactRendererConfig<TDetails> {
-	tool: string;
-	target: (details: TDetails) => string;
-	meta: (details: TDetails) => string;
-	status?: (details: TDetails) => CollapsedToolStatus;
-}
-
-type TextCtor = new (text?: string, padX?: number, padY?: number) => TextLike;
-
-// Preview cap is intentionally generous — dimming already de-emphasises the
-// body, so we only truncate to keep pathological multi-thousand-line dumps
-// from flooding the viewport. Expanding shows everything.
-const MAX_PREVIEW_LINES = 32;
-
-let TextComponent: TextCtor | undefined;
-try {
-	TextComponent = (require("@earendil-works/pi-tui") as { Text: TextCtor }).Text;
-} catch {
-	TextComponent = undefined;
-}
-
-function getText(lastComponent: Component | undefined): TextLike {
-	if (lastComponent && "setText" in lastComponent)
-		return unframeToolResult(lastComponent as TextLike);
-	// TextComponent is always present in the interactive TUI (pi-tui peer); the
-	// require() guard only matters for headless/test contexts where renderers
-	// are never invoked.
-	if (TextComponent) return new TextComponent("", 0, 0);
-	throw new Error("pi-tui Text component unavailable");
-}
-
-function allText(result: AgentToolResult<unknown>): string {
-	return (result.content ?? [])
-		.filter((block): block is { type: "text"; text: string } => block.type === "text")
-		.map((block) => block.text)
-		.join("\n");
-}
-
-function dimBody(body: string, theme: Theme, expanded: boolean): string {
-	const lines = body.split("\n");
-	const maxShow = expanded ? lines.length : MAX_PREVIEW_LINES;
-	const shown = lines.slice(0, maxShow);
-	const out = shown.map((line) => `  ${theme.fg("dim", line)}`);
-	const remaining = lines.length - maxShow;
-	if (remaining > 0) {
-		out.push(`  ${theme.fg("muted", `… ${remaining} more lines`)}`);
-	}
-	return out.join("\n");
-}
-
-/** Build a `renderCall` that shows `<title> <secondary arg>`. */
-export function makeRenderCall<TArgs>(title: string, pickArg: (args: TArgs) => string) {
-	return (args: TArgs, theme: Theme, ctx: RenderCtx): Component => {
-		const text = getText(ctx.lastComponent);
-		if (
-			hideCollapsedToolCall(ctx.state as CollapseState, ctx.expanded, (value) =>
-				text.setText(value),
-			)
-		)
-			return text;
-		const arg = pickArg(args);
-		text.setText(`${theme.fg("toolTitle", theme.bold(title))} ${theme.fg("dim", arg)}`);
-		return text;
-	};
-}
-
-/** Build a `renderResult` with a bounded detail preview and optional compact terminal row. */
-export function makeRenderResult<TDetails>(config?: CompactRendererConfig<TDetails>) {
-	return (
-		result: AgentToolResult<unknown>,
-		opts: ToolRenderResultOptions,
-		theme: Theme,
-		ctx: RenderCtx,
-	): Component => {
-		const text = getText(ctx.lastComponent);
-		const body = allText(result);
-
-		if (ctx.isError) {
-			text.setText(`  ${theme.fg("error", body || "Error")}`);
-			return opts.isPartial ? text : frameToolResult(text, theme, true);
-		}
-		if (!body.trim()) {
-			text.setText(`  ${theme.fg("muted", "(empty)")}`);
-			return opts.isPartial ? text : frameToolResult(text, theme, false);
-		}
-
-		const details = result.details as TDetails | undefined;
-		const isError = Boolean(details && config?.status?.(details) === "error");
-		if (
-			!opts.isPartial &&
-			config &&
-			details &&
-			tickCollapse(config.tool, ctx.state as CollapseState, ctx.invalidate, opts.expanded)
-		) {
-			text.setText(
-				formatCollapsedToolRow(
-					theme,
-					config.tool,
-					config.target(details),
-					config.meta(details),
-					config.status?.(details) ?? "success",
-				),
-			);
-			return text;
-		}
-
-		text.setText(dimBody(body, theme, opts.expanded));
-		return opts.isPartial ? text : frameToolResult(text, theme, isError);
-	};
-}

package/src/search.ts

@@ -1,282 +0,0 @@
-/**
- * search.ts — search tool via 9Router API, with curl fallback.
- *
- * Environment:
- *   ROUTER_API_BASE  — router API base URL (default: https://9router.example.com/v1)
- *   ROUTER_API_KEY   — bearer token for the router
- */
-
-import { StringEnum } from "@earendil-works/pi-ai";
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { Type } from "typebox";
-import { routerBaseUrl } from "./data.js";
-import { apiPost, auth, curl, isCancelled } from "./http.js";
-import { makeRenderCall, makeRenderResult } from "./render.js";
-
-type RouterOutcome = "running" | "fallback" | "success" | "cancelled" | "error";
-type SearchType = "web" | "news";
-
-export interface SearchResultDetails {
-	_type: "searchResult";
-	outcome: RouterOutcome;
-	query: string;
-	searchType: SearchType;
-	source?: "api" | "curl-fallback" | "failed";
-	resultCount?: number;
-}
-
-export interface SearchParams {
-	query: string;
-	search_type?: SearchType;
-	max_results?: number;
-}
-
-interface SearchResult {
-	content: { type: "text"; text: string }[];
-	details: SearchResultDetails;
-	isError?: boolean;
-}
-
-interface SearchDependencies {
-	apiPost: typeof apiPost;
-	curl: typeof curl;
-}
-
-interface SearchResultItem {
-	title?: string | null;
-	url?: string | null;
-	snippet?: string | null;
-	published_at?: string | null;
-	metadata?: { author?: string | null } | null;
-	author?: string | null;
-}
-
-/** Parse and format a raw exa search response with its authoritative result count. */
-export function parseSearchResults(raw: string): { text: string; count: number } {
-	let parsed: unknown;
-	try {
-		parsed = JSON.parse(raw);
-	} catch {
-		return { text: raw.slice(0, 20_000), count: 0 };
-	}
-
-	const obj = parsed as Record<string, unknown>;
-	const results = obj?.results;
-	if (!Array.isArray(results)) {
-		return { text: JSON.stringify(parsed, null, 2).slice(0, 20_000), count: 0 };
-	}
-
-	if (results.length === 0) return { text: "No results.", count: 0 };
-
-	const lines: string[] = [];
-	results.forEach((r: SearchResultItem, i: number) => {
-		const title = (r.title ?? "").trim() || r.url || "(untitled)";
-		lines.push(`${i + 1}. ${title}`);
-		if (r.url) lines.push(`   ${r.url}`);
-		const meta: string[] = [];
-		const author = r.author ?? r.metadata?.author;
-		if (author) meta.push(author);
-		if (r.published_at) meta.push(r.published_at.slice(0, 10));
-		if (meta.length > 0) lines.push(`   (${meta.join(" — ")})`);
-		const snippet = (r.snippet ?? "").trim();
-		if (snippet) lines.push(`   ${snippet.replace(/\s+/g, " ").slice(0, 300)}`);
-		lines.push("");
-	});
-
-	return {
-		text: lines.join("\n").trim().slice(0, 20_000),
-		count: results.length,
-	};
-}
-
-/** Format a raw exa search response as a compact markdown list. */
-export function formatSearchResults(raw: string): string {
-	return parseSearchResults(raw).text;
-}
-
-export async function executeSearch(
-	params: SearchParams,
-	signal?: AbortSignal,
-	onUpdate?: (result: SearchResult) => void,
-	dependencies: SearchDependencies = { apiPost, curl },
-): Promise<SearchResult> {
-	const max = Math.min(params.max_results ?? 5, 10);
-	const searchType = params.search_type ?? "web";
-	let apiMsg = "";
-
-	try {
-		onUpdate?.({
-			content: [{ type: "text", text: `Searching (${searchType}): ${params.query}...` }],
-			details: {
-				_type: "searchResult",
-				outcome: "running",
-				query: params.query,
-				searchType,
-			},
-		});
-
-		const raw = await dependencies.apiPost(
-			"/search",
-			{
-				model: "exa",
-				query: params.query,
-				search_type: searchType,
-				max_results: max,
-			},
-			signal,
-		);
-		const parsed = parseSearchResults(raw);
-
-		return {
-			content: [{ type: "text", text: parsed.text }],
-			details: {
-				_type: "searchResult",
-				outcome: "success",
-				query: params.query,
-				searchType,
-				source: "api",
-				resultCount: parsed.count,
-			},
-		};
-	} catch (apiErr: unknown) {
-		apiMsg = apiErr instanceof Error ? apiErr.message : String(apiErr);
-		if (isCancelled(apiErr, signal)) {
-			return {
-				content: [{ type: "text", text: `Search cancelled.\nAPI: ${apiMsg}` }],
-				details: {
-					_type: "searchResult",
-					outcome: "cancelled",
-					query: params.query,
-					searchType,
-				},
-			};
-		}
-		onUpdate?.({
-			content: [
-				{
-					type: "text",
-					text: `API failed: ${apiMsg}\nFalling back to curl...`,
-				},
-			],
-			details: {
-				_type: "searchResult",
-				outcome: "running",
-				query: params.query,
-				searchType,
-			},
-		});
-	}
-
-	try {
-		const body = JSON.stringify({
-			model: "exa",
-			query: params.query,
-			search_type: searchType,
-			max_results: max,
-		});
-		const raw = await dependencies.curl([
-			"-X",
-			"POST",
-			"-H",
-			"Content-Type: application/json",
-			...(auth() ? ["-H", `Authorization: Bearer ${auth()}`] : []),
-			"-d",
-			body,
-			`${routerBaseUrl()}/search`,
-		]);
-		const parsed = parseSearchResults(raw);
-
-		return {
-			content: [
-				{
-					type: "text",
-					text: `[FALLBACK — curl] API called via curl instead of fetch.\n\n${parsed.text.slice(0, 19_500)}`,
-				},
-			],
-			details: {
-				_type: "searchResult",
-				outcome: "fallback",
-				query: params.query,
-				searchType,
-				source: "curl-fallback",
-				resultCount: parsed.count,
-			},
-		};
-	} catch (curlErr: unknown) {
-		const curlMsg = curlErr instanceof Error ? curlErr.message : String(curlErr);
-		return {
-			content: [
-				{
-					type: "text",
-					text: `Search failed (both API and curl).\nAPI: ${apiMsg}\nCurl: ${curlMsg}`,
-				},
-			],
-			details: {
-				_type: "searchResult",
-				outcome: "error",
-				query: params.query,
-				searchType,
-				source: "failed",
-			},
-			isError: true,
-		};
-	}
-}
-
-export default function registerSearch(pi: ExtensionAPI): void {
-	pi.registerTool({
-		name: "search",
-		label: "Search",
-		renderShell: "self",
-		description: "Web or news search via exa. Returns title, url, and snippet for each result.",
-		promptSnippet:
-			"search(query, search_type, max_results?) — search_type: 'web' or 'news'. Defaults to 5 results, max 10.",
-		promptGuidelines: [
-			"search: search_type='web' for general web results, 'news' for recent news articles.",
-		],
-		renderCall: makeRenderCall<unknown>("search", (args) =>
-			String((args as Partial<SearchParams>).query ?? ""),
-		),
-		renderResult: (
-			(renderTerminal) => (result, options, theme, context) =>
-				renderTerminal(result, options, theme, {
-					...context,
-					isError: context.isError && options.expanded,
-				})
-		)(
-			makeRenderResult<SearchResultDetails>({
-				tool: "search",
-				target: (details) => `“${details.query}”`,
-				meta: (details) => {
-					if (details.outcome === "error") return "failed";
-					if (details.outcome === "cancelled") return "cancelled";
-					const count = details.resultCount ?? 0;
-					return `${count} ${count === 1 ? "result" : "results"} · ${details.searchType}`;
-				},
-				status: (details) =>
-					details.outcome === "error"
-						? "error"
-						: details.outcome === "fallback" || details.outcome === "cancelled"
-							? "warning"
-							: "success",
-			}),
-		),
-		parameters: Type.Object({
-			query: Type.String({ description: "Search query" }),
-			search_type: StringEnum(["web", "news"] as const, {
-				description:
-					'Required choice. Enter exactly "web" for general web results or "news" for recent news articles.',
-			}),
-			max_results: Type.Optional(
-				Type.Number({
-					description: "Max results (default 5, max 10)",
-					default: 5,
-				}),
-			),
-		}),
-
-		async execute(_toolCallId, params, signal, onUpdate) {
-			return executeSearch(params as SearchParams, signal, onUpdate);
-		},
-	});
-}