update: safety settings

zhu327 · Jan 1, 2024 · cd1ba65 · cd1ba65
1 parent f519857
commit cd1ba65
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pkg/adapter/chat.go b/pkg/adapter/chat.go
@@ -324,19 +324,19 @@ func setGenaiModelByOpenaiRequest(model *genai.GenerativeModel, req *ChatComplet
 	model.SafetySettings = []*genai.SafetySetting{
 		{
 			Category:  genai.HarmCategoryHarassment,
-			Threshold: genai.HarmBlockOnlyHigh,
+			Threshold: genai.HarmBlockNone,
 		},
 		{
 			Category:  genai.HarmCategoryHateSpeech,
-			Threshold: genai.HarmBlockOnlyHigh,
+			Threshold: genai.HarmBlockNone,
 		},
 		{
 			Category:  genai.HarmCategorySexuallyExplicit,
-			Threshold: genai.HarmBlockOnlyHigh,
+			Threshold: genai.HarmBlockNone,
 		},
 		{
 			Category:  genai.HarmCategoryDangerousContent,
-			Threshold: genai.HarmBlockOnlyHigh,
+			Threshold: genai.HarmBlockNone,
 		},
 	}
 }