Add gemini-exp-1121

timothycarambat · timothycarambat · commit 246152c02414 · 2024-11-21T11:02:43.000-08:00
resolves #2657
diff --git a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx
@@ -51,6 +51,7 @@ export default function GeminiLLMOptions({ settings }) {
                     "gemini-1.5-flash-exp-0827",
                     "gemini-1.5-flash-8b-exp-0827",
                     "gemini-exp-1114",
+                    "gemini-exp-1121",
                     "learnlm-1.5-pro-experimental",
                   ].map((model) => {
                     return (
diff --git a/frontend/src/hooks/useGetProvidersModels.js b/frontend/src/hooks/useGetProvidersModels.js
@@ -21,6 +21,7 @@ const PROVIDER_DEFAULT_MODELS = {
     "gemini-1.5-flash-exp-0827",
     "gemini-1.5-flash-8b-exp-0827",
     "gemini-exp-1114",
+    "gemini-exp-1121",
     "learnlm-1.5-pro-experimental",
   ],
   anthropic: [
diff --git a/server/utils/AiProviders/gemini/index.js b/server/utils/AiProviders/gemini/index.js
@@ -114,6 +114,7 @@ class GeminiLLM {
       "gemini-1.5-flash-exp-0827",
       "gemini-1.5-flash-8b-exp-0827",
       "gemini-exp-1114",
+      "gemini-exp-1121",
       "learnlm-1.5-pro-experimental",
     ];
     return validModels.includes(modelName);
diff --git a/server/utils/AiProviders/modelMap.js b/server/utils/AiProviders/modelMap.js
@@ -35,6 +35,7 @@ const MODEL_MAP = {
     "gemini-1.5-flash-exp-0827": 1_048_576,
     "gemini-1.5-flash-8b-exp-0827": 1_048_576,
     "gemini-exp-1114": 32_767,
+    "gemini-exp-1121": 32_767,
     "learnlm-1.5-pro-experimental": 32_767,
   },
   groq: {
diff --git a/server/utils/helpers/updateENV.js b/server/utils/helpers/updateENV.js
@@ -702,6 +702,7 @@ function validGeminiModel(input = "") {
     "gemini-1.5-flash-exp-0827",
     "gemini-1.5-flash-8b-exp-0827",
     "gemini-exp-1114",
+    "gemini-exp-1121",
     "learnlm-1.5-pro-experimental",
   ];
   return validModels.includes(input)
