diff --git a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx index 5a2affb4fcf484742c9842f4fb15beb8744953a7..e7a873b9de47e993bace1b43bb8639d3738d8ca6 100644 --- a/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx +++ b/frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx @@ -35,6 +35,7 @@ export default function GeminiLLMOptions({ settings }) { "gemini-1.0-pro", "gemini-1.5-pro-latest", "gemini-1.5-flash-latest", + "gemini-1.5-pro-exp-0801", ].map((model) => { return ( <option key={model} value={model}> diff --git a/frontend/src/hooks/useGetProvidersModels.js b/frontend/src/hooks/useGetProvidersModels.js index c56f42fdca0ed34212da3f2cd5ecbc06727786dd..fc0311a592c1df8977b3ddc262a37c5b66379b56 100644 --- a/frontend/src/hooks/useGetProvidersModels.js +++ b/frontend/src/hooks/useGetProvidersModels.js @@ -16,6 +16,7 @@ const PROVIDER_DEFAULT_MODELS = { "gemini-1.0-pro", "gemini-1.5-pro-latest", "gemini-1.5-flash-latest", + "gemini-1.5-pro-exp-0801", ], anthropic: [ "claude-instant-1.2", diff --git a/server/utils/AiProviders/gemini/index.js b/server/utils/AiProviders/gemini/index.js index f2a948428cfcdb52b32e60b92a37a3b3556a3869..b78c8db6ef27e805b2eb6321a27a85eca17aa330 100644 --- a/server/utils/AiProviders/gemini/index.js +++ b/server/utils/AiProviders/gemini/index.js @@ -96,7 +96,9 @@ class GeminiLLM { case "gemini-1.5-flash-latest": return 1_048_576; case "gemini-1.5-pro-latest": - return 1_048_576; + return 2_097_152; + case "gemini-1.5-pro-exp-0801": + return 2_097_152; default: return 30_720; // assume a gemini-pro model } @@ -108,6 +110,7 @@ class GeminiLLM { "gemini-1.0-pro", "gemini-1.5-pro-latest", "gemini-1.5-flash-latest", + "gemini-1.5-pro-exp-0801", ]; return validModels.includes(modelName); } diff --git a/server/utils/helpers/updateENV.js b/server/utils/helpers/updateENV.js index afacb7279fb86c41a70a418293c73bfc0200408c..f1cedb707d8b7596669aee3accf5a9c4943be39d 100644 --- a/server/utils/helpers/updateENV.js +++ b/server/utils/helpers/updateENV.js @@ -585,6 +585,7 @@ function validGeminiModel(input = "") { "gemini-1.0-pro", "gemini-1.5-pro-latest", "gemini-1.5-flash-latest", + "gemini-1.5-pro-exp-0801", ]; return validModels.includes(input) ? null