(textRequest *gemini.GeminiChatRequest)
| 45 | } |
| 46 | |
| 47 | func checkGeminiInputSensitive(textRequest *gemini.GeminiChatRequest) ([]string, error) { |
| 48 | var inputTexts []string |
| 49 | for _, content := range textRequest.Contents { |
| 50 | for _, part := range content.Parts { |
| 51 | if part.Text != "" { |
| 52 | inputTexts = append(inputTexts, part.Text) |
| 53 | } |
| 54 | } |
| 55 | } |
| 56 | if len(inputTexts) == 0 { |
| 57 | return nil, nil |
| 58 | } |
| 59 | |
| 60 | sensitiveWords, err := service.CheckSensitiveInput(inputTexts) |
| 61 | return sensitiveWords, err |
| 62 | } |
| 63 | |
| 64 | func getGeminiInputTokens(req *gemini.GeminiChatRequest, info *relaycommon.RelayInfo) int { |
| 65 | // 计算输入 token 数量 |
no test coverage detected