fix(provider): use gateway-specific model IDs for vercel transform

The gateway uses google/gemini-3-flash (no -preview suffix) unlike the
direct Google API. Give the vercel provider its own transform instead
of delegating to sub-provider transforms blindly.
This commit is contained in:
Matan Kushner
2026-04-13 12:51:56 +09:00
parent 6f72ec754d
commit 542dc890f9
3 changed files with 21 additions and 12 deletions
+11 -2
View File
@@ -6,17 +6,26 @@ function inferSubProvider(model: string): string | undefined {
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}/${transformModelForProvider(subProvider, subModel)}`
return `${subProvider}/${transformModelForGateway(subModel)}`
}
const subProvider = inferSubProvider(model)
if (subProvider) {
return `${subProvider}/${transformModelForProvider(subProvider, model)}`
return `${subProvider}/${transformModelForGateway(model)}`
}
return model
}