@xynogen/pix-models 0.1.5 → 0.1.7

package/README.md

@@ -1,10 +1,10 @@
 # pix-models
 
-Pi extension — enhanced `/models` picker with BenchLM ranks.
+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 model name, provider, context window, cost per million tokens, and BenchLM rank/score when available. The list is sorted by BenchLM rank (best first), then alphabetically for unranked 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`; BenchLM data is fetched and cached at startup. 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 model name, provider, context window, cost per million tokens, and a coding-focused score/rank 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.
 
 ## Install
 

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@xynogen/pix-models",
-	"version": "0.1.5",
+	"version": "0.1.7",
 	"description": "Pi extension — enhanced /models picker with BenchLM ranks",
 	"type": "module",
 	"main": "src/index.ts",

package/src/models.ts

@@ -23,7 +23,11 @@ import {
 	Text,
 	visibleWidth,
 } from "@earendil-works/pi-tui";
-import { lookupBenchmark, lookupModelsDev } from "@xynogen/pix-data";
+import {
+	benchScoreColor,
+	lookupBenchmark,
+	lookupModelsDev,
+} from "@xynogen/pix-data";
 import { patchOutBuiltinModelCommand } from "./patch-builtin";
 
 // ─── Pure logic (exported for tests) ─────────────────────────────────────────
@@ -175,7 +179,10 @@ async function showEnhancedPicker(
 					let rankPrefix: string;
 					if (localRank) {
 						const rankStr = String(localRank).padEnd(maxRankWidth);
-						rankPrefix = mute("#") + theme.fg("warning", rankStr);
+						// Color rank by the model's bench score (same scale as ⚡score),
+						// not by list position — keeps the two colors consistent.
+						const rankColor = benchScoreColor(bench?.overallScore);
+						rankPrefix = mute("#") + theme.fg(rankColor, rankStr);
 					} else {
 						rankPrefix = " ".repeat(maxRankWidth + 1);
 					}
@@ -200,6 +207,7 @@ async function showEnhancedPicker(
 					if (bench) {
 						const score = bench.overallScore ?? "?";
 						const s = bench.overallScore;
+						const scoreColor = benchScoreColor(s);
 						let filled = 1;
 						if (typeof s === "number") {
 							if (s >= 90) filled = 5;
@@ -208,9 +216,9 @@ async function showEnhancedPicker(
 							else if (s >= 50) filled = 2;
 						}
 						const starBar =
-							theme.fg("warning", "★".repeat(filled)) +
+							theme.fg(scoreColor, "★".repeat(filled)) +
 							mute("☆".repeat(5 - filled));
-						benchSeg = `⚡${theme.fg("warning", String(score))} ${starBar}`;
+						benchSeg = `⚡${theme.fg(scoreColor, String(score))} ${starBar}`;
 					}
 					const desc = [ctxStr, costSeg, benchSeg].filter(Boolean).join(sep);