🪙 fix(Google): Update maxOutputTokens Condition (danny-avila#2434)

- generalize condition to fix max output tokens for gemini-1.5
noobskies · Apr 16, 2024 · bf4e64c · bf4e64c
1 parent 9d854da
commit bf4e64c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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;