Cohere: unbreak requires for old models

293676feffea07771761c2df8ca37404f80a1bcb

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

Parents: b9543054d8b89fdffeee6e1d12d9af7df326531e
1 files changed, +5 -1Ignore whitespace
src/endpoints/backends/chat-completions.js+5 -1
@@ -570,9 +570,13 @@ async function sendCohereRequest(request, response) {
570570 documents: [],
571571 tools: tools,
572572 search_queries_only: false,
573- safety_mode: 'NONE',
574573 };
575574
575+ const canDoSafetyMode = String(request.body.model).endsWith('08-2024');
576+ if (canDoSafetyMode) {
577+ requestBody.safety_mode = 'NONE';
578+ }
579+
576580 console.log('Cohere request:', requestBody);
577581
578582 const config = {