mirror of
https://github.com/EveryInc/compound-engineering-plugin.git
synced 2026-09-19 01:09:55 +08:00
Trevin Chow
95 lines
3.5 KiB
TypeScript
95 lines
3.5 KiB
TypeScript
/**
|
|
* Shared model normalization utilities for cross-platform conversion.
|
|
*
|
|
* Claude Code uses bare family aliases (`model: sonnet`) that must be
|
|
* resolved differently depending on the target platform.
|
|
*/
|
|
|
|
/**
|
|
* Bare Claude family aliases used in Claude Code (e.g. `model: haiku`).
|
|
* Maps alias -> canonical model name (without provider prefix).
|
|
* Update these when new model generations are released.
|
|
*/
|
|
export const CLAUDE_FAMILY_ALIASES: Record<string, string> = {
|
|
haiku: "claude-haiku-4-5",
|
|
sonnet: "claude-sonnet-5",
|
|
opus: "claude-opus-4-8",
|
|
}
|
|
|
|
/**
|
|
* Canonical Claude model IDs that reject non-default sampling params
|
|
* (`temperature`/`top_p`/`top_k`) with HTTP 400. Emitting an inferred
|
|
* temperature alongside one of these produces a config that fails at runtime.
|
|
* Keep in sync with CLAUDE_FAMILY_ALIASES when new generations are released.
|
|
* See the Sonnet 5 and Opus 4.7/4.8 migration notes.
|
|
*/
|
|
const SAMPLING_PARAM_REJECTING_MODELS: ReadonlySet<string> = new Set([
|
|
"claude-sonnet-5",
|
|
"claude-opus-4-7",
|
|
"claude-opus-4-8",
|
|
])
|
|
|
|
/**
|
|
* Resolve a bare Claude family alias to its canonical model name.
|
|
* Returns the input unchanged if not a recognized alias.
|
|
*
|
|
* "sonnet" -> "claude-sonnet-5"
|
|
* "claude-sonnet-4-20250514" -> "claude-sonnet-4-20250514" (unchanged)
|
|
*/
|
|
export function resolveClaudeFamilyAlias(model: string): string {
|
|
return CLAUDE_FAMILY_ALIASES[model] ?? model
|
|
}
|
|
|
|
/**
|
|
* Add a provider prefix based on model naming conventions.
|
|
* Returns the input unchanged if already prefixed (contains "/").
|
|
*
|
|
* "claude-sonnet-5" -> "anthropic/claude-sonnet-5"
|
|
* "gpt-5.6-sol" -> "openai/gpt-5.6-sol"
|
|
* "gemini-2.0" -> "google/gemini-2.0"
|
|
* "minimax-m3" -> "minimax/minimax-m3"
|
|
* "anthropic/foo" -> "anthropic/foo" (unchanged)
|
|
*/
|
|
export function addProviderPrefix(model: string): string {
|
|
if (model.includes("/")) return model
|
|
if (/^claude-/.test(model)) return `anthropic/${model}`
|
|
if (/^(gpt-|o1-|o3-)/.test(model)) return `openai/${model}`
|
|
if (/^gemini-/.test(model)) return `google/${model}`
|
|
if (/^qwen-/.test(model)) return `qwen/${model}`
|
|
if (/^minimax-/i.test(model)) return `minimax/${model}`
|
|
return `anthropic/${model}`
|
|
}
|
|
|
|
/**
|
|
* Normalize a model for targets that use provider-prefixed IDs.
|
|
* Resolves bare aliases and adds provider prefix.
|
|
*
|
|
* "sonnet" -> "anthropic/claude-sonnet-5"
|
|
* "claude-sonnet-4-20250514" -> "anthropic/claude-sonnet-4-20250514"
|
|
* "anthropic/claude-opus" -> "anthropic/claude-opus" (unchanged)
|
|
*/
|
|
export function normalizeModelWithProvider(model: string): string {
|
|
if (model.includes("/")) return model
|
|
const resolved = resolveClaudeFamilyAlias(model)
|
|
if (resolved !== model) {
|
|
console.warn(
|
|
`Warning: bare model alias "${model}" mapped to "anthropic/${resolved}". ` +
|
|
`Update CLAUDE_FAMILY_ALIASES if a newer version is available.`,
|
|
)
|
|
}
|
|
return addProviderPrefix(resolved)
|
|
}
|
|
|
|
/**
|
|
* Whether a model rejects non-default sampling params. Accepts a bare alias,
|
|
* a canonical ID, or a provider-prefixed ID; resolves aliases and strips the
|
|
* provider prefix so `sonnet` and `anthropic/claude-sonnet-5` both match.
|
|
*
|
|
* "sonnet" -> true (resolves to claude-sonnet-5)
|
|
* "claude-sonnet-4-20250514" -> false (dated Sonnet 4 accepts sampling params)
|
|
*/
|
|
export function rejectsSamplingParams(model: string): boolean {
|
|
const canonical = resolveClaudeFamilyAlias(model).replace(/^anthropic\//, "")
|
|
return SAMPLING_PARAM_REJECTING_MODELS.has(canonical)
|
|
}
|