refactor(web): 优化 WebSocket 错误消息发送

- 移除了未使用的 fmt 包
- 更新了 Rag 结构的注释
- 简化了错误消息的发送逻辑,直接使用 errcode.ErrMsgForUser[code] 作为消息内容
This commit is contained in:
Havoc412 2024-11-18 00:27:33 +08:00
parent dba6b16693
commit d2f6f5da5a

View File

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