From d2f6f5da5a330ca2e9efce6096dcfd3ec18ebe7c Mon Sep 17 00:00:00 2001 From: Havoc412 <2993167370@qq.com> Date: Mon, 18 Nov 2024 00:27:33 +0800 Subject: [PATCH] =?UTF-8?q?refactor(web):=20=E4=BC=98=E5=8C=96=20WebSocket?= =?UTF-8?q?=20=E9=94=99=E8=AF=AF=E6=B6=88=E6=81=AF=E5=8F=91=E9=80=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 移除了未使用的 fmt 包 - 更新了 Rag 结构的注释 - 简化了错误消息的发送逻辑，直接使用 errcode.ErrMsgForUser[code] 作为消息内容 --- app/http/controller/web/rag_controller.go | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/app/http/controller/web/rag_controller.go b/app/http/controller/web/rag_controller.go index 06fc83d..7c98fd4 100644 --- a/app/http/controller/web/rag_controller.go +++ b/app/http/controller/web/rag_controller.go @@ -6,7 +6,6 @@ import ( "catface/app/model_es" "catface/app/service/nlp" "catface/app/utils/response" - "fmt" "io" "net/http" @@ -18,7 +17,7 @@ import ( type Rag struct { } -// v1 Http-POST 版本 +// v1 Http-POST 版本; chat 需要不使用 ch 的版本。 // func (r *Rag) Chat(context *gin.Context) { // // 1. query embedding // query := context.GetString(consts.ValidatorPrefix + "query") @@ -119,7 +118,7 @@ func (r *Rag) ChatWebSocket(context *gin.Context) { embedding, ok := nlp.GetEmbedding(query) if !ok { code := errcode.ErrServerDown - err := ws.WriteMessage(websocket.TextMessage, []byte(fmt.Sprintf("%d:%s", code, errcode.ErrMsgForUser[code]))) + 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)) } @@ -132,7 +131,7 @@ func (r *Rag) ChatWebSocket(context *gin.Context) { variable.ZapLog.Error("ES TopK error", zap.Error(err)) code := errcode.ErrNoDocFound - err := ws.WriteMessage(websocket.TextMessage, []byte(fmt.Sprintf("%d:%s", code, errcode.ErrMsgForUser[code]))) + 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)) }