Kit Langton
46 lines
1.7 KiB
TypeScript
46 lines
1.7 KiB
TypeScript
import type { Provider } from "@opencode-ai/sdk/v2"
|
|
|
|
export function parse(value: string) {
|
|
const [providerID, ...modelID] = value.split("/")
|
|
return { providerID, modelID: modelID.join("/") }
|
|
}
|
|
|
|
export function index(list: Provider[] | undefined) {
|
|
return new Map((list ?? []).map((item) => [item.id, item] as const))
|
|
}
|
|
|
|
export function get(list: Provider[] | ReadonlyMap<string, Provider> | undefined, providerID: string, modelID: string) {
|
|
const provider =
|
|
list instanceof Map
|
|
? list.get(providerID)
|
|
: Array.isArray(list)
|
|
? list.find((item) => item.id === providerID)
|
|
: undefined
|
|
return provider?.models[modelID]
|
|
}
|
|
|
|
export function name(
|
|
list: Provider[] | ReadonlyMap<string, Provider> | undefined,
|
|
providerID: string,
|
|
modelID: string,
|
|
) {
|
|
return get(list, providerID, modelID)?.name ?? modelID
|
|
}
|
|
|
|
export function formatRef(model: { providerID: string; id: string; variant?: string }) {
|
|
return [model.providerID, model.id, model.variant].filter((value) => value !== undefined).join("/")
|
|
}
|
|
|
|
export function switchLabel(
|
|
model: { providerID: string; id: string; variant?: string },
|
|
models?: readonly { providerID: string; id: string; name: string }[],
|
|
previous?: { providerID: string; id: string; variant?: string },
|
|
) {
|
|
if (previous?.providerID === model.providerID && previous.id === model.id)
|
|
return `Switched variant to ${model.variant ?? "default"}`
|
|
const display = models?.find((item) => item.providerID === model.providerID && item.id === model.id)?.name
|
|
if (display === undefined) return `Switched model to ${formatRef(model)}`
|
|
const variant = model.variant && model.variant !== "default" ? ` (${model.variant})` : ""
|
|
return `Switched model to ${display}${variant}`
|
|
}
|