gemini: Adjust safety filters - new value "OFF" and new category "HAR…

…M_CATEGORY_CIVIC_INTEGRITY"
Brawl345 · Jan 9, 2025 · 251d348 · 251d348
1 parent 0eec16a
commit 251d348
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/plugin/gemini/api.go b/plugin/gemini/api.go
@@ -26,8 +26,8 @@ type (
 	}
 
 	SafetySetting struct {
-		Category  string `json:"category"`
-		Threshold string `json:"threshold"`
+		Category  string `json:"category"`  // https://ai.google.dev/api/generate-content#v1beta.HarmCategory
+		Threshold string `json:"threshold"` // https://ai.google.dev/api/generate-content#HarmBlockThreshold
 	}
 
 	GenerationConfig struct {

diff --git a/plugin/gemini/gemini.go b/plugin/gemini/gemini.go
@@ -237,19 +237,23 @@ func (p *Plugin) onGemini(b *gotgbot.Bot, c plugin.GobotContext) error {
 		SafetySettings: []SafetySetting{
 			{
 				Category:  "HARM_CATEGORY_HARASSMENT",
-				Threshold: "BLOCK_NONE",
+				Threshold: "OFF",
 			},
 			{
 				Category:  "HARM_CATEGORY_HATE_SPEECH",
-				Threshold: "BLOCK_NONE",
+				Threshold: "OFF",
 			},
 			{
 				Category:  "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-				Threshold: "BLOCK_NONE",
+				Threshold: "OFF",
 			},
 			{
 				Category:  "HARM_CATEGORY_DANGEROUS_CONTENT",
-				Threshold: "BLOCK_NONE",
+				Threshold: "OFF",
+			},
+			{
+				Category:  "HARM_CATEGORY_CIVIC_INTEGRITY",
+				Threshold: "OFF",
 			},
 		},
 		GenerationConfig: GenerationConfig{