From a53d04616c808f230fdc02fc5bb1d6d751ff2c8b Mon Sep 17 00:00:00 2001
From: timothycarambat <rambat1010@gmail.com>
Date: Wed, 20 Nov 2024 09:51:38 -0800
Subject: [PATCH] Add more expiermental models from Gemini

---
 .../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, 12 insertions(+)

diff --git a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx
index f3635977a81..528d01049ff 100644
--- a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx
+++ b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx
@@ -50,6 +50,8 @@ export default function GeminiLLMOptions({ settings }) {
                     "gemini-1.5-pro-exp-0827",
                     "gemini-1.5-flash-exp-0827",
                     "gemini-1.5-flash-8b-exp-0827",
+                    "gemini-exp-1114",
+                    "learnlm-1.5-pro-experimental",
                   ].map((model) => {
                     return (
                       <option key={model} value={model}>
diff --git a/frontend/src/hooks/useGetProvidersModels.js b/frontend/src/hooks/useGetProvidersModels.js
index 645189d5264..3e3e797e718 100644
--- a/frontend/src/hooks/useGetProvidersModels.js
+++ b/frontend/src/hooks/useGetProvidersModels.js
@@ -20,6 +20,8 @@ const PROVIDER_DEFAULT_MODELS = {
     "gemini-1.5-pro-exp-0827",
     "gemini-1.5-flash-exp-0827",
     "gemini-1.5-flash-8b-exp-0827",
+    "gemini-exp-1114",
+    "learnlm-1.5-pro-experimental",
   ],
   anthropic: [
     "claude-instant-1.2",
diff --git a/server/utils/AiProviders/gemini/index.js b/server/utils/AiProviders/gemini/index.js
index a35adab849b..aeb8b432fe3 100644
--- a/server/utils/AiProviders/gemini/index.js
+++ b/server/utils/AiProviders/gemini/index.js
@@ -26,6 +26,8 @@ class GeminiLLM {
           "gemini-1.5-pro-exp-0827",
           "gemini-1.5-flash-exp-0827",
           "gemini-1.5-flash-8b-exp-0827",
+          "gemini-exp-1114",
+          "learnlm-1.5-pro-experimental",
         ].includes(this.model)
           ? "v1beta"
           : "v1",
@@ -111,6 +113,8 @@ class GeminiLLM {
       "gemini-1.5-pro-exp-0827",
       "gemini-1.5-flash-exp-0827",
       "gemini-1.5-flash-8b-exp-0827",
+      "gemini-exp-1114",
+      "learnlm-1.5-pro-experimental",
     ];
     return validModels.includes(modelName);
   }
diff --git a/server/utils/AiProviders/modelMap.js b/server/utils/AiProviders/modelMap.js
index d96f02928e5..34dc20b926c 100644
--- a/server/utils/AiProviders/modelMap.js
+++ b/server/utils/AiProviders/modelMap.js
@@ -34,6 +34,8 @@ const MODEL_MAP = {
     "gemini-1.5-pro-exp-0827": 2_097_152,
     "gemini-1.5-flash-exp-0827": 1_048_576,
     "gemini-1.5-flash-8b-exp-0827": 1_048_576,
+    "gemini-exp-1114": 32_767,
+    "learnlm-1.5-pro-experimental": 32_767,
   },
   groq: {
     "gemma2-9b-it": 8192,
diff --git a/server/utils/helpers/updateENV.js b/server/utils/helpers/updateENV.js
index 07100fae542..917969f4189 100644
--- a/server/utils/helpers/updateENV.js
+++ b/server/utils/helpers/updateENV.js
@@ -701,6 +701,8 @@ function validGeminiModel(input = "") {
     "gemini-1.5-pro-exp-0827",
     "gemini-1.5-flash-exp-0827",
     "gemini-1.5-flash-8b-exp-0827",
+    "gemini-exp-1114",
+    "learnlm-1.5-pro-experimental",
   ];
   return validModels.includes(input)
     ? null