51 lines
1.5 KiB
TypeScript
51 lines
1.5 KiB
TypeScript
import memoize from 'lodash-es/memoize.js'
|
|
import { getAPIProvider } from './providers.js'
|
|
|
|
export type ModelCapabilityOverride =
|
|
| 'effort'
|
|
| 'max_effort'
|
|
| 'thinking'
|
|
| 'adaptive_thinking'
|
|
| 'interleaved_thinking'
|
|
|
|
const TIERS = [
|
|
{
|
|
modelEnvVar: 'ANTHROPIC_DEFAULT_OPUS_MODEL',
|
|
capabilitiesEnvVar: 'ANTHROPIC_DEFAULT_OPUS_MODEL_SUPPORTED_CAPABILITIES',
|
|
},
|
|
{
|
|
modelEnvVar: 'ANTHROPIC_DEFAULT_SONNET_MODEL',
|
|
capabilitiesEnvVar: 'ANTHROPIC_DEFAULT_SONNET_MODEL_SUPPORTED_CAPABILITIES',
|
|
},
|
|
{
|
|
modelEnvVar: 'ANTHROPIC_DEFAULT_HAIKU_MODEL',
|
|
capabilitiesEnvVar: 'ANTHROPIC_DEFAULT_HAIKU_MODEL_SUPPORTED_CAPABILITIES',
|
|
},
|
|
] as const
|
|
|
|
/**
|
|
* Check whether a 3p model capability override is set for a model that matches one of
|
|
* the pinned ANTHROPIC_DEFAULT_*_MODEL env vars.
|
|
*/
|
|
export const get3PModelCapabilityOverride = memoize(
|
|
(model: string, capability: ModelCapabilityOverride): boolean | undefined => {
|
|
if (getAPIProvider() === 'firstParty') {
|
|
return undefined
|
|
}
|
|
const m = model.toLowerCase()
|
|
for (const tier of TIERS) {
|
|
const pinned = process.env[tier.modelEnvVar]
|
|
const capabilities = process.env[tier.capabilitiesEnvVar]
|
|
if (!pinned || capabilities === undefined) continue
|
|
if (m !== pinned.toLowerCase()) continue
|
|
return capabilities
|
|
.toLowerCase()
|
|
.split(',')
|
|
.map(s => s.trim())
|
|
.includes(capability)
|
|
}
|
|
return undefined
|
|
},
|
|
(model, capability) => `${model.toLowerCase()}:${capability}`,
|
|
)
|