diff --git a/app/constant.ts b/app/constant.ts index 77d83c378..6a51cae9d 100644 --- a/app/constant.ts +++ b/app/constant.ts @@ -161,6 +161,7 @@ const openaiModels = [ const googleModels = [ "gemini-1.0-pro", "gemini-1.5-pro-latest", + "gemini-1.5-flash-latest", "gemini-pro-vision", ]; diff --git a/app/utils.ts b/app/utils.ts index b509c3f0d..0b01b91f1 100644 --- a/app/utils.ts +++ b/app/utils.ts @@ -296,13 +296,12 @@ export function isVisionModel(model: string) { "vision", "claude-3", "gemini-1.5-pro", - "gpt-4-turbo", - "gpt-4o", + "gemini-1.5-flash", ]; - const isGpt4TurboPreview = model === "gpt-4-turbo-preview"; + const isGpt4Turbo = + model.includes("gpt-4-turbo") && !model.includes("preview"); return ( - visionKeywords.some((keyword) => model.includes(keyword)) && - !isGpt4TurboPreview + visionKeywords.some((keyword) => model.includes(keyword)) || isGpt4Turbo ); }