diff --git a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx index f3635977a..528d01049 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 645189d52..3e3e797e7 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 a35adab84..aeb8b432f 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 d96f02928..34dc20b92 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 07100fae5..917969f41 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