mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-20 13:02:15 +00:00
fix(coding-agent): improve model scope toggle
This commit is contained in:
parent
f711340136
commit
2217cde74c
1 changed files with 90 additions and 39 deletions
|
|
@ -13,6 +13,7 @@ import type { ModelRegistry } from "../../../core/model-registry.js";
|
||||||
import type { SettingsManager } from "../../../core/settings-manager.js";
|
import type { SettingsManager } from "../../../core/settings-manager.js";
|
||||||
import { theme } from "../theme/theme.js";
|
import { theme } from "../theme/theme.js";
|
||||||
import { DynamicBorder } from "./dynamic-border.js";
|
import { DynamicBorder } from "./dynamic-border.js";
|
||||||
|
import { keyHint } from "./keybinding-hints.js";
|
||||||
|
|
||||||
interface ModelItem {
|
interface ModelItem {
|
||||||
provider: string;
|
provider: string;
|
||||||
|
|
@ -25,6 +26,8 @@ interface ScopedModelItem {
|
||||||
thinkingLevel: string;
|
thinkingLevel: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type ModelScope = "all" | "scoped";
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Component that renders a model selector with search
|
* Component that renders a model selector with search
|
||||||
*/
|
*/
|
||||||
|
|
@ -42,6 +45,8 @@ export class ModelSelectorComponent extends Container implements Focusable {
|
||||||
}
|
}
|
||||||
private listContainer: Container;
|
private listContainer: Container;
|
||||||
private allModels: ModelItem[] = [];
|
private allModels: ModelItem[] = [];
|
||||||
|
private scopedModelItems: ModelItem[] = [];
|
||||||
|
private activeModels: ModelItem[] = [];
|
||||||
private filteredModels: ModelItem[] = [];
|
private filteredModels: ModelItem[] = [];
|
||||||
private selectedIndex: number = 0;
|
private selectedIndex: number = 0;
|
||||||
private currentModel?: Model<any>;
|
private currentModel?: Model<any>;
|
||||||
|
|
@ -52,6 +57,9 @@ export class ModelSelectorComponent extends Container implements Focusable {
|
||||||
private errorMessage?: string;
|
private errorMessage?: string;
|
||||||
private tui: TUI;
|
private tui: TUI;
|
||||||
private scopedModels: ReadonlyArray<ScopedModelItem>;
|
private scopedModels: ReadonlyArray<ScopedModelItem>;
|
||||||
|
private scope: ModelScope = "all";
|
||||||
|
private scopeText?: Text;
|
||||||
|
private scopeHintText?: Text;
|
||||||
|
|
||||||
constructor(
|
constructor(
|
||||||
tui: TUI,
|
tui: TUI,
|
||||||
|
|
@ -70,6 +78,7 @@ export class ModelSelectorComponent extends Container implements Focusable {
|
||||||
this.settingsManager = settingsManager;
|
this.settingsManager = settingsManager;
|
||||||
this.modelRegistry = modelRegistry;
|
this.modelRegistry = modelRegistry;
|
||||||
this.scopedModels = scopedModels;
|
this.scopedModels = scopedModels;
|
||||||
|
this.scope = scopedModels.length > 0 ? "scoped" : "all";
|
||||||
this.onSelectCallback = onSelect;
|
this.onSelectCallback = onSelect;
|
||||||
this.onCancelCallback = onCancel;
|
this.onCancelCallback = onCancel;
|
||||||
|
|
||||||
|
|
@ -78,11 +87,15 @@ export class ModelSelectorComponent extends Container implements Focusable {
|
||||||
this.addChild(new Spacer(1));
|
this.addChild(new Spacer(1));
|
||||||
|
|
||||||
// Add hint about model filtering
|
// Add hint about model filtering
|
||||||
const hintText =
|
if (scopedModels.length > 0) {
|
||||||
scopedModels.length > 0
|
this.scopeText = new Text(this.getScopeText(), 0, 0);
|
||||||
? "Showing models from --models scope"
|
this.addChild(this.scopeText);
|
||||||
: "Only showing models with configured API keys (see README for details)";
|
this.scopeHintText = new Text(this.getScopeHintText(), 0, 0);
|
||||||
|
this.addChild(this.scopeHintText);
|
||||||
|
} else {
|
||||||
|
const hintText = "Only showing models with configured API keys (see README for details)";
|
||||||
this.addChild(new Text(theme.fg("warning", hintText), 0, 0));
|
this.addChild(new Text(theme.fg("warning", hintText), 0, 0));
|
||||||
|
}
|
||||||
this.addChild(new Spacer(1));
|
this.addChild(new Spacer(1));
|
||||||
|
|
||||||
// Create search input
|
// Create search input
|
||||||
|
|
@ -124,14 +137,6 @@ export class ModelSelectorComponent extends Container implements Focusable {
|
||||||
private async loadModels(): Promise<void> {
|
private async loadModels(): Promise<void> {
|
||||||
let models: ModelItem[];
|
let models: ModelItem[];
|
||||||
|
|
||||||
// Use scoped models if provided via --models flag
|
|
||||||
if (this.scopedModels.length > 0) {
|
|
||||||
models = this.scopedModels.map((scoped) => ({
|
|
||||||
provider: scoped.model.provider,
|
|
||||||
id: scoped.model.id,
|
|
||||||
model: scoped.model,
|
|
||||||
}));
|
|
||||||
} else {
|
|
||||||
// Refresh to pick up any changes to models.json
|
// Refresh to pick up any changes to models.json
|
||||||
this.modelRegistry.refresh();
|
this.modelRegistry.refresh();
|
||||||
|
|
||||||
|
|
@ -151,28 +156,64 @@ export class ModelSelectorComponent extends Container implements Focusable {
|
||||||
}));
|
}));
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
this.allModels = [];
|
this.allModels = [];
|
||||||
|
this.scopedModelItems = [];
|
||||||
|
this.activeModels = [];
|
||||||
this.filteredModels = [];
|
this.filteredModels = [];
|
||||||
this.errorMessage = error instanceof Error ? error.message : String(error);
|
this.errorMessage = error instanceof Error ? error.message : String(error);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
this.allModels = this.sortModels(models);
|
||||||
|
this.scopedModelItems = this.sortModels(
|
||||||
|
this.scopedModels.map((scoped) => ({
|
||||||
|
provider: scoped.model.provider,
|
||||||
|
id: scoped.model.id,
|
||||||
|
model: scoped.model,
|
||||||
|
})),
|
||||||
|
);
|
||||||
|
this.activeModels = this.scope === "scoped" ? this.scopedModelItems : this.allModels;
|
||||||
|
this.filteredModels = this.activeModels;
|
||||||
|
this.selectedIndex = Math.min(this.selectedIndex, Math.max(0, this.filteredModels.length - 1));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private sortModels(models: ModelItem[]): ModelItem[] {
|
||||||
|
const sorted = [...models];
|
||||||
// Sort: current model first, then by provider
|
// Sort: current model first, then by provider
|
||||||
models.sort((a, b) => {
|
sorted.sort((a, b) => {
|
||||||
const aIsCurrent = modelsAreEqual(this.currentModel, a.model);
|
const aIsCurrent = modelsAreEqual(this.currentModel, a.model);
|
||||||
const bIsCurrent = modelsAreEqual(this.currentModel, b.model);
|
const bIsCurrent = modelsAreEqual(this.currentModel, b.model);
|
||||||
if (aIsCurrent && !bIsCurrent) return -1;
|
if (aIsCurrent && !bIsCurrent) return -1;
|
||||||
if (!aIsCurrent && bIsCurrent) return 1;
|
if (!aIsCurrent && bIsCurrent) return 1;
|
||||||
return a.provider.localeCompare(b.provider);
|
return a.provider.localeCompare(b.provider);
|
||||||
});
|
});
|
||||||
|
return sorted;
|
||||||
|
}
|
||||||
|
|
||||||
this.allModels = models;
|
private getScopeText(): string {
|
||||||
this.filteredModels = models;
|
const allText = this.scope === "all" ? theme.fg("accent", "all") : theme.fg("muted", "all");
|
||||||
this.selectedIndex = Math.min(this.selectedIndex, Math.max(0, models.length - 1));
|
const scopedText = this.scope === "scoped" ? theme.fg("accent", "scoped") : theme.fg("muted", "scoped");
|
||||||
|
return `${theme.fg("muted", "Scope: ")}${allText}${theme.fg("muted", " | ")}${scopedText}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
private getScopeHintText(): string {
|
||||||
|
return keyHint("tab", "scope") + theme.fg("muted", " (all/scoped)");
|
||||||
|
}
|
||||||
|
|
||||||
|
private setScope(scope: ModelScope): void {
|
||||||
|
if (this.scope === scope) return;
|
||||||
|
this.scope = scope;
|
||||||
|
this.activeModels = this.scope === "scoped" ? this.scopedModelItems : this.allModels;
|
||||||
|
this.selectedIndex = 0;
|
||||||
|
this.filterModels(this.searchInput.getValue());
|
||||||
|
if (this.scopeText) {
|
||||||
|
this.scopeText.setText(this.getScopeText());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private filterModels(query: string): void {
|
private filterModels(query: string): void {
|
||||||
this.filteredModels = fuzzyFilter(this.allModels, query, ({ id, provider }) => `${id} ${provider}`);
|
this.filteredModels = query
|
||||||
|
? fuzzyFilter(this.activeModels, query, ({ id, provider }) => `${id} ${provider}`)
|
||||||
|
: this.activeModels;
|
||||||
this.selectedIndex = Math.min(this.selectedIndex, Math.max(0, this.filteredModels.length - 1));
|
this.selectedIndex = Math.min(this.selectedIndex, Math.max(0, this.filteredModels.length - 1));
|
||||||
this.updateList();
|
this.updateList();
|
||||||
}
|
}
|
||||||
|
|
@ -232,6 +273,16 @@ export class ModelSelectorComponent extends Container implements Focusable {
|
||||||
|
|
||||||
handleInput(keyData: string): void {
|
handleInput(keyData: string): void {
|
||||||
const kb = getEditorKeybindings();
|
const kb = getEditorKeybindings();
|
||||||
|
if (kb.matches(keyData, "tab")) {
|
||||||
|
if (this.scopedModelItems.length > 0) {
|
||||||
|
const nextScope: ModelScope = this.scope === "all" ? "scoped" : "all";
|
||||||
|
this.setScope(nextScope);
|
||||||
|
if (this.scopeHintText) {
|
||||||
|
this.scopeHintText.setText(this.getScopeHintText());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
// Up arrow - wrap to bottom when at top
|
// Up arrow - wrap to bottom when at top
|
||||||
if (kb.matches(keyData, "selectUp")) {
|
if (kb.matches(keyData, "selectUp")) {
|
||||||
if (this.filteredModels.length === 0) return;
|
if (this.filteredModels.length === 0) return;
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue