pi-model-costs 1.0.0 → 1.1.0

package/README.md

@@ -11,13 +11,16 @@ the real cost of each model before you switch.
   - context window and max output tokens
   - cache read/write rates and pricing tiers
   - reasoning support and modality (text / text+image)
+  - **`ctrl+s` cost sorting** — cycle through default, input ↑↓, output ↑↓
+    and total (input + output) ↑↓; the active mode is shown in the footer
   - same flow as `/model`: arrows to navigate, type to fuzzy-filter, `Tab`
     toggles all/scoped (when scoped models are configured), `Enter` selects,
     `Esc` cancels
 - **Footer status** — pricing of the currently active model in the footer
   (disable it by setting `SHOW_STATUS = false` in the source).
 
-The current model is marked with a ✓ and sorted to the top.
+In default order the current model is marked with a ✓ and sorted to the top;
+cost sorts rank models purely by price (ties broken by provider/id).
 
 ![model-cost picker](images/model-cost.png)
 

package/model-costs.ts

@@ -7,7 +7,9 @@
  *                          window, cache rates, tiers and reasoning support.
  *                          Same flow as /model: arrows to navigate, type to
  *                          fuzzy-filter, Tab toggles all/scoped (when scoped
- *                          models are configured), Enter selects, Esc cancels.
+ *                          models are configured), Enter selects, Esc
+ *                          cancels. ctrl+s cycles cost sorting (input /
+ *                          output / total, ascending or descending).
  *
  *   Footer status        — pricing of the currently active model
  *                          (set SHOW_STATUS = false below to disable).
@@ -71,6 +73,22 @@ interface Picked {
 	thinkingLevel?: ThinkingLevel;
 }
 
+type SortMode = "default" | "input-asc" | "input-desc" | "output-asc" | "output-desc" | "total-asc" | "total-desc";
+
+const SORT_ORDER: SortMode[] = ["default", "input-asc", "input-desc", "output-asc", "output-desc", "total-asc", "total-desc"];
+
+function sortLabel(mode: SortMode): string {
+	switch (mode) {
+		case "input-asc": return "in↑";
+		case "input-desc": return "in↓";
+		case "output-asc": return "out↑";
+		case "output-desc": return "out↓";
+		case "total-asc": return "total↑";
+		case "total-desc": return "total↓";
+		default: return "default";
+	}
+}
+
 class ModelCostPicker extends Container implements Focusable {
 	private searchInput = new Input();
 	private listContainer = new Container();
@@ -93,6 +111,8 @@ class ModelCostPicker extends Container implements Focusable {
 	private selectedIndex = 0;
 	private scope: "all" | "scoped";
 	private scopeText: Text | undefined;
+	private sortMode: SortMode = "default";
+	private footerText!: Text;
 
 	constructor(
 		private tui: import("@earendil-works/pi-tui").TUI,
@@ -135,7 +155,8 @@ class ModelCostPicker extends Container implements Focusable {
 		this.addChild(this.detailContainer);
 
 		this.addChild(new Spacer(1));
-		this.addChild(new Text(theme.fg("dim", "  ↑↓ navigate · Enter select · Esc cancel · prices per 1M tokens"), 0, 0));
+		this.footerText = new Text(this.renderFooter(), 0, 0);
+		this.addChild(this.footerText);
 		this.addChild(new DynamicBorder((s) => theme.fg("accent", s)));
 
 		const currentIndex = this.activeItems.findIndex((item) => sameModel(this.currentModel, item.model));
@@ -143,17 +164,44 @@ class ModelCostPicker extends Container implements Focusable {
 		this.applyFilter(initialQuery ?? "");
 	}
 
-	private sortItems(items: ModelItem[]): ModelItem[] {
-		const sorted = [...items];
-		sorted.sort((a, b) => {
+	private compareItems(a: ModelItem, b: ModelItem): number {
+		// Pin current model on top only in default mode; cost sorts rank it purely
+		if (this.sortMode === "default") {
 			const aCurrent = sameModel(this.currentModel, a.model);
 			const bCurrent = sameModel(this.currentModel, b.model);
 			if (aCurrent && !bCurrent) return -1;
 			if (!aCurrent && bCurrent) return 1;
-			const byProvider = a.model.provider.localeCompare(b.model.provider);
-			return byProvider !== 0 ? byProvider : a.model.id.localeCompare(b.model.id);
-		});
-		return sorted;
+		}
+		const dir = this.sortMode.endsWith("desc") ? -1 : 1;
+		let diff = 0;
+		if (this.sortMode.startsWith("input")) diff = a.model.cost.input - b.model.cost.input;
+		else if (this.sortMode.startsWith("output")) diff = a.model.cost.output - b.model.cost.output;
+		else if (this.sortMode.startsWith("total")) diff = (a.model.cost.input + a.model.cost.output) - (b.model.cost.input + b.model.cost.output);
+		if (diff !== 0) return diff * dir;
+		const byProvider = a.model.provider.localeCompare(b.model.provider);
+		return byProvider !== 0 ? byProvider : a.model.id.localeCompare(b.model.id);
+	}
+
+	private sortItems(items: ModelItem[]): ModelItem[] {
+		return [...items].sort((a, b) => this.compareItems(a, b));
+	}
+
+	private renderFooter(): string {
+		return this.theme.fg("dim", `  ↑↓ navigate · Enter select · Esc cancel · ctrl+s sort (${sortLabel(this.sortMode)}) · prices per 1M tokens`);
+	}
+
+	private cycleSort(): void {
+		this.sortMode = SORT_ORDER[(SORT_ORDER.indexOf(this.sortMode) + 1) % SORT_ORDER.length] as SortMode;
+		// Re-sort cached lists in place, full re-query if model count grows large
+		this.allItems = this.sortItems(this.allItems);
+		this.scopedItems = this.sortItems(this.scopedItems);
+		this.activeItems = this.scope === "scoped" ? this.scopedItems : this.allItems;
+		this.footerText.setText(this.renderFooter());
+		this.applyFilter(this.searchInput.getValue());
+		// Jump selection to current model so it stays visible after re-sort
+		const curIdx = this.filtered.findIndex((item) => sameModel(this.currentModel, item.model));
+		this.selectedIndex = curIdx >= 0 ? curIdx : 0;
+		this.updateList();
 	}
 
 	private renderScopeText(): string {
@@ -276,6 +324,8 @@ class ModelCostPicker extends Container implements Focusable {
 			this.selectedIndex = currentIndex >= 0 ? currentIndex : 0;
 			if (this.scopeText) this.scopeText.setText(this.renderScopeText());
 			this.applyFilter(this.searchInput.getValue());
+		} else if (matchesKey(data, "ctrl+s")) {
+			this.cycleSort();
 		} else if (matchesKey(data, "up")) {
 			if (this.filtered.length === 0) return;
 			this.selectedIndex = this.selectedIndex === 0 ? this.filtered.length - 1 : this.selectedIndex - 1;

package/package.json

@@ -1,8 +1,8 @@
 {
 	"name": "pi-model-costs",
 	"displayName": "model-costs",
-	"version": "1.0.0",
-	"description": "Pricing-aware model picker for pi: select models with per-1M-token cost, context window, cache rates, tiers and reasoning support.",
+	"version": "1.1.0",
+	"description": "Pricing-aware model picker for pi: select models with per-1M-token cost, context window, cache rates, tiers and reasoning support. Sort models by input/output/total cost with ctrl+s.",
 	"type": "module",
 	"keywords": [
 		"pi-package",