add profanity content filter

sashabaranov · Oct 1, 2024 · 8033f54 · 8033f54
1 parent 37540d2
commit 8033f54
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/chat.go b/chat.go
@@ -46,12 +46,18 @@ type JailBreak struct {
 	Detected bool `json:"detected"`
 }
 
+type Profanity struct {
+	Filtered bool `json:"filtered"`
+	Detected bool `json:"detected"`
+}
+
 type ContentFilterResults struct {
 	Hate      Hate      `json:"hate,omitempty"`
 	SelfHarm  SelfHarm  `json:"self_harm,omitempty"`
 	Sexual    Sexual    `json:"sexual,omitempty"`
 	Violence  Violence  `json:"violence,omitempty"`
 	JailBreak JailBreak `json:"jailbreak,omitempty"`
+	Profanity Profanity `json:"profanity,omitempty"`
 }
 
 type PromptAnnotation struct {