getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-19 01:04:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Filter model selector to only show models with configured API keys

Browse source
This commit is contained in:
Carlos Gutierrez 2025-11-15 18:49:30 -07:00
parent 3932c5e04c
commit 58eefa8d3d
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
packages/coding-agent/src/tui/model-selector.ts
View file

@ -1,4 +1,4 @@
import { getModels, getProviders, type Model } from "@mariozechner/pi-ai"; import { getApiKey, getModels, getProviders, type Model } from "@mariozechner/pi-ai";
import { Container, Input, Spacer, Text } from "@mariozechner/pi-tui"; import { Container, Input, Spacer, Text } from "@mariozechner/pi-tui";
import chalk from "chalk"; import chalk from "chalk";
@ -71,8 +71,11 @@ export class ModelSelectorComponent extends Container {
} }
} }
// Filter out models from providers without API keys
const filteredModels = models.filter((item) => getApiKey(item.provider as any) !== undefined);
// Sort: current model first, then by provider // Sort: current model first, then by provider
models.sort((a, b) => { filteredModels.sort((a, b) => {
const aIsCurrent = this.currentModel?.id === a.model.id; const aIsCurrent = this.currentModel?.id === a.model.id;
const bIsCurrent = this.currentModel?.id === b.model.id; const bIsCurrent = this.currentModel?.id === b.model.id;
if (aIsCurrent && !bIsCurrent) return -1; if (aIsCurrent && !bIsCurrent) return -1;
@ -80,8 +83,8 @@ export class ModelSelectorComponent extends Container {
return a.provider.localeCompare(b.provider); return a.provider.localeCompare(b.provider);
}); });
this.allModels = models; this.allModels = filteredModels;
this.filteredModels = models; this.filteredModels = filteredModels;
} }
private filterModels(query: string): void { private filterModels(query: string): void {