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

setting/model_setting.go

@@ -0,0 +1,45 @@
+package setting
+
+import (
+	"encoding/json"
+	"one-api/common"
+)
+
+var geminiSafetySettings = map[string]string{
+	"default":                       "OFF",
+	"HARM_CATEGORY_CIVIC_INTEGRITY": "BLOCK_NONE",
+}
+
+func GetGeminiSafetySetting(key string) string {
+	if value, ok := geminiSafetySettings[key]; ok {
+		return value
+	}
+	return geminiSafetySettings["default"]
+}
+
+func GeminiSafetySettingFromJsonString(jsonString string) {
+	geminiSafetySettings = map[string]string{}
+	err := json.Unmarshal([]byte(jsonString), &geminiSafetySettings)
+	if err != nil {
+		geminiSafetySettings = map[string]string{
+			"default":                       "OFF",
+			"HARM_CATEGORY_CIVIC_INTEGRITY": "BLOCK_NONE",
+		}
+	}
+	// check must have default
+	if _, ok := geminiSafetySettings["default"]; !ok {
+		geminiSafetySettings["default"] = common.GeminiSafetySetting
+	}
+}
+
+func GeminiSafetySettingsJsonString() string {
+	// check must have default
+	if _, ok := geminiSafetySettings["default"]; !ok {
+		geminiSafetySettings["default"] = common.GeminiSafetySetting
+	}
+	jsonString, err := json.Marshal(geminiSafetySettings)
+	if err != nil {
+		return "{}"
+	}
+	return string(jsonString)
+}