@xynogen/pix-models 0.1.27 → 0.1.31

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@xynogen/pix-models",
-	"version": "0.1.27",
+	"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) {
@@ -346,7 +360,10 @@ 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;