From 69b672b625cc9fc8e4eba3793610b8c8c5cad46f Mon Sep 17 00:00:00 2001
From: timothycarambat <rambat1010@gmail.com>
Date: Wed, 11 Dec 2024 09:04:29 -0800
Subject: [PATCH] add gemini 1206 and gemini-2.0-flash exp models connect #2788
---
.../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 3b903d273..1533a6c5a 100644
--- a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx
+++ b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx
@@ -52,7 +52,9 @@ export default function GeminiLLMOptions({ settings }) {
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"gemini-exp-1121",
+ "gemini-exp-1206",
"learnlm-1.5-pro-experimental",
+ "gemini-2.0-flash-exp-1206",
].map((model) => {
return (
<option key={model} value={model}>
diff --git a/frontend/src/hooks/useGetProvidersModels.js b/frontend/src/hooks/useGetProvidersModels.js
index 52e84122b..29615a41e 100644
--- a/frontend/src/hooks/useGetProvidersModels.js
+++ b/frontend/src/hooks/useGetProvidersModels.js
@@ -22,7 +22,9 @@ const PROVIDER_DEFAULT_MODELS = {
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"gemini-exp-1121",
+ "gemini-exp-1206",
"learnlm-1.5-pro-experimental",
+ "gemini-2.0-flash-exp-1206",
],
anthropic: [
"claude-instant-1.2",
diff --git a/server/utils/AiProviders/gemini/index.js b/server/utils/AiProviders/gemini/index.js
index 055b455c1..4ce374d5f 100644
--- a/server/utils/AiProviders/gemini/index.js
+++ b/server/utils/AiProviders/gemini/index.js
@@ -28,7 +28,9 @@ class GeminiLLM {
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"gemini-exp-1121",
+ "gemini-exp-1206",
"learnlm-1.5-pro-experimental",
+ "gemini-2.0-flash-exp-1206",
].includes(this.model)
? "v1beta"
: "v1",
@@ -116,7 +118,9 @@ class GeminiLLM {
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"gemini-exp-1121",
+ "gemini-exp-1206",
"learnlm-1.5-pro-experimental",
+ "gemini-2.0-flash-exp-1206",
];
return validModels.includes(modelName);
}
diff --git a/server/utils/AiProviders/modelMap.js b/server/utils/AiProviders/modelMap.js
index f9bf00672..bff2f262a 100644
--- a/server/utils/AiProviders/modelMap.js
+++ b/server/utils/AiProviders/modelMap.js
@@ -36,7 +36,9 @@ const MODEL_MAP = {
"gemini-1.5-flash-8b-exp-0827": 1_048_576,
"gemini-exp-1114": 32_767,
"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,
},
groq: {
"gemma2-9b-it": 8192,
diff --git a/server/utils/helpers/updateENV.js b/server/utils/helpers/updateENV.js
index 2ae930a6c..312cbd4b7 100644
--- a/server/utils/helpers/updateENV.js
+++ b/server/utils/helpers/updateENV.js
@@ -731,7 +731,9 @@ function validGeminiModel(input = "") {
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"gemini-exp-1121",
+ "gemini-exp-1206",
"learnlm-1.5-pro-experimental",
+ "gemini-2.0-flash-exp-1206",
];
return validModels.includes(input)
? null
--
GitLab