fix: cohere SafetyMode

This commit is contained in:
CalciumIon
2024-09-11 19:12:32 +08:00
parent 89713975ca
commit f50cf75d1a
2 changed files with 4 additions and 2 deletions
+1 -1
View File
@@ -8,7 +8,7 @@ type CohereRequest struct {
Message string `json:"message"`
Stream bool `json:"stream"`
MaxTokens int `json:"max_tokens"`
SafetyMode string `json:"safety_mode"`
SafetyMode string `json:"safety_mode,omitempty"`
}
type ChatHistory struct {
+3 -1
View File
@@ -22,7 +22,9 @@ func requestOpenAI2Cohere(textRequest dto.GeneralOpenAIRequest) *CohereRequest {
Message: "",
Stream: textRequest.Stream,
MaxTokens: textRequest.GetMaxTokens(),
SafetyMode: common.CohereSafetySetting,
}
if common.CohereSafetySetting != "NONE" {
cohereReq.SafetyMode = common.CohereSafetySetting
}
if cohereReq.MaxTokens == 0 {
cohereReq.MaxTokens = 4000