@xynogen/pix-models 0.1.26 → 0.1.31

package/README.md

@@ -4,7 +4,7 @@ Pi extension — enhanced `/models` picker with coding score/rank.
 
 ## What it does
 
-Registers a `/models` slash command that replaces Pi's built-in `/model` selector with a richer TUI picker. Each row shows the model id, context window, per-million-token cost, and a coding-focused score/rank (with star bar) when available. The list is sorted by coding score (best first), then alphabetically for unscored models. Fuzzy search filters the list as you type. Selecting a model switches the active model for the session. Model metadata is sourced from `~/.cache/pi/` via `pix-data`; the coding score/rank is computed locally from the modelgrep catalog (best = #1). Requires `@xynogen/pix-data` as a dependency.
+Registers a `/models` slash command that replaces Pi's built-in `/model` selector with a richer TUI picker. Each row shows the model id, context window, per-million-token cost, and a coding-focused score/rank (with star bar) when available. The list is sorted by coding score (best first), then alphabetically for unscored models. Fuzzy search filters the list as you type. Left/right changes the thinking level (`off` → `minimal` → `low` → `medium` → `high` → `xhigh`), with the effective level shown live in the picker header. Selecting a model switches the active model for the session. Model metadata is sourced from `~/.cache/pi/` via `pix-data`; the coding score/rank is computed locally from the modelgrep catalog (best = #1). Requires `@xynogen/pix-data` as a dependency.
 
 ## Install
 

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@xynogen/pix-models",
-	"version": "0.1.26",
+	"version": "0.1.31",
 	"description": "Pi extension — enhanced /models picker with BenchLM ranks",
 	"type": "module",
 	"main": "src/index.ts",
@@ -38,9 +38,9 @@
 		"access": "public"
 	},
 	"dependencies": {
-		"@xynogen/pix-data": "^0.4.0",
-		"@xynogen/pix-pretty": "^1.7.9",
-		"@xynogen/pix-runtime": "^0.4.0"
+		"@xynogen/pix-data": "^0.4.3",
+		"@xynogen/pix-pretty": "^1.11.2",
+		"@xynogen/pix-runtime": "^0.5.3"
 	},
 	"peerDependencies": {
 		"@earendil-works/pi-coding-agent": "*",

package/src/models.test.ts

@@ -6,6 +6,7 @@ import {
 	fmtCtx,
 	type ModelSearchLookup,
 	normalizeModelText,
+	resolveContextWindow,
 	sortModels,
 	stepEffectiveThinkingLevel,
 	stepThinkingLevel,
@@ -57,6 +58,25 @@ describe("stepEffectiveThinkingLevel", () => {
 	});
 });
 
+describe("resolveContextWindow", () => {
+	it("prefers provider contextWindow over fallback", () => {
+		expect(resolveContextWindow({ contextWindow: 200_000 }, 128_000)).toBe(200_000);
+	});
+	it("falls back to dev limit when provider is 0/missing", () => {
+		expect(resolveContextWindow({ contextWindow: 0 }, 128_000)).toBe(128_000);
+		expect(resolveContextWindow({}, 64_000)).toBe(64_000);
+	});
+	it("ignores non-finite and negative values", () => {
+		expect(resolveContextWindow({ contextWindow: Number.NaN }, 128_000)).toBe(128_000);
+		expect(resolveContextWindow({ contextWindow: Number.POSITIVE_INFINITY }, 50_000)).toBe(50_000);
+		expect(resolveContextWindow({ contextWindow: -1 }, 32_000)).toBe(32_000);
+	});
+	it("returns 0 when both missing/invalid", () => {
+		expect(resolveContextWindow({}, undefined)).toBe(0);
+		expect(resolveContextWindow({ contextWindow: null }, null)).toBe(0);
+	});
+});
+
 describe("fmtCtx", () => {
 	it("formats 0 as 0", () => expect(fmtCtx(0)).toBe("0"));
 	it("formats small numbers as-is", () => expect(fmtCtx(512)).toBe("512"));

package/src/models.ts

@@ -31,6 +31,20 @@ import { patchOutBuiltinModelCommand } from "./patch-builtin";
 
 // ─── Pure logic (exported for tests) ─────────────────────────────────────────
 
+export function resolveContextWindow(
+	model: { contextWindow?: number | null },
+	fallback?: number | null,
+): number {
+	if (
+		typeof model.contextWindow === "number" &&
+		Number.isFinite(model.contextWindow) &&
+		model.contextWindow > 0
+	)
+		return model.contextWindow;
+	if (typeof fallback === "number" && Number.isFinite(fallback) && fallback > 0) return fallback;
+	return 0;
+}
+
 export function fmtCtx(n: number): string {
 	if (!n || n < 1_000) return `${n}`;
 	if (n >= 1_000_000) {
@@ -286,6 +300,14 @@ async function showEnhancedPicker(pi: ExtensionAPI, ctx: ExtensionContext): Prom
 				1,
 			);
 
+			// Widest cost string so the cost column pads to a common width and the
+			// following ⚡score/stars stay column-aligned (e.g. "10.00/50.00" is 11
+			// chars — a fixed pad of 10 shifted those rows right by one).
+			const maxCostWidth = Math.max(
+				...dedupedRows.map((r) => fmtCost(r.dev).length),
+				"free".length,
+			);
+
 			// Mute low-info parts (separators, padding, #, ☆) so the actual values pop.
 			const mute = (s: string) => theme.fg("muted", s);
 			const sep = mute(" · ");
@@ -338,16 +360,19 @@ async function showEnhancedPicker(pi: ExtensionAPI, ctx: ExtensionContext): Prom
 
 				// Description: ctx · cost · score stars
 				// Colors: ctx muted · cost success (free muted) · score+stars warning
-				const ctxRaw = fmtCtx(dev?.limit?.context ?? 0);
+				// Context: provider's `contextWindow` (source of truth) → fallback to modelgrep `dev.limit.context`.
+				const ctxRaw = fmtCtx(
+					resolveContextWindow(m as { contextWindow?: number }, dev?.limit?.context),
+				);
 				const ctxStr = mute(ctxRaw.padStart(4));
 				const rawCost = fmtCost(dev);
 				let costSeg: string;
 				if (rawCost === "—") {
-					costSeg = theme.fg("dim", "—".padEnd(10));
+					costSeg = theme.fg("dim", "—".padEnd(maxCostWidth));
 				} else if (rawCost === "free") {
-					costSeg = mute("free".padEnd(10));
+					costSeg = mute("free".padEnd(maxCostWidth));
 				} else {
-					costSeg = theme.fg("success", rawCost.padEnd(10));
+					costSeg = theme.fg("success", rawCost.padEnd(maxCostWidth));
 				}
 				let benchSeg = "";
 				if (bench) {