Эх сурвалжийг харах

wip:增加敏感词回复

wangkaiyue 2 жил өмнө
parent
commit
a1c4caa387

+ 5 - 2
internal/model/ws.go

@@ -34,9 +34,12 @@ func (m *WsChat) Handle(ws *ghttp.WebSocket, msg []byte) {
 		glog.Errorf(m.Ctx, "%d 接收消息Unmarshal出错:%v", jSession.AccountId, err)
 		return
 	}
-	// 问题敏感词过滤
-	req.Prompt = fsw.Repl(req.Prompt)
+
 	reply, errMsg := func() (string, error) {
+		// 问题敏感词过滤
+		if fsw.Match(req.Prompt) {
+			return "", fmt.Errorf(g.Cfg().MustGet(m.Ctx, "limit.fswMsg", "您的问题可能包含敏感词汇,请修改后再提问!").String())
+		}
 		reply, from, err := Question.DetailQuestion(m.Ctx, req)
 		// 回答敏感词过滤
 		reply = fsw.Repl(reply)

+ 2 - 1
manifest/config/config.yaml

@@ -48,7 +48,8 @@ database:
 
 limit:
   rateMinute: 3 #每分钟速率
-  exceedMsg: 您的提问过于频繁,请等待一会儿再试 # 超出提示语句
+  exceedMsg: "您的提问过于频繁,请等待一会儿再试" # 超出提示语句
+  fswMsg: "您的问题可能包含敏感词汇,请修改后再提问!"# 敏感词回复
 
 logger:
   path: "logs"        # 日志文件路径。默认为空,表示关闭,仅输出到终端