🪙 fix(Google): Update maxOutputTokens Condition (#2434)

- generalize condition to fix max output tokens for gemini-1.5
This commit is contained in:
Danny Avila 2024-04-16 10:23:13 -04:00 committed by GitHub
parent 9d854dac07
commit bf4e64ce63
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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;