apply gemini safety settings as extra_body in OpenRouter requests (#4256) * apply gemini safety settings as extra_body in OpenRouter requests * revert adding new gemini safety category * Make it actually work --------- Co-authored-by: Cohee <18619528+Cohee1207@users.noreply.github.com>

71a3ad0edb6b5da40926126061a476c7a53381ca

Frédéric Bolvin <frederic@bol.vin>

Signed
Parents: 556fb4ba087a432fea7667079194cbb7bc48aebc
1 files changed, +5 -0Showing whitespace changes
src/endpoints/backends/chat-completions.js+5 -0
@@ -1486,6 +1486,11 @@ router.post('/generate', function (request, response) {
14861486 if (Number.isInteger(cachingAtDepth) && cachingAtDepth >= 0 && isClaude3or4) {
14871487 cachingAtDepthForOpenRouterClaude(request.body.messages, cachingAtDepth, cacheTTL);
14881488 }
1489+
1490+ const isGemini = /google\/gemini/.test(request.body.model);
1491+ if (isGemini) {
1492+ bodyParams['safety_settings'] = GEMINI_SAFETY;
1493+ }
14891494 } else if (request.body.chat_completion_source === CHAT_COMPLETION_SOURCES.CUSTOM) {
14901495 apiUrl = request.body.custom_url;
14911496 apiKey = readSecret(request.user.directories, SECRET_KEYS.CUSTOM);