diff --git a/src/endpoints/backends/chat-completions.js b/src/endpoints/backends/chat-completions.js index 72b7edbeb..a7689fa1c 100644 --- a/src/endpoints/backends/chat-completions.js +++ b/src/endpoints/backends/chat-completions.js @@ -570,9 +570,13 @@ async function sendCohereRequest(request, response) { documents: [], tools: tools, search_queries_only: false, - safety_mode: 'NONE', }; + const canDoSafetyMode = String(request.body.model).endsWith('08-2024'); + if (canDoSafetyMode) { + requestBody.safety_mode = 'NONE'; + } + console.log('Cohere request:', requestBody); const config = {