From f8e91e1ffacc501085849b8f5b6cfb0784dcfa13 Mon Sep 17 00:00:00 2001
From: timothycarambat <rambat1010@gmail.com>
Date: Wed, 11 Dec 2024 16:52:31 -0800
Subject: [PATCH] patch gemini-2.0-key

---
 .../src/components/LLMSelection/GeminiLLMOptions/index.jsx    | 2 +-
 frontend/src/hooks/useGetProvidersModels.js                   | 2 +-
 server/utils/AiProviders/gemini/index.js                      | 4 ++--
 server/utils/AiProviders/modelMap.js                          | 2 +-
 server/utils/helpers/updateENV.js                             | 2 +-
 5 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx
index 1533a6c5ae..79c72fffa6 100644
--- a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx
+++ b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx
@@ -54,7 +54,7 @@ export default function GeminiLLMOptions({ settings }) {
                     "gemini-exp-1121",
                     "gemini-exp-1206",
                     "learnlm-1.5-pro-experimental",
-                    "gemini-2.0-flash-exp-1206",
+                    "gemini-2.0-flash-exp",
                   ].map((model) => {
                     return (
                       <option key={model} value={model}>
diff --git a/frontend/src/hooks/useGetProvidersModels.js b/frontend/src/hooks/useGetProvidersModels.js
index 29615a41ef..5b98911d48 100644
--- a/frontend/src/hooks/useGetProvidersModels.js
+++ b/frontend/src/hooks/useGetProvidersModels.js
@@ -24,7 +24,7 @@ const PROVIDER_DEFAULT_MODELS = {
     "gemini-exp-1121",
     "gemini-exp-1206",
     "learnlm-1.5-pro-experimental",
-    "gemini-2.0-flash-exp-1206",
+    "gemini-2.0-flash-exp",
   ],
   anthropic: [
     "claude-instant-1.2",
diff --git a/server/utils/AiProviders/gemini/index.js b/server/utils/AiProviders/gemini/index.js
index 4ce374d5fd..28349f492b 100644
--- a/server/utils/AiProviders/gemini/index.js
+++ b/server/utils/AiProviders/gemini/index.js
@@ -30,7 +30,7 @@ class GeminiLLM {
           "gemini-exp-1121",
           "gemini-exp-1206",
           "learnlm-1.5-pro-experimental",
-          "gemini-2.0-flash-exp-1206",
+          "gemini-2.0-flash-exp",
         ].includes(this.model)
           ? "v1beta"
           : "v1",
@@ -120,7 +120,7 @@ class GeminiLLM {
       "gemini-exp-1121",
       "gemini-exp-1206",
       "learnlm-1.5-pro-experimental",
-      "gemini-2.0-flash-exp-1206",
+      "gemini-2.0-flash-exp",
     ];
     return validModels.includes(modelName);
   }
diff --git a/server/utils/AiProviders/modelMap.js b/server/utils/AiProviders/modelMap.js
index bff2f262af..153285d4c7 100644
--- a/server/utils/AiProviders/modelMap.js
+++ b/server/utils/AiProviders/modelMap.js
@@ -38,7 +38,7 @@ const MODEL_MAP = {
     "gemini-exp-1121": 32_767,
     "gemini-exp-1206": 32_767,
     "learnlm-1.5-pro-experimental": 32_767,
-    "gemini-2.0-flash-exp-1206": 1_048_576,
+    "gemini-2.0-flash-exp": 1_048_576,
   },
   groq: {
     "gemma2-9b-it": 8192,
diff --git a/server/utils/helpers/updateENV.js b/server/utils/helpers/updateENV.js
index 312cbd4b74..3cfc13e6e1 100644
--- a/server/utils/helpers/updateENV.js
+++ b/server/utils/helpers/updateENV.js
@@ -733,7 +733,7 @@ function validGeminiModel(input = "") {
     "gemini-exp-1121",
     "gemini-exp-1206",
     "learnlm-1.5-pro-experimental",
-    "gemini-2.0-flash-exp-1206",
+    "gemini-2.0-flash-exp",
   ];
   return validModels.includes(input)
     ? null