🐛 修正 ws 错误状体的返回。
This commit is contained in:
parent
0c2903e4f8
commit
5ca6966b2f
@ -15,6 +15,6 @@ const (
|
|||||||
ErrGeneralStart = ErrGeneral + iota
|
ErrGeneralStart = ErrGeneral + iota
|
||||||
ErrInvalidData
|
ErrInvalidData
|
||||||
ErrInternalError
|
ErrInternalError
|
||||||
|
|
||||||
ErrDataNoFound
|
ErrDataNoFound
|
||||||
|
ErrServerDown
|
||||||
)
|
)
|
||||||
|
@ -19,6 +19,7 @@ func init() {
|
|||||||
|
|
||||||
// INGO
|
// INGO
|
||||||
ErrMsgForUser = make(msg)
|
ErrMsgForUser = make(msg)
|
||||||
|
GeneralMsgUserInit(ErrMsgForUser)
|
||||||
AnimalMsgUserInit(ErrMsgForUser)
|
AnimalMsgUserInit(ErrMsgForUser)
|
||||||
EncounterMsgUserInit(ErrMsgForUser)
|
EncounterMsgUserInit(ErrMsgForUser)
|
||||||
KnowledgeMsgUserInit(ErrMsgForUser)
|
KnowledgeMsgUserInit(ErrMsgForUser)
|
||||||
@ -31,3 +32,7 @@ func GeneralMsgInit(m msg) {
|
|||||||
m[ErrInternalError] = "内部服务器错误"
|
m[ErrInternalError] = "内部服务器错误"
|
||||||
m[ErrDataNoFound] = "无数据查询"
|
m[ErrDataNoFound] = "无数据查询"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func GeneralMsgUserInit(m msg) {
|
||||||
|
m[ErrServerDown] = "后端服务未启动,此功能暂时无法使用。"
|
||||||
|
}
|
||||||
|
@ -12,5 +12,5 @@ func NlpMsgInit(m msg) {
|
|||||||
|
|
||||||
func NlpMsgUserInit(m msg) {
|
func NlpMsgUserInit(m msg) {
|
||||||
m[ErrNoContent] = "请输入内容"
|
m[ErrNoContent] = "请输入内容"
|
||||||
m[ErrNoDocFound] = "没有找到相关文档"
|
m[ErrNoDocFound] = "小护没有在知识库中找到相关文档。😿😿😿"
|
||||||
}
|
}
|
||||||
|
@ -6,6 +6,7 @@ import (
|
|||||||
"catface/app/model_es"
|
"catface/app/model_es"
|
||||||
"catface/app/service/nlp"
|
"catface/app/service/nlp"
|
||||||
"catface/app/utils/response"
|
"catface/app/utils/response"
|
||||||
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
@ -117,8 +118,11 @@ func (r *Rag) ChatWebSocket(context *gin.Context) {
|
|||||||
// 1. query embedding
|
// 1. query embedding
|
||||||
embedding, ok := nlp.GetEmbedding(query)
|
embedding, ok := nlp.GetEmbedding(query)
|
||||||
if !ok {
|
if !ok {
|
||||||
code := errcode.ErrPythonService
|
code := errcode.ErrServerDown
|
||||||
response.Fail(context, code, errcode.ErrMsg[code], "")
|
err := ws.WriteMessage(websocket.TextMessage, []byte(fmt.Sprintf("%d:%s", code, errcode.ErrMsgForUser[code])))
|
||||||
|
if err != nil {
|
||||||
|
variable.ZapLog.Error("Failed to send error message via WebSocket", zap.Error(err))
|
||||||
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -128,7 +132,11 @@ func (r *Rag) ChatWebSocket(context *gin.Context) {
|
|||||||
variable.ZapLog.Error("ES TopK error", zap.Error(err))
|
variable.ZapLog.Error("ES TopK error", zap.Error(err))
|
||||||
|
|
||||||
code := errcode.ErrNoDocFound
|
code := errcode.ErrNoDocFound
|
||||||
response.Fail(context, code, errcode.ErrMsg[code], errcode.ErrMsgForUser[code])
|
err := ws.WriteMessage(websocket.TextMessage, []byte(fmt.Sprintf("%d:%s", code, errcode.ErrMsgForUser[code])))
|
||||||
|
if err != nil {
|
||||||
|
variable.ZapLog.Error("Failed to send error message via WebSocket", zap.Error(err))
|
||||||
|
}
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 3.
|
// 3.
|
||||||
|
Loading…
x
Reference in New Issue
Block a user