56 lines
1.5 KiB
TypeScript
56 lines
1.5 KiB
TypeScript
|
import type { ModelRequirement } from "../../../shared/model-requirements"
|
||
|
|
import type { OmoConfig } from "./model-resolution-types"
|
||
|
|
|
||
|
|
export function formatModelWithVariant(model: string, variant?: string): string {
|
||
|
|
return variant ? `${model} (${variant})` : model
|
||
|
|
}
|
||
|
|
|
||
|
|
function getAgentOverride(
|
||
|
|
agentName: string,
|
||
|
|
config: OmoConfig
|
||
|
|
): { variant?: string; category?: string } | undefined {
|
||
|
|
const agentOverrides = config.agents
|
||
|
|
if (!agentOverrides) return undefined
|
||
|
|
|
||
|
|
return (
|
||
|
|
agentOverrides[agentName] ??
|
||
|
|
Object.entries(agentOverrides).find(([key]) => key.toLowerCase() === agentName.toLowerCase())?.[1]
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
export function getEffectiveVariant(
|
||
|
|
agentName: string,
|
||
|
|
requirement: ModelRequirement,
|
||
|
|
config: OmoConfig
|
||
|
|
): string | undefined {
|
||
|
|
const agentOverride = getAgentOverride(agentName, config)
|
||
|
|
|
||
|
|
if (agentOverride?.variant) {
|
||
|
|
return agentOverride.variant
|
||
|
|
}
|
||
|
|
|
||
|
|
const categoryName = agentOverride?.category
|
||
|
|
if (categoryName) {
|
||
|
|
const categoryVariant = config.categories?.[categoryName]?.variant
|
||
|
|
if (categoryVariant) {
|
||
|
|
return categoryVariant
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const firstEntry = requirement.fallbackChain[0]
|
||
|
|
return firstEntry?.variant ?? requirement.variant
|
||
|
|
}
|
||
|
|
|
||
|
|
export function getCategoryEffectiveVariant(
|
||
|
|
categoryName: string,
|
||
|
|
requirement: ModelRequirement,
|
||
|
|
config: OmoConfig
|
||
|
|
): string | undefined {
|
||
|
|
const categoryVariant = config.categories?.[categoryName]?.variant
|
||
|
|
if (categoryVariant) {
|
||
|
|
return categoryVariant
|
||
|
|
}
|
||
|
|
const firstEntry = requirement.fallbackChain[0]
|
||
|
|
return firstEntry?.variant ?? requirement.variant
|
||
|
|
}
|