Merge pull request #3271 from tincansimagine/geminifeat2 Fix: Ensure "OFF" safety settings are applied to gemini-2.0-flash-exp for unfiltered responses

f91e1b3d2feac37c2e06e5ea1b51eb0e1ef628c9

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed
Parents: 0f03dcfc0fb6b7bc48cab580c230074dff2316f1 b66c4d720691e907c78f29cc90b1dae8dd7315d3
1 files changed, +7 -1Showing whitespace changes
src/endpoints/backends/chat-completions.js+7 -1
@@ -308,9 +308,15 @@ async function sendMakerSuiteRequest(request, response) {
308 ) && request.body.use_makersuite_sysprompt;308 ) && request.body.use_makersuite_sysprompt;
309309
310 const prompt = convertGooglePrompt(request.body.messages, model, should_use_system_prompt, getPromptNames(request));310 const prompt = convertGooglePrompt(request.body.messages, model, should_use_system_prompt, getPromptNames(request));
311 let safetySettings = GEMINI_SAFETY;
312
313 if (model.includes('gemini-2.0-flash-exp')) {
314 safetySettings = GEMINI_SAFETY.map(setting => ({ ...setting, threshold: 'OFF' }));
315 }
316
311 let body = {317 let body = {
312 contents: prompt.contents,318 contents: prompt.contents,
313 safetySettings: GEMINI_SAFETY,319 safetySettings: safetySettings,
314 generationConfig: generationConfig,320 generationConfig: generationConfig,
315 };321 };
316322