mirror of
https://github.com/claude-code-best/claude-code.git
synced 2026-06-15 12:55:51 +00:00
31 lines
800 B
TypeScript
31 lines
800 B
TypeScript
function getModelFamily(model: string): 'haiku' | 'sonnet' | 'opus' | null {
|
|
if (/haiku/i.test(model)) return 'haiku'
|
|
if (/opus/i.test(model)) return 'opus'
|
|
if (/sonnet/i.test(model)) return 'sonnet'
|
|
return null
|
|
}
|
|
|
|
export function resolveGeminiModel(anthropicModel: string): string {
|
|
if (process.env.GEMINI_MODEL) {
|
|
return process.env.GEMINI_MODEL
|
|
}
|
|
|
|
const cleanModel = anthropicModel.replace(/\[1m\]$/i, '')
|
|
const family = getModelFamily(cleanModel)
|
|
|
|
if (!family) {
|
|
return cleanModel
|
|
}
|
|
|
|
const sharedEnvVar = `ANTHROPIC_DEFAULT_${family.toUpperCase()}_MODEL`
|
|
const resolvedModel = process.env[sharedEnvVar]
|
|
if (resolvedModel) {
|
|
return resolvedModel
|
|
}
|
|
|
|
throw new Error(
|
|
`Gemini provider requires GEMINI_MODEL or ${sharedEnvVar} to be configured.`,
|
|
)
|
|
}
|
|
|