Matan Kushner
34e334eabe
Adds vercel to every fallback entry where the model exists on the gateway (kimi-k2.5, glm-5, glm-4.6v, minimax-m2.7, grok-code-fast-1, gpt-5-nano). Adds inferSubProvider mappings for minimax, moonshotai, and zai. Updates librarian/explore special cases to prefer minimax via gateway over claude-haiku.
58 lines
2.1 KiB
TypeScript
58 lines
2.1 KiB
TypeScript
function inferSubProvider(model: string): string | undefined {
|
|
if (model.startsWith("claude-")) return "anthropic"
|
|
if (model.startsWith("gpt-")) return "openai"
|
|
if (model.startsWith("gemini-")) return "google"
|
|
if (model.startsWith("grok-")) return "xai"
|
|
if (model.startsWith("minimax-")) return "minimax"
|
|
if (model.startsWith("kimi-")) return "moonshotai"
|
|
if (model.startsWith("glm-")) return "zai"
|
|
return undefined
|
|
}
|
|
|
|
function transformModelForGateway(model: string): string {
|
|
return model
|
|
.replace("claude-opus-4-6", "claude-opus-4.6")
|
|
.replace("claude-sonnet-4-6", "claude-sonnet-4.6")
|
|
.replace("claude-sonnet-4-5", "claude-sonnet-4.5")
|
|
.replace("claude-haiku-4-5", "claude-haiku-4.5")
|
|
.replace(/gemini-3\.1-pro(?!-)/g, "gemini-3.1-pro-preview")
|
|
}
|
|
|
|
export function transformModelForProvider(provider: string, model: string): string {
|
|
if (provider === "vercel") {
|
|
const slashIndex = model.indexOf("/")
|
|
if (slashIndex !== -1) {
|
|
const subProvider = model.substring(0, slashIndex)
|
|
const subModel = model.substring(slashIndex + 1)
|
|
return `${subProvider}/${transformModelForGateway(subModel)}`
|
|
}
|
|
const subProvider = inferSubProvider(model)
|
|
if (subProvider) {
|
|
return `${subProvider}/${transformModelForGateway(model)}`
|
|
}
|
|
return model
|
|
}
|
|
if (provider === "github-copilot") {
|
|
return model
|
|
.replace("claude-opus-4-6", "claude-opus-4.6")
|
|
.replace("claude-sonnet-4-6", "claude-sonnet-4.6")
|
|
.replace("claude-sonnet-4-5", "claude-sonnet-4.5")
|
|
.replace("claude-haiku-4-5", "claude-haiku-4.5")
|
|
.replace("claude-sonnet-4", "claude-sonnet-4")
|
|
.replace(/gemini-3\.1-pro(?!-)/g, "gemini-3.1-pro-preview")
|
|
.replace(/gemini-3-flash(?!-)/g, "gemini-3-flash-preview")
|
|
}
|
|
if (provider === "google") {
|
|
return model
|
|
.replace(/gemini-3\.1-pro(?!-)/g, "gemini-3.1-pro-preview")
|
|
.replace(/gemini-3-flash(?!-)/g, "gemini-3-flash-preview")
|
|
}
|
|
if (provider === "anthropic") {
|
|
return model
|
|
.replace("claude-opus-4-6", "claude-opus-4.6")
|
|
.replace("claude-sonnet-4-6", "claude-sonnet-4.6")
|
|
.replace("claude-haiku-4-5", "claude-haiku-4.5")
|
|
}
|
|
return model
|
|
}
|