feat: update vision model list

Hk-Gosuto · Feb 14, 2025 · b1b8be6 · b1b8be6
1 parent f14e966
commit b1b8be6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/app/utils.ts b/app/utils.ts
@@ -286,13 +286,19 @@ export function isVisionModel(model: string) {
     "gpt-4o",
     "gpt-4o-mini",
   ];
+
+  var googleModels = DEFAULT_MODELS.filter(
+    (model) => model.provider.id === "google",
+  ).map((model) => model.name);
+
   const isGpt4Turbo =
     model.includes("gpt-4-turbo") && !model.includes("preview");
 
   return (
     visionKeywords.some((keyword) => model.includes(keyword)) ||
     isGpt4Turbo ||
-    isDalle3(model)
+    isDalle3(model) ||
+    googleModels.some((keyword) => model.includes(keyword))
   );
 }