@victor-software-house/pi-openai-proxy 4.2.1 → 4.2.3

package/extensions/proxy.ts

@@ -33,6 +33,9 @@ import {
 import {
 	type Component,
 	Container,
+	fuzzyFilter,
+	getKeybindings,
+	Input,
 	type SettingItem,
 	SettingsList,
 	Text,
@@ -550,71 +553,179 @@ export default function proxyExtension(pi: ExtensionAPI): void {
 		_currentValue: string,
 		done: (selectedValue?: string) => void,
 	): Component {
-		const models = getAvailableModels();
+		interface ModelEntry {
+			canonical: string;
+			provider: string;
+		}
+
+		const models: ModelEntry[] = getAvailableModels().map((m) => ({
+			canonical: `${m.provider}/${m.id}`,
+			provider: m.provider,
+		}));
+
 		const selected = new Set(config.customModels);
 		let selectedIndex = 0;
+		const searchInput = new Input();
+		let filtered: ModelEntry[] = models;
+		const maxVisible = 20;
+
+		// Precompute provider boundary indices in the full list
+		function providerBoundaries(list: ModelEntry[]): {
+			firstOf: Map<string, number>;
+			lastOf: Map<string, number>;
+		} {
+			const firstOf = new Map<string, number>();
+			const lastOf = new Map<string, number>();
+			for (let i = 0; i < list.length; i++) {
+				const entry = list[i];
+				if (entry === undefined) continue;
+				if (!firstOf.has(entry.provider)) firstOf.set(entry.provider, i);
+				lastOf.set(entry.provider, i);
+			}
+			return { firstOf, lastOf };
+		}
+
+		function jumpProvider(direction: "prev" | "next"): void {
+			if (filtered.length === 0) return;
+			const { firstOf, lastOf } = providerBoundaries(filtered);
+			const current = filtered[selectedIndex];
+			if (current === undefined) return;
+
+			const providers = [...firstOf.keys()]; // insertion-order = list order
+			const provIdx = providers.indexOf(current.provider);
+
+			if (direction === "prev") {
+				if (provIdx <= 0) {
+					selectedIndex = 0;
+				} else {
+					const prevProvider = providers[provIdx - 1];
+					if (prevProvider !== undefined) {
+						selectedIndex = lastOf.get(prevProvider) ?? 0;
+					}
+				}
+			} else {
+				if (provIdx >= providers.length - 1) {
+					selectedIndex = filtered.length - 1;
+				} else {
+					const nextProvider = providers[provIdx + 1];
+					if (nextProvider !== undefined) {
+						selectedIndex = firstOf.get(nextProvider) ?? filtered.length - 1;
+					}
+				}
+			}
+		}
 
-		function toggle(canonicalId: string): void {
-			if (selected.has(canonicalId)) {
-				selected.delete(canonicalId);
+		function toggle(idx: number): void {
+			const entry = filtered[idx];
+			if (entry === undefined) return;
+			if (selected.has(entry.canonical)) {
+				selected.delete(entry.canonical);
 			} else {
-				selected.add(canonicalId);
+				selected.add(entry.canonical);
 			}
 			config = { ...config, customModels: [...selected] };
 			saveConfigToFile(config);
 			config = loadConfigFromFile();
 		}
 
+		function applyFilter(query: string): void {
+			if (query === "") {
+				filtered = models;
+			} else {
+				filtered = fuzzyFilter(models, query, (m) => m.canonical);
+			}
+			selectedIndex = 0;
+		}
+
+		const theme = getSettingsListTheme();
+
 		return {
 			render(width: number): string[] {
 				const lines: string[] = [];
-				lines.push("  Select models (Space: toggle, Esc: done)");
+				lines.push(...searchInput.render(width));
 				lines.push("");
 
-				if (models.length === 0) {
-					lines.push("  No models available (no auth configured)");
+				if (filtered.length === 0) {
+					lines.push(theme.hint("  No matching models"));
+					lines.push("");
+					lines.push(theme.hint("  Type to filter | Left/Right: jump provider | Esc: done"));
 					return lines;
 				}
 
-				const maxVisible = 15;
 				const start = Math.max(
 					0,
-					Math.min(selectedIndex - Math.floor(maxVisible / 2), models.length - maxVisible),
+					Math.min(selectedIndex - Math.floor(maxVisible / 2), filtered.length - maxVisible),
 				);
-				const end = Math.min(start + maxVisible, models.length);
+				const end = Math.min(start + maxVisible, filtered.length);
 
+				let lastProvider = "";
 				for (let i = start; i < end; i++) {
-					const m = models[i];
-					if (m === undefined) continue;
-					const canonical = `${m.provider}/${m.id}`;
-					const check = selected.has(canonical) ? "[x]" : "[ ]";
-					const cursor = i === selectedIndex ? "> " : "  ";
-					const line = `${cursor}${check} ${canonical}`;
-					const truncated = line.length > width ? line.slice(0, width - 1) : line;
-					lines.push(truncated);
+					const entry = filtered[i];
+					if (entry === undefined) continue;
+
+					// Provider separator
+					if (entry.provider !== lastProvider) {
+						if (lastProvider !== "") lines.push("");
+						lastProvider = entry.provider;
+					}
+
+					const check = selected.has(entry.canonical) ? "[x]" : "[ ]";
+					const cursor = i === selectedIndex ? theme.cursor : "  ";
+					const label =
+						i === selectedIndex
+							? theme.label(`${check} ${entry.canonical}`, true)
+							: theme.label(`${check} ${entry.canonical}`, false);
+					lines.push(`${cursor}${label}`.slice(0, width));
+				}
+
+				if (start > 0 || end < filtered.length) {
+					lines.push(theme.hint(`  (${String(selectedIndex + 1)}/${String(filtered.length)})`));
 				}
 
 				lines.push("");
-				lines.push(`  ${String(selected.size)} of ${String(models.length)} selected`);
+				lines.push(
+					theme.description(`  ${String(selected.size)} of ${String(models.length)} selected`),
+				);
+				lines.push("");
+				lines.push(
+					theme.hint("  Type to filter | Space: toggle | Left/Right: jump provider | Esc: done"),
+				);
 				return lines;
 			},
-			invalidate(): void {
-				// no-op
-			},
+			invalidate(): void {},
 			handleInput(data: string): void {
-				if (data === "\x1B" || data === "q") {
+				const kb = getKeybindings();
+
+				if (kb.matches(data, "tui.select.cancel")) {
 					done(`${String(selected.size)} selected`);
 					return;
 				}
-				if (models.length === 0) return;
-				if (data === "\x1B[A") {
-					selectedIndex = selectedIndex <= 0 ? models.length - 1 : selectedIndex - 1;
-				} else if (data === "\x1B[B") {
-					selectedIndex = selectedIndex >= models.length - 1 ? 0 : selectedIndex + 1;
-				} else if (data === " " || data === "\r") {
-					const m = models[selectedIndex];
-					if (m !== undefined) {
-						toggle(`${m.provider}/${m.id}`);
+				if (filtered.length === 0) {
+					// Still allow typing to refine filter
+					const sanitized = data.replace(/ /g, "");
+					if (sanitized !== "") {
+						searchInput.handleInput(sanitized);
+						applyFilter(searchInput.getValue());
+					}
+					return;
+				}
+				if (kb.matches(data, "tui.select.up")) {
+					selectedIndex = selectedIndex <= 0 ? filtered.length - 1 : selectedIndex - 1;
+				} else if (kb.matches(data, "tui.select.down")) {
+					selectedIndex = selectedIndex >= filtered.length - 1 ? 0 : selectedIndex + 1;
+				} else if (data === "\x1B[D") {
+					// Left arrow: previous provider
+					jumpProvider("prev");
+				} else if (data === "\x1B[C") {
+					// Right arrow: next provider
+					jumpProvider("next");
+				} else if (kb.matches(data, "tui.select.confirm") || data === " ") {
+					toggle(selectedIndex);
+				} else {
+					const sanitized = data.replace(/ /g, "");
+					if (sanitized !== "") {
+						searchInput.handleInput(sanitized);
+						applyFilter(searchInput.getValue());
 					}
 				}
 			},

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@victor-software-house/pi-openai-proxy",
-	"version": "4.2.1",
+	"version": "4.2.3",
 	"description": "OpenAI-compatible HTTP proxy for pi's multi-provider model registry",
 	"license": "MIT",
 	"author": "Victor Software House",