Something went wrong. Try again.
source dump of claude code forked from oppi.li/claude-code
Something went wrong. Try again.
1.5 kB · 50 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051import 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}`,)