feat: publish v1.0.0-rc.36-custom.1

relay/common/relay_info_gemini_settings_test.go

@@ -0,0 +1,25 @@
+package common
+
+import (
+	"testing"
+
+	"github.com/QuantumNous/new-api/setting/model_setting"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRelayInfoConvOptionsUsesNormalizedGeminiSafetySettings(t *testing.T) {
+	settings := model_setting.GetGeminiSettings()
+	original := settings.SafetySettings
+	t.Cleanup(func() {
+		settings.SafetySettings = original
+	})
+	settings.SafetySettings = map[string]string{
+		"HARM_CATEGORY_HATE_SPEECH":       "",
+		"HARM_CATEGORY_DANGEROUS_CONTENT": "BLOCK_ONLY_HIGH",
+	}
+
+	options := (&RelayInfo{}).ConvOptions()
+
+	assert.Equal(t, "OFF", options.Gemini.SafetySetting("HARM_CATEGORY_HATE_SPEECH"))
+	assert.Equal(t, "BLOCK_ONLY_HIGH", options.Gemini.SafetySetting("HARM_CATEGORY_DANGEROUS_CONTENT"))
+}