diff --git a/app/global/errcode/nlp.go b/app/global/errcode/nlp.go index 98f47ca..368a3c1 100644 --- a/app/global/errcode/nlp.go +++ b/app/global/errcode/nlp.go @@ -3,6 +3,7 @@ package errcode const ( ErrNoContent = ErrNlp + iota ErrNoDocFound + ErrPythonServierDown ) func NlpMsgInit(m msg) { @@ -12,5 +13,6 @@ func NlpMsgInit(m msg) { func NlpMsgUserInit(m msg) { m[ErrNoContent] = "请输入内容" - m[ErrNoDocFound] = "小护没有在知识库中找到相关文档。😿😿😿" + m[ErrNoDocFound] = "小护没有在知识库中找到相关文档。😿" + m[ErrPythonServierDown] = "小护的🐍python服务挂了,请稍后再试。😿" } diff --git a/app/http/controller/web/rag_controller.go b/app/http/controller/web/rag_controller.go index 7c98fd4..ef291c0 100644 --- a/app/http/controller/web/rag_controller.go +++ b/app/http/controller/web/rag_controller.go @@ -117,7 +117,7 @@ func (r *Rag) ChatWebSocket(context *gin.Context) { // 1. query embedding embedding, ok := nlp.GetEmbedding(query) if !ok { - code := errcode.ErrServerDown + code := errcode.ErrPythonServierDown err := ws.WriteMessage(websocket.TextMessage, []byte(errcode.ErrMsgForUser[code])) if err != nil { variable.ZapLog.Error("Failed to send error message via WebSocket", zap.Error(err))