Prechádzať zdrojové kódy

wip:删除单子段,返回过滤敏感词

wangkaiyue 2 rokov pred
rodič
commit
b4255ace1d
1 zmenil súbory, kde vykonal 3 pridanie a 2 odobranie
  1. 3 2
      internal/model/ws.go

+ 3 - 2
internal/model/ws.go

@@ -115,6 +115,7 @@ func (m *WsChat) Handle(ws *ghttp.WebSocket, msg []byte) {
 				break
 			}
 			if _, data := parseEventStream(line); data != nil {
+				data.Response = fsw.Repl(data.Response)
 				if data.Finished {
 					replyId := ChatHistory.Save(m.Ctx, &ChatRecord{
 						Content:    data.Response,
@@ -125,9 +126,9 @@ func (m *WsChat) Handle(ws *ghttp.WebSocket, msg []byte) {
 						Item:       Answer_ChatGPT,
 						CreateTime: time.Now().Format(date.Date_Full_Layout),
 					})
-					_ = ws.WriteJSON(g.Map{"error_code": 0, "error_msg": "", "data": g.Map{"id": encrypt.SE.Encode2Hex(fmt.Sprintf("%d", replyId)), "reply": data.Response, "replyByte": data.Delta, "isEnd": data.Finished}})
+					_ = ws.WriteJSON(g.Map{"error_code": 0, "error_msg": "", "data": g.Map{"id": encrypt.SE.Encode2Hex(fmt.Sprintf("%d", replyId)), "reply": data.Response, "isEnd": data.Finished}})
 				} else {
-					_ = ws.WriteJSON(g.Map{"error_code": 0, "error_msg": "", "data": g.Map{"reply": data.Response, "replyByte": data.Delta, "isEnd": data.Finished}})
+					_ = ws.WriteJSON(g.Map{"error_code": 0, "error_msg": "", "data": g.Map{"reply": data.Response, "isEnd": data.Finished}})
 				}
 			}
 		}