Havoc420/catface_backend_go
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐱

Browse Source
This commit is contained in:
Havoc412 2024-11-18 00:39:36 +08:00
parent d2f6f5da5a
commit c06bc1f84e
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/global/errcode/nlp.go
View File

@ -3,6 +3,7 @@ package errcode
const ( const (
ErrNoContent = ErrNlp + iota ErrNoContent = ErrNlp + iota
ErrNoDocFound ErrNoDocFound
ErrPythonServierDown
) )
func NlpMsgInit(m msg) { func NlpMsgInit(m msg) {
@ -12,5 +13,6 @@ func NlpMsgInit(m msg) {
func NlpMsgUserInit(m msg) { func NlpMsgUserInit(m msg) {
m[ErrNoContent] = "请输入内容" m[ErrNoContent] = "请输入内容"
m[ErrNoDocFound] = "小护没有在知识库中找到相关文档。😿😿😿" m[ErrNoDocFound] = "小护没有在知识库中找到相关文档。😿"
m[ErrPythonServierDown] = "小护的🐍python服务挂了，请稍后再试。😿"
} }

2
app/http/controller/web/rag_controller.go
View File

@ -117,7 +117,7 @@ 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.ErrServerDown code := errcode.ErrPythonServierDown
err := ws.WriteMessage(websocket.TextMessage, []byte(errcode.ErrMsgForUser[code])) err := ws.WriteMessage(websocket.TextMessage, []byte(errcode.ErrMsgForUser[code]))
if err != nil { if err != nil {
variable.ZapLog.Error("Failed to send error message via WebSocket", zap.Error(err)) variable.ZapLog.Error("Failed to send error message via WebSocket", zap.Error(err))