Explorar o código

feat:错误信息为配置时error_code为0

fuwencai hai 1 ano
pai
achega
49f2d965ac
Modificáronse 1 ficheiros con 11 adicións e 5 borrados
  1. 11 5
      internal/model/answer.go

+ 11 - 5
internal/model/answer.go

@@ -105,11 +105,7 @@ func (l *cAnswer) FindAnswer(ctx context.Context, answerReq *v1.FindAnswerReq) (
 	}
 	if from != Answer_ChatGPT {
 		if errMsg != nil {
-			if errMsg.Error() == g.Cfg().MustGet(ctx, "limit.errMsg").String() {
-				rs = &v1.FindAnswerRes{ErrorCode: 0, ErrorMsg: errMsg.Error(), Data: nil}
-			} else {
-				rs = &v1.FindAnswerRes{ErrorCode: -1, ErrorMsg: errMsg.Error(), Data: nil}
-			}
+			rs = &v1.FindAnswerRes{ErrorCode: -1, ErrorMsg: errMsg.Error(), Data: nil}
 		} else {
 			rs = &v1.FindAnswerRes{ErrorCode: 0, ErrorMsg: "", Data: g.Map{"id": encrypt.SE.Encode2Hex(fmt.Sprintf("%d", replyId)), "reply": content, "isEnd": true}}
 		}
@@ -144,6 +140,16 @@ func (l *cAnswer) FindAnswer(ctx context.Context, answerReq *v1.FindAnswerReq) (
 			rs = &v1.FindAnswerRes{ErrorCode: -1, ErrorMsg: finalReply}
 		}
 	}
+	if rs.ErrorMsg != "" {
+		exceedMsg := g.Cfg().MustGet(ctx, "limit.exceedMsg").String()
+		fswMsg := g.Cfg().MustGet(ctx, "limit.fswMsg").String()
+		blackMsg := g.Cfg().MustGet(ctx, "limit.blackMsg").String()
+		errMsg := g.Cfg().MustGet(ctx, "limit.errMsg").String()
+		emptyMsg := g.Cfg().MustGet(ctx, "limit.emptyMsg").String()
+		if rs.ErrorMsg == exceedMsg || rs.ErrorMsg == fswMsg || rs.ErrorMsg == blackMsg || rs.ErrorMsg == errMsg || rs.ErrorMsg == emptyMsg {
+			rs.ErrorCode = 0
+		}
+	}
 	return rs, nil
 }