feat: Add Gemini safety settings configuration support (close #703)

2025-02-26 16:54:43 +08:00
parent f451268830
commit e19b244e73
7 changed files with 267 additions and 22 deletions
--- a/relay/channel/gemini/constant.go
+++ b/relay/channel/gemini/constant.go
@@ -20,4 +20,12 @@ var ModelList = []string{
 	"imagen-3.0-generate-002",
 }

+var SafetySettingList = []string{
+	"HARM_CATEGORY_HARASSMENT",
+	"HARM_CATEGORY_VIOLENCE",
+	"HARM_CATEGORY_SEXUALLY_EXPLICIT",
+	"HARM_CATEGORY_DANGEROUS_CONTENT",
+	"HARM_CATEGORY_CIVIC_INTEGRITY",
+}
+
 var ChannelName = "google gemini"