@xynogen/pix-9router 0.1.1 → 0.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@xynogen/pix-9router",
-	"version": "0.1.1",
+	"version": "0.2.0",
 	"description": "Pi extension — 9Router provider + fetch/search tools via router API",
 	"type": "module",
 	"main": "src/index.ts",

package/src/fetch.ts

@@ -11,6 +11,7 @@ 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 { makeRenderCall, makeRenderResult } from "./render.js";
 
 const REQUEST_TIMEOUT_MS = 30_000;
 
@@ -119,6 +120,8 @@ export default function registerFetch(pi: ExtensionAPI): void {
 			"Set max_characters to cap response size (default 1000, 0 = unlimited). Use 5000-10000 for typical pages.",
 			"Always prefer fetch over raw curl/browser for reading page content.",
 		],
+		renderCall: makeRenderCall("fetch", (args) => String(args.url ?? "")),
+		renderResult: makeRenderResult(),
 		parameters: Type.Object({
 			url: Type.String({ description: "URL to fetch" }),
 			format: StringEnum(["markdown", "text", "html"] as const, {

package/src/render.test.ts

@@ -0,0 +1,78 @@
+import { expect, test } from "bun:test";
+import { makeRenderCall, makeRenderResult } from "./render.js";
+
+const theme = {
+	fg: (t: string, s: string) => `[${t}]${s}`,
+	bold: (s: string) => `*${s}*`,
+} as never;
+
+function stubComponent() {
+	let captured = "";
+	const component = {
+		setText: (s: string) => {
+			captured = s;
+		},
+		render: () => [],
+		invalidate: () => {},
+	};
+	return { component, get: () => captured };
+}
+
+test("renderCall shows title + accent arg", () => {
+	const stub = stubComponent();
+	const rc = makeRenderCall<{ url: string }>("fetch", (a) => a.url);
+	rc({ url: "https://x.com" }, theme, {
+		lastComponent: stub.component as never,
+		isError: false,
+	});
+	expect(stub.get()).toBe("[toolTitle]*fetch* [accent]https://x.com");
+});
+
+test("renderResult dims every line", () => {
+	const stub = stubComponent();
+	const rr = makeRenderResult();
+	rr(
+		{ content: [{ type: "text", text: "a\nb" }] } as never,
+		{ expanded: true, isPartial: false },
+		theme,
+		{ lastComponent: stub.component as never, isError: false },
+	);
+	expect(stub.get()).toBe("  [dim]a\n  [dim]b");
+});
+
+test("renderResult caps preview when not expanded", () => {
+	const stub = stubComponent();
+	const rr = makeRenderResult();
+	const body = Array.from({ length: 45 }, (_, i) => `L${i}`).join("\n");
+	rr(
+		{ content: [{ type: "text", text: body }] } as never,
+		{ expanded: false, isPartial: false },
+		theme,
+		{ lastComponent: stub.component as never, isError: false },
+	);
+	expect(stub.get()).toContain("… 5 more lines");
+});
+
+test("renderResult renders empty body marker", () => {
+	const stub = stubComponent();
+	const rr = makeRenderResult();
+	rr(
+		{ content: [{ type: "text", text: "   " }] } as never,
+		{ expanded: false, isPartial: false },
+		theme,
+		{ lastComponent: stub.component as never, isError: false },
+	);
+	expect(stub.get()).toBe("  [dim](empty)");
+});
+
+test("renderResult uses error token on failure", () => {
+	const stub = stubComponent();
+	const rr = makeRenderResult();
+	rr(
+		{ content: [{ type: "text", text: "boom" }] } as never,
+		{ expanded: false, isPartial: false },
+		theme,
+		{ lastComponent: stub.component as never, isError: true },
+	);
+	expect(stub.get()).toBe("  [error]boom");
+});

package/src/render.ts

@@ -0,0 +1,111 @@
+/**
+ * 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";
+
+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;
+}
+
+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 = 40;
+
+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 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 firstText(result: AgentToolResult<unknown>): string {
+	const block = result.content?.find((c) => c.type === "text");
+	return (block as { text?: string } | undefined)?.text ?? "";
+}
+
+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("dim", `… ${remaining} more lines`)}`);
+	}
+	return out.join("\n");
+}
+
+/** Build a `renderCall` that shows `<title> <accent arg>`. */
+export function makeRenderCall<TArgs>(
+	title: string,
+	pickArg: (args: TArgs) => string,
+) {
+	return (args: TArgs, theme: Theme, ctx: RenderCtx): Component => {
+		const text = getText(ctx.lastComponent);
+		const arg = pickArg(args);
+		text.setText(
+			`${theme.fg("toolTitle", theme.bold(title))} ${theme.fg("accent", arg)}`,
+		);
+		return text;
+	};
+}
+
+/** Build a `renderResult` that dims the whole result body. */
+export function makeRenderResult() {
+	return (
+		result: AgentToolResult<unknown>,
+		opts: ToolRenderResultOptions,
+		theme: Theme,
+		ctx: RenderCtx,
+	): Component => {
+		const text = getText(ctx.lastComponent);
+		const body = firstText(result);
+
+		if (ctx.isError) {
+			text.setText(`  ${theme.fg("error", body || "Error")}`);
+			return text;
+		}
+		if (!body.trim()) {
+			text.setText(`  ${theme.fg("dim", "(empty)")}`);
+			return text;
+		}
+
+		text.setText(dimBody(body, theme, opts.expanded));
+		return text;
+	};
+}

package/src/search.ts

@@ -11,6 +11,7 @@ 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 { makeRenderCall, makeRenderResult } from "./render.js";
 
 const REQUEST_TIMEOUT_MS = 30_000;
 
@@ -136,6 +137,8 @@ export default function registerSearch(pi: ExtensionAPI): void {
 			"Set search_type='web' for general web results, search_type='news' for recent news articles.",
 			"Always prefer search over raw curl/browser requests for information lookup.",
 		],
+		renderCall: makeRenderCall("search", (args) => String(args.query ?? "")),
+		renderResult: makeRenderResult(),
 		parameters: Type.Object({
 			query: Type.String({ description: "Search query" }),
 			search_type: StringEnum(["web", "news"] as const, {