70baa6f7db
Add xAI Grok as a new API provider. Reuses OpenAI-compatible message/tool
converters and stream adapter with Grok-specific client and model mapping.
Default model mapping:
opus → grok-4.20-reasoning
sonnet → grok-3-mini-fast
haiku → grok-3-mini-fast
Users can customize mapping via:
- GROK_MODEL env var (override all)
- GROK_MODEL_MAP env var (JSON family map, e.g. {"opus":"grok-4"})
- GROK_DEFAULT_{FAMILY}_MODEL env vars
Activation: CLAUDE_CODE_USE_GROK=1 or modelType: "grok" in settings.json
Also integrates with /provider command for runtime switching.
108 lines
3.4 KiB
TypeScript
108 lines
3.4 KiB
TypeScript
/**
|
|
* Default mapping from Anthropic model names to Grok model names.
|
|
*
|
|
* Users can override per-family via GROK_DEFAULT_{FAMILY}_MODEL env vars,
|
|
* or override the entire mapping via GROK_MODEL_MAP env var (JSON string):
|
|
* GROK_MODEL_MAP='{"opus":"grok-4","sonnet":"grok-3","haiku":"grok-3-mini-fast"}'
|
|
*/
|
|
const DEFAULT_MODEL_MAP: Record<string, string> = {
|
|
'claude-sonnet-4-20250514': 'grok-3-mini-fast',
|
|
'claude-sonnet-4-5-20250929': 'grok-3-mini-fast',
|
|
'claude-sonnet-4-6': 'grok-3-mini-fast',
|
|
'claude-opus-4-20250514': 'grok-4.20-reasoning',
|
|
'claude-opus-4-1-20250805': 'grok-4.20-reasoning',
|
|
'claude-opus-4-5-20251101': 'grok-4.20-reasoning',
|
|
'claude-opus-4-6': 'grok-4.20-reasoning',
|
|
'claude-haiku-4-5-20251001': 'grok-3-mini-fast',
|
|
'claude-3-5-haiku-20241022': 'grok-3-mini-fast',
|
|
'claude-3-7-sonnet-20250219': 'grok-3-mini-fast',
|
|
'claude-3-5-sonnet-20241022': 'grok-3-mini-fast',
|
|
}
|
|
|
|
/**
|
|
* Family-level mapping defaults (used by GROK_MODEL_MAP).
|
|
*/
|
|
const DEFAULT_FAMILY_MAP: Record<string, string> = {
|
|
opus: 'grok-4.20-reasoning',
|
|
sonnet: 'grok-3-mini-fast',
|
|
haiku: 'grok-3-mini-fast',
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
/**
|
|
* Parse user-provided model map from GROK_MODEL_MAP env var.
|
|
* Accepts JSON like: {"opus":"grok-4","sonnet":"grok-3","haiku":"grok-3-mini-fast"}
|
|
*/
|
|
function getUserModelMap(): Record<string, string> | null {
|
|
const raw = process.env.GROK_MODEL_MAP
|
|
if (!raw) return null
|
|
try {
|
|
const parsed = JSON.parse(raw)
|
|
if (parsed && typeof parsed === 'object' && !Array.isArray(parsed)) {
|
|
return parsed as Record<string, string>
|
|
}
|
|
} catch {
|
|
// ignore invalid JSON
|
|
}
|
|
return null
|
|
}
|
|
|
|
/**
|
|
* Resolve the Grok model name for a given Anthropic model.
|
|
*
|
|
* Priority:
|
|
* 1. GROK_MODEL env var (override all)
|
|
* 2. GROK_MODEL_MAP env var — JSON family map (e.g. {"opus":"grok-4"})
|
|
* 3. GROK_DEFAULT_{FAMILY}_MODEL env var (e.g. GROK_DEFAULT_OPUS_MODEL)
|
|
* 4. ANTHROPIC_DEFAULT_{FAMILY}_MODEL env var (backward compat)
|
|
* 5. DEFAULT_MODEL_MAP lookup
|
|
* 6. Family-level default
|
|
* 7. Pass through original model name
|
|
*/
|
|
export function resolveGrokModel(anthropicModel: string): string {
|
|
// 1. Global override
|
|
if (process.env.GROK_MODEL) {
|
|
return process.env.GROK_MODEL
|
|
}
|
|
|
|
const cleanModel = anthropicModel.replace(/\[1m\]$/, '')
|
|
const family = getModelFamily(cleanModel)
|
|
|
|
// 2. User-provided model map
|
|
const userMap = getUserModelMap()
|
|
if (userMap && family && userMap[family]) {
|
|
return userMap[family]
|
|
}
|
|
|
|
if (family) {
|
|
// 3. Grok-specific family override
|
|
const grokEnvVar = `GROK_DEFAULT_${family.toUpperCase()}_MODEL`
|
|
const grokOverride = process.env[grokEnvVar]
|
|
if (grokOverride) return grokOverride
|
|
|
|
// 4. Anthropic env var (backward compat)
|
|
const anthropicEnvVar = `ANTHROPIC_DEFAULT_${family.toUpperCase()}_MODEL`
|
|
const anthropicOverride = process.env[anthropicEnvVar]
|
|
if (anthropicOverride) return anthropicOverride
|
|
}
|
|
|
|
// 5. Exact model name lookup
|
|
if (DEFAULT_MODEL_MAP[cleanModel]) {
|
|
return DEFAULT_MODEL_MAP[cleanModel]
|
|
}
|
|
|
|
// 6. Family-level default
|
|
if (family && DEFAULT_FAMILY_MAP[family]) {
|
|
return DEFAULT_FAMILY_MAP[family]
|
|
}
|
|
|
|
// 7. Pass through
|
|
return cleanModel
|
|
}
|