CaIon 1 год назад
Родитель
Сommit
47b9f48544
2 измененных файлов с 4 добавлено и 4 удалено
  1. 1 1
      relay/channel/openai/relay-openai.go
  2. 3 3
      service/sensitive.go

+ 1 - 1
relay/channel/openai/relay-openai.go

@@ -56,7 +56,7 @@ func OpenaiStreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*d
 			sensitive := false
 			if checkSensitive {
 				// check sensitive
-				sensitive, _, data = service.SensitiveWordReplace(data, constant.StopOnSensitiveEnabled)
+				sensitive, _, data = service.SensitiveWordReplace(data, false)
 			}
 			dataChan <- data
 			data = data[6:]

+ 3 - 3
service/sensitive.go

@@ -25,15 +25,15 @@ func SensitiveWordContains(text string) (bool, []string) {
 
 // SensitiveWordReplace 敏感词替换,返回是否包含敏感词和替换后的文本
 func SensitiveWordReplace(text string, returnImmediately bool) (bool, []string, string) {
-	text = strings.ToLower(text)
+	checkText := strings.ToLower(text)
 	m := initAc()
-	hits := m.MultiPatternSearch([]rune(text), returnImmediately)
+	hits := m.MultiPatternSearch([]rune(checkText), returnImmediately)
 	if len(hits) > 0 {
 		words := make([]string, 0)
 		for _, hit := range hits {
 			pos := hit.Pos
 			word := string(hit.Word)
-			text = text[:pos] + " *###* " + text[pos+len(word):]
+			text = text[:pos] + "*###*" + text[pos+len(word):]
 			words = append(words, word)
 		}
 		return true, words, text