|
@@ -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
|
|
|
}
|
|
|
|