David Brailovsky
31 lines
1.2 KiB
TypeScript
31 lines
1.2 KiB
TypeScript
import type { ThinkingLevel, ThinkingLevelMap } from "../src/types.ts";
|
|
|
|
export type ModelsDevReasoningOption =
|
|
| { type: "toggle" }
|
|
| {
|
|
type: "effort";
|
|
values: Array<"none" | "minimal" | "low" | "medium" | "high" | "xhigh" | "max" | "default" | null>;
|
|
}
|
|
| { type: "budget_tokens"; min?: number; max?: number };
|
|
|
|
const THINKING_LEVELS: readonly ThinkingLevel[] = ["minimal", "low", "medium", "high", "xhigh", "max"];
|
|
|
|
/**
|
|
* Converts models.dev verified effort values into Pi's selectable thinking levels.
|
|
* Values without a Pi equivalent (`default` and JSON `null`) are intentionally
|
|
* omitted.
|
|
*/
|
|
export function getEffortThinkingLevelMap(options: readonly ModelsDevReasoningOption[]): ThinkingLevelMap | undefined {
|
|
const effortValues = options.flatMap((option) => (option.type === "effort" ? option.values : []));
|
|
if (effortValues.length === 0) return undefined;
|
|
|
|
const supported = new Set(effortValues);
|
|
if (!THINKING_LEVELS.some((level) => supported.has(level)) && !supported.has("none")) return undefined;
|
|
|
|
const map: ThinkingLevelMap = { off: supported.has("none") ? "none" : null };
|
|
for (const level of THINKING_LEVELS) {
|
|
map[level] = supported.has(level) ? level : null;
|
|
}
|
|
return map;
|
|
}
|