From bf4e64ce63e65ff35d3784c318db84ee29cfa5cd Mon Sep 17 00:00:00 2001
From: Danny Avila <danny@librechat.ai>
Date: Tue, 16 Apr 2024 10:23:13 -0400
Subject: [PATCH] =?UTF-8?q?=F0=9F=AA=99=20fix(Google):=20Update=20`maxOutp?=
 =?UTF-8?q?utTokens`=20Condition=20(#2434)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- generalize condition to fix max output tokens for gemini-1.5
---
 client/src/components/Endpoints/Settings/Google.tsx | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/client/src/components/Endpoints/Settings/Google.tsx b/client/src/components/Endpoints/Settings/Google.tsx
index 0d77a8d553..2b550894e8 100644
--- a/client/src/components/Endpoints/Settings/Google.tsx
+++ b/client/src/components/Endpoints/Settings/Google.tsx
@@ -22,12 +22,12 @@ export default function Settings({ conversation, setOption, models, readonly }:
   const { model, modelLabel, promptPrefix, temperature, topP, topK, maxOutputTokens } =
     conversation ?? {};
 
-  const isGeminiPro = model?.toLowerCase()?.includes('gemini-pro');
+  const isGemini = model?.toLowerCase()?.includes('gemini');
 
-  const maxOutputTokensMax = isGeminiPro
+  const maxOutputTokensMax = isGemini
     ? google.maxOutputTokens.maxGeminiPro
     : google.maxOutputTokens.max;
-  const maxOutputTokensDefault = isGeminiPro
+  const maxOutputTokensDefault = isGemini
     ? google.maxOutputTokens.defaultGeminiPro
     : google.maxOutputTokens.default;