diff --git a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx index 3b903d27314eb80feb7d8c497053418a957a5a1a..1533a6c5aee26e83093abb7510072f05786ef444 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 52e84122bae04a5a2fa8696b982eb0704a5a344d..29615a41ef67bec9eaae27f822b3d292e2091dce 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 055b455c1167cc8750900b11ec0081106efe7ad8..4ce374d5fda42b3da511a868777d47a4a750d9fe 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 f9bf00672c6757b19d074298c9c87578c2124690..bff2f262afd365855563e0cb06fad0546fe58542 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 2ae930a6c3e87fbad51b85710471be8fc096086e..312cbd4b74614933a82fe1cd2b2c4a3d330c3570 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