diff --git a/controller/relay-gemini.go b/controller/relay-gemini.go index 2458458e..3b83becb 100644 --- a/controller/relay-gemini.go +++ b/controller/relay-gemini.go @@ -56,24 +56,24 @@ type GeminiChatGenerationConfig struct { func requestOpenAI2Gemini(textRequest GeneralOpenAIRequest) *GeminiChatRequest { geminiRequest := GeminiChatRequest{ Contents: make([]GeminiChatContent, 0, len(textRequest.Messages)), - //SafetySettings: []GeminiChatSafetySettings{ - // { - // Category: "HARM_CATEGORY_HARASSMENT", - // Threshold: "BLOCK_ONLY_HIGH", - // }, - // { - // Category: "HARM_CATEGORY_HATE_SPEECH", - // Threshold: "BLOCK_ONLY_HIGH", - // }, - // { - // Category: "HARM_CATEGORY_SEXUALLY_EXPLICIT", - // Threshold: "BLOCK_ONLY_HIGH", - // }, - // { - // Category: "HARM_CATEGORY_DANGEROUS_CONTENT", - // Threshold: "BLOCK_ONLY_HIGH", - // }, - //}, + SafetySettings: []GeminiChatSafetySettings{ + { + Category: "HARM_CATEGORY_HARASSMENT", + Threshold: "BLOCK_NONE", + }, + { + Category: "HARM_CATEGORY_HATE_SPEECH", + Threshold: "BLOCK_NONE", + }, + { + Category: "HARM_CATEGORY_SEXUALLY_EXPLICIT", + Threshold: "BLOCK_NONE", + }, + { + Category: "HARM_CATEGORY_DANGEROUS_CONTENT", + Threshold: "BLOCK_NONE", + }, + }, GenerationConfig: GeminiChatGenerationConfig{ Temperature: textRequest.Temperature, TopP: textRequest.TopP,