@xynogen/pix-9router 0.3.5 → 0.3.7

package/package.json

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

package/src/data.ts

@@ -41,6 +41,12 @@ export interface RouterModel {
 	max_tokens?: number;
 	maxTokens?: number;
 	owned_by?: string;
+	/** 9router v1 API nests these inside capabilities */
+	capabilities?: {
+		contextWindow?: number;
+		maxOutput?: number;
+		[key: string]: unknown;
+	};
 }
 
 interface RouterModelsResponse {

package/src/fetch.ts

@@ -264,6 +264,7 @@ export default function registerFetch(pi: ExtensionAPI): void {
 	pi.registerTool({
 		name: "fetch",
 		label: "Fetch",
+		renderShell: "self",
 		description: "Fetch a web page as markdown, text, or raw HTML via exa. Returns page content.",
 		promptSnippet:
 			"fetch(url, format, max_characters?) — format: 'markdown', 'text', or 'html'. Read page content via exa.",

package/src/format.test.ts

@@ -282,6 +282,17 @@ const renderTheme = {
 	bold: (text: string) => `*${text}*`,
 } as never;
 
+function captureRegisteredTool(register: (pi: never) => void): Record<string, unknown> {
+	let tool: Record<string, unknown> | undefined;
+	register({
+		registerTool(value: Record<string, unknown>) {
+			tool = value;
+		},
+	} as never);
+	if (!tool) throw new Error("tool was not registered");
+	return tool;
+}
+
 function renderRegisteredResult(
 	register: (pi: never) => void,
 	result: Record<string, unknown>,
@@ -313,6 +324,11 @@ function renderRegisteredResult(
 }
 
 describe("fetch and search compact renderers", () => {
+	it("use the self-rendered shell so status marks have no leading box padding", () => {
+		expect(captureRegisteredTool(registerFetch).renderShell).toBe("self");
+		expect(captureRegisteredTool(registerSearch).renderShell).toBe("self");
+	});
+
 	it("renders fallback and failure rows from metadata", () => {
 		const fetchRow = renderRegisteredResult(registerFetch, {
 			content: [{ type: "text", text: "fallback body" }],

package/src/provider.test.ts

@@ -1,48 +1,12 @@
 import { describe, expect, it } from "bun:test";
 import type { ModelsDevModel, RouterModel } from "./data.ts";
-
-// ── Re-export internal helpers for testing via module augmentation ────────────
-// provider.ts exports only the default fn; we test the pure mapping logic
-// inline here to avoid coupling tests to private internals.
-
-const DEFAULT_CONTEXT_WINDOW = 128_000;
-const DEFAULT_MAX_TOKENS = 16_384;
-
-const IMAGE_CAPABLE_PATTERNS = [/claude/i, /gpt-5/i, /gpt-4/i, /kimi-k2/i, /hy3/i];
-
-function getInputTypes(model: RouterModel, devModel?: ModelsDevModel): ("text" | "image")[] {
-	if (devModel?.modalities?.input) {
-		const inputs = devModel.modalities.input.filter(
-			(i): i is "text" | "image" => i === "text" || i === "image",
-		);
-		if (inputs.length > 0) return inputs;
-	}
-	const id = model.id ?? "";
-	if (IMAGE_CAPABLE_PATTERNS.some((p) => p.test(id))) return ["text", "image"];
-	return ["text"];
-}
-
-function getModelName(model: RouterModel, devModel?: ModelsDevModel): string {
-	return model.name || devModel?.name || model.id || "unknown";
-}
-
-function getContextWindow(model: RouterModel, devModel?: ModelsDevModel): number {
-	return (
-		model.context_window ||
-		model.contextWindow ||
-		devModel?.limit?.context ||
-		DEFAULT_CONTEXT_WINDOW
-	);
-}
-
-function getMaxTokens(model: RouterModel, devModel?: ModelsDevModel): number {
-	return model.max_tokens || model.maxTokens || devModel?.limit?.output || DEFAULT_MAX_TOKENS;
-}
-
-function getReasoning(model: RouterModel, devModel?: ModelsDevModel): boolean {
-	if (typeof devModel?.reasoning === "boolean") return devModel.reasoning;
-	return /reasoner|thinking|xhigh|high|max|pro|codex|opus|sonnet/i.test(model.id ?? "");
-}
+import {
+	getContextWindow,
+	getInputTypes,
+	getMaxTokens,
+	getModelName,
+	getReasoning,
+} from "./provider.ts";
 
 // ── getInputTypes ────────────────────────────────────────────────────────────
 
@@ -129,12 +93,37 @@ describe("getContextWindow", () => {
 		expect(getContextWindow({ id: "x", contextWindow: 64_000 })).toBe(64_000);
 	});
 
+	it("falls back to capabilities.contextWindow (9router v1 nested shape)", () => {
+		expect(getContextWindow({ id: "x", capabilities: { contextWindow: 1_000_000 } })).toBe(
+			1_000_000,
+		);
+	});
+
+	it("top-level fields beat capabilities.contextWindow", () => {
+		expect(
+			getContextWindow({
+				id: "x",
+				contextWindow: 64_000,
+				capabilities: { contextWindow: 1_000_000 },
+			}),
+		).toBe(64_000);
+	});
+
+	it("capabilities.contextWindow beats devModel.limit.context", () => {
+		expect(
+			getContextWindow(
+				{ id: "x", capabilities: { contextWindow: 1_000_000 } },
+				{ id: "x", limit: { context: 200_000 } },
+			),
+		).toBe(1_000_000);
+	});
+
 	it("falls back to devModel.limit.context", () => {
 		expect(getContextWindow({ id: "x" }, { id: "x", limit: { context: 200_000 } })).toBe(200_000);
 	});
 
 	it("falls back to DEFAULT_CONTEXT_WINDOW", () => {
-		expect(getContextWindow({ id: "x" })).toBe(DEFAULT_CONTEXT_WINDOW);
+		expect(getContextWindow({ id: "x" })).toBe(128_000);
 	});
 });
 
@@ -151,12 +140,40 @@ describe("getMaxTokens", () => {
 		expect(getMaxTokens({ id: "x", maxTokens: 8_192 })).toBe(8_192);
 	});
 
+	it("falls back to capabilities.maxOutput (9router v1 nested shape)", () => {
+		expect(getMaxTokens({ id: "x", capabilities: { maxOutput: 64_000 } })).toBe(64_000);
+	});
+
+	it("top-level fields beat capabilities.maxOutput", () => {
+		expect(getMaxTokens({ id: "x", maxTokens: 8_192, capabilities: { maxOutput: 64_000 } })).toBe(
+			8_192,
+		);
+	});
+
+	it("capabilities.maxOutput beats devModel.limit.output", () => {
+		expect(
+			getMaxTokens(
+				{ id: "x", capabilities: { maxOutput: 64_000 } },
+				{ id: "x", limit: { output: 16_000 } },
+			),
+		).toBe(64_000);
+	});
+
 	it("falls back to devModel.limit.output", () => {
 		expect(getMaxTokens({ id: "x" }, { id: "x", limit: { output: 16_000 } })).toBe(16_000);
 	});
 
 	it("falls back to DEFAULT_MAX_TOKENS", () => {
-		expect(getMaxTokens({ id: "x" })).toBe(DEFAULT_MAX_TOKENS);
+		expect(getMaxTokens({ id: "x" })).toBe(16_384);
+	});
+
+	it("handles the PR #9 regression case (qwen3-coder-flash nested capabilities)", () => {
+		const model: RouterModel = {
+			id: "alims-intl/qwen3-coder-flash-2025-07-28",
+			capabilities: { contextWindow: 1_000_000, maxOutput: 64_000 },
+		};
+		expect(getContextWindow(model)).toBe(1_000_000);
+		expect(getMaxTokens(model)).toBe(64_000);
 	});
 });
 

package/src/provider.ts

@@ -26,7 +26,7 @@ const ZERO_COST = {
 // Fallback pattern-based detection if models.dev lookup fails
 const IMAGE_CAPABLE_PATTERNS = [/claude/i, /gpt-5/i, /gpt-4/i, /kimi-k2/i, /hy3/i];
 
-function getInputTypes(model: RouterModel, devModel?: ModelsDevModel): ("text" | "image")[] {
+export function getInputTypes(model: RouterModel, devModel?: ModelsDevModel): ("text" | "image")[] {
 	if (devModel?.modalities?.input) {
 		const inputs = devModel.modalities.input.filter(
 			(i): i is "text" | "image" => i === "text" || i === "image",
@@ -38,24 +38,31 @@ function getInputTypes(model: RouterModel, devModel?: ModelsDevModel): ("text" |
 	return ["text"];
 }
 
-function getModelName(model: RouterModel, devModel?: ModelsDevModel): string {
+export function getModelName(model: RouterModel, devModel?: ModelsDevModel): string {
 	return model.name || devModel?.name || model.id || "unknown";
 }
 
-function getContextWindow(model: RouterModel, devModel?: ModelsDevModel): number {
+export function getContextWindow(model: RouterModel, devModel?: ModelsDevModel): number {
 	return (
 		model.context_window ||
 		model.contextWindow ||
+		model.capabilities?.contextWindow ||
 		devModel?.limit?.context ||
 		DEFAULT_CONTEXT_WINDOW
 	);
 }
 
-function getMaxTokens(model: RouterModel, devModel?: ModelsDevModel): number {
-	return model.max_tokens || model.maxTokens || devModel?.limit?.output || DEFAULT_MAX_TOKENS;
+export function getMaxTokens(model: RouterModel, devModel?: ModelsDevModel): number {
+	return (
+		model.max_tokens ||
+		model.maxTokens ||
+		model.capabilities?.maxOutput ||
+		devModel?.limit?.output ||
+		DEFAULT_MAX_TOKENS
+	);
 }
 
-function getReasoning(model: RouterModel, devModel?: ModelsDevModel): boolean {
+export function getReasoning(model: RouterModel, devModel?: ModelsDevModel): boolean {
 	if (typeof devModel?.reasoning === "boolean") return devModel.reasoning;
 	return /reasoner|thinking|xhigh|high|max|pro|codex|opus|sonnet/i.test(model.id ?? "");
 }

package/src/search.ts

@@ -293,6 +293,7 @@ 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.",

package/src/transcribe.test.ts

@@ -416,9 +416,9 @@ const theme = {
 } as never;
 
 function captureTranscribeRenderer() {
-	let tool: Record<string, (...args: never[]) => unknown> | undefined;
+	let tool: Record<string, unknown> | undefined;
 	registerTranscribe({
-		registerTool(value: Record<string, (...args: never[]) => unknown>) {
+		registerTool(value: Record<string, unknown>) {
 			tool = value;
 		},
 	} as never);
@@ -440,7 +440,8 @@ function renderTranscribe(
 		invalidate: () => {},
 	};
 	const tool = captureTranscribeRenderer();
-	tool.renderResult?.(result as never, { expanded, isPartial: false } as never, theme, {
+	const renderResult = tool.renderResult as ((...args: never[]) => unknown) | undefined;
+	renderResult?.(result as never, { expanded, isPartial: false } as never, theme, {
 		lastComponent: component,
 		isError,
 		state: { collapsed: true },
@@ -451,6 +452,10 @@ function renderTranscribe(
 }
 
 describe("transcribe compact renderer", () => {
+	it("uses the self-rendered shell so the status mark has no leading box padding", () => {
+		expect(captureTranscribeRenderer().renderShell).toBe("self");
+	});
+
 	it("summarizes inline and written transcripts", () => {
 		const inline = renderTranscribe({
 			content: [{ type: "text", text: "full transcript" }],

package/src/transcribe.ts

@@ -358,6 +358,7 @@ export default function registerTranscribe(pi: ExtensionAPI): void {
 	pi.registerTool({
 		name: "transcribe",
 		label: "Transcribe",
+		renderShell: "self",
 		description:
 			"Convert speech to text. Transcribes an audio file using the 9Router audio transcription API (Deepgram Nova 3). Optionally writes the full text to a file on disk.",
 		promptSnippet: