shumengya/sproutclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #19 from CarlosGtrz/filter-model-selector-by-api-key

Browse Source 
Filter model selector to only show models with configured API keys
This commit is contained in:
Mario Zechner
2025-11-16 20:43:02 +01:00
committed by GitHub
parent 3932c5e04c 58eefa8d3d
commit 977dfec5c9
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 {