can1357
65 lines
1.7 KiB
TypeScript
65 lines
1.7 KiB
TypeScript
import type { ThinkingLevel } from "@oh-my-pi/pi-agent-core";
|
|
import { Container, type SelectItem, SelectList } from "@oh-my-pi/pi-tui";
|
|
import { getSelectListTheme } from "../../modes/theme/theme";
|
|
import { DynamicBorder } from "./dynamic-border";
|
|
|
|
const LEVEL_DESCRIPTIONS: Record<ThinkingLevel, string> = {
|
|
off: "No reasoning",
|
|
minimal: "Very brief reasoning (~1k tokens)",
|
|
low: "Light reasoning (~2k tokens)",
|
|
medium: "Moderate reasoning (~8k tokens)",
|
|
high: "Deep reasoning (~16k tokens)",
|
|
xhigh: "Maximum reasoning (~32k tokens)",
|
|
};
|
|
|
|
/**
|
|
* Component that renders a thinking level selector with borders
|
|
*/
|
|
export class ThinkingSelectorComponent extends Container {
|
|
#selectList: SelectList;
|
|
|
|
constructor(
|
|
currentLevel: ThinkingLevel,
|
|
availableLevels: ThinkingLevel[],
|
|
onSelect: (level: ThinkingLevel) => void,
|
|
onCancel: () => void,
|
|
) {
|
|
super();
|
|
|
|
const thinkingLevels: SelectItem[] = availableLevels.map(level => ({
|
|
value: level,
|
|
label: level,
|
|
description: LEVEL_DESCRIPTIONS[level],
|
|
}));
|
|
|
|
// Add top border
|
|
this.addChild(new DynamicBorder());
|
|
|
|
// Create selector
|
|
this.#selectList = new SelectList(thinkingLevels, thinkingLevels.length, getSelectListTheme());
|
|
|
|
// Preselect current level
|
|
const currentIndex = thinkingLevels.findIndex(item => item.value === currentLevel);
|
|
if (currentIndex !== -1) {
|
|
this.#selectList.setSelectedIndex(currentIndex);
|
|
}
|
|
|
|
this.#selectList.onSelect = item => {
|
|
onSelect(item.value as ThinkingLevel);
|
|
};
|
|
|
|
this.#selectList.onCancel = () => {
|
|
onCancel();
|
|
};
|
|
|
|
this.addChild(this.#selectList);
|
|
|
|
// Add bottom border
|
|
this.addChild(new DynamicBorder());
|
|
}
|
|
|
|
getSelectList(): SelectList {
|
|
return this.#selectList;
|
|
}
|
|
}
|