From 657e44b5016ebe766071dbcb3eb5a9f1b6b8efb7 Mon Sep 17 00:00:00 2001
From: Hk-Gosuto <yominogosuto@gmail.com>
Date: Fri, 14 Feb 2025 13:15:46 +0800
Subject: [PATCH] feat: update vision model list

---
 app/utils.ts | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/app/utils.ts b/app/utils.ts
index 80ab75053..5e8ee9ba2 100644
--- 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))
   );
 }