diff --git a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx index f3635977a81dc7a7f92dc1c8494d245068be2723..528d01049ff0b317dd34011508980baf3ae92749 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 645189d526444f2a65418097af7a3dde9e500056..3e3e797e718c0d0282af57e7ff7b2e017ce5c68f 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 a35adab849b91f72f1417f9a28a2d87458e911ba..aeb8b432fe3d54f7ddbc8c9ffb10fdc9215b9a36 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 d96f02928e5e3e20d25cb24807e2cc8f3c264070..34dc20b926c9c9b21455020755a582e07208d338 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 07100fae542735f44ed97d00ea1419094d7738a3..917969f4189e49ffe8765d80d70919c1fbb3a269 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