From 9192656c6714f34d295ecc4f58633cfede76baf8 Mon Sep 17 00:00:00 2001
From: dupl <dupl@mail.ustc.edu.cn>
Date: Fri, 20 Dec 2024 16:09:18 +0800
Subject: [PATCH] support gemini-2.0-flash-exp...

---
 app/constant.ts | 15 +++++++++++++++
 app/utils.ts    |  3 +++
 2 files changed, 18 insertions(+)

diff --git a/app/constant.ts b/app/constant.ts
index 25c8d98eae3..7e9cf82e206 100644
--- a/app/constant.ts
+++ b/app/constant.ts
@@ -324,6 +324,21 @@ const googleModels = [
   "gemini-exp-1121",
   "learnlm-1.5-pro-experimental",
   "gemini-pro-vision",
+  "gemini-1.5-flash",
+  "gemini-1.5-flash-002",
+  "gemini-1.5-flash-8b",
+  "gemini-1.5-flash-exp-0827",
+  "gemini-1.5-pro",
+  "gemini-1.5-pro-002",
+  "gemini-1.5-pro-exp-0827",
+  "gemini-2.0-flash-exp",
+  "gemini-exp-1206",
+  "gemini-ultra",
+  "text-embedding-004",
+  "aqa",
+  "gemma-2-2b-it",
+  "gemma-2-9b-it",
+  "gemma-2-27b-it",
 ];
 
 const anthropicModels = [
diff --git a/app/utils.ts b/app/utils.ts
index c52e85265b2..8b1cfd49a91 100644
--- a/app/utils.ts
+++ b/app/utils.ts
@@ -265,6 +265,9 @@ export function isVisionModel(model: string) {
     "qwen-vl",
     "qwen2-vl",
     "glm-4v",
+    "gemini-2.0",
+    "gemini-ultra",
+    "gemma-2",
   ];
   const isGpt4Turbo =
     model.includes("gpt-4-turbo") && !model.includes("preview");