From 293676feffea07771761c2df8ca37404f80a1bcb Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Fri, 30 Aug 2024 23:32:09 +0300
Subject: [PATCH] Cohere: unbreak requires for old models

---
 src/endpoints/backends/chat-completions.js | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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 = {