fix: cohere SafetyMode

relay/channel/cohere/dto.go

@@ -8,7 +8,7 @@ type CohereRequest struct {
 	Message     string        `json:"message"`
 	Message     string        `json:"message"`
 	Stream      bool          `json:"stream"`
 	Stream      bool          `json:"stream"`
 	MaxTokens   int           `json:"max_tokens"`
 	MaxTokens   int           `json:"max_tokens"`
-	SafetyMode  string        `json:"safety_mode"`
+	SafetyMode  string        `json:"safety_mode,omitempty"`
 }
 }
 
 
 type ChatHistory struct {
 type ChatHistory struct {

relay/channel/cohere/relay-cohere.go

@@ -22,7 +22,9 @@ func requestOpenAI2Cohere(textRequest dto.GeneralOpenAIRequest) *CohereRequest {
 		Message:     "",
 		Message:     "",
 		Stream:      textRequest.Stream,
 		Stream:      textRequest.Stream,
 		MaxTokens:   textRequest.GetMaxTokens(),
 		MaxTokens:   textRequest.GetMaxTokens(),
-		SafetyMode:  common.CohereSafetySetting,
+	}
+	if common.CohereSafetySetting != "NONE" {
+		cohereReq.SafetyMode = common.CohereSafetySetting
 	}
 	}
 	if cohereReq.MaxTokens == 0 {
 	if cohereReq.MaxTokens == 0 {
 		cohereReq.MaxTokens = 4000
 		cohereReq.MaxTokens = 4000