support gemini-2.0-flash-exp...

dupl · Dec 20, 2024 · 9192656 · 9192656
1 parent d64d190
commit 9192656
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git 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
@@ -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");