refactor(provider): replace per-model string replacements with generic regex

Claude version transforms now use a single regex
claude-(\w+)-(\d+)-(\d+) -> claude-$1-$2.$3 instead of one
.replace() per model. New claude models need zero changes.
This commit is contained in:
Matan Kushner
2026-04-13 14:01:17 +09:00
parent 34e334eabe
commit effae16c14
+19 -23
View File
@@ -9,13 +9,17 @@ function inferSubProvider(model: string): string | undefined {
return undefined return undefined
} }
function transformModelForGateway(model: string): string { const CLAUDE_VERSION_DOT = /claude-(\w+)-(\d+)-(\d+)/g
return model const GEMINI_31_PRO_PREVIEW = /gemini-3\.1-pro(?!-)/g
.replace("claude-opus-4-6", "claude-opus-4.6") const GEMINI_3_FLASH_PREVIEW = /gemini-3-flash(?!-)/g
.replace("claude-sonnet-4-6", "claude-sonnet-4.6")
.replace("claude-sonnet-4-5", "claude-sonnet-4.5") function claudeVersionDot(model: string): string {
.replace("claude-haiku-4-5", "claude-haiku-4.5") return model.replace(CLAUDE_VERSION_DOT, "claude-$1-$2.$3")
.replace(/gemini-3\.1-pro(?!-)/g, "gemini-3.1-pro-preview") }
function applyGatewayTransforms(model: string): string {
return claudeVersionDot(model)
.replace(GEMINI_31_PRO_PREVIEW, "gemini-3.1-pro-preview")
} }
export function transformModelForProvider(provider: string, model: string): string { export function transformModelForProvider(provider: string, model: string): string {
@@ -24,34 +28,26 @@ export function transformModelForProvider(provider: string, model: string): stri
if (slashIndex !== -1) { if (slashIndex !== -1) {
const subProvider = model.substring(0, slashIndex) const subProvider = model.substring(0, slashIndex)
const subModel = model.substring(slashIndex + 1) const subModel = model.substring(slashIndex + 1)
return `${subProvider}/${transformModelForGateway(subModel)}` return `${subProvider}/${applyGatewayTransforms(subModel)}`
} }
const subProvider = inferSubProvider(model) const subProvider = inferSubProvider(model)
if (subProvider) { if (subProvider) {
return `${subProvider}/${transformModelForGateway(model)}` return `${subProvider}/${applyGatewayTransforms(model)}`
} }
return model return model
} }
if (provider === "github-copilot") { if (provider === "github-copilot") {
return model return claudeVersionDot(model)
.replace("claude-opus-4-6", "claude-opus-4.6") .replace(GEMINI_31_PRO_PREVIEW, "gemini-3.1-pro-preview")
.replace("claude-sonnet-4-6", "claude-sonnet-4.6") .replace(GEMINI_3_FLASH_PREVIEW, "gemini-3-flash-preview")
.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") { if (provider === "google") {
return model return model
.replace(/gemini-3\.1-pro(?!-)/g, "gemini-3.1-pro-preview") .replace(GEMINI_31_PRO_PREVIEW, "gemini-3.1-pro-preview")
.replace(/gemini-3-flash(?!-)/g, "gemini-3-flash-preview") .replace(GEMINI_3_FLASH_PREVIEW, "gemini-3-flash-preview")
} }
if (provider === "anthropic") { if (provider === "anthropic") {
return model return claudeVersionDot(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 return model
} }