浏览代码

wip:历史记录截取

wangkaiyue 2 年之前
父节点
当前提交
f941aa5064
共有 1 个文件被更改,包括 6 次插入3 次删除
  1. 6 3
      internal/model/question.go

+ 6 - 3
internal/model/question.go

@@ -40,15 +40,18 @@ type QuestionReq struct {
 // ParseHistoryFsw 过滤历史记录敏感词
 func (r *BaseQuestion) ParseHistoryFsw() {
 	var newHistory [][]string
-	for _, h := range r.History {
+	for i := len(r.History) - 1; i >= 0; i++ {
 		var pass bool = true
-		for _, v := range h {
+		for _, v := range r.History[i] {
 			if fsw.Match(v) {
 				pass = false
 			}
 		}
 		if pass {
-			newHistory = append(newHistory, h)
+			newHistory = append(newHistory, r.History[i])
+			if len(newHistory) > 3 {
+				break
+			}
 		}
 	}
 	r.History = newHistory