diff --git a/server/chat_handler.go b/server/chat_handler.go index 3d0bb061..9699176a 100644 --- a/server/chat_handler.go +++ b/server/chat_handler.go @@ -176,6 +176,8 @@ func (s *Server) sendMessage(session types.ChatSession, role types.ChatRole, pro if err != nil { // 数据解析出错 logger.Error(err, line) replyChunkMessage(ws, types.WsMessage{Type: types.WsEnd, IsHelloMsg: false}) + replyMessage(ws, ErrorMsg, false) + replyMessage(ws, "![](images/wx.png)", true) break } // 初始化 role diff --git a/server/server.go b/server/server.go index 88a773ee..3c288cc8 100644 --- a/server/server.go +++ b/server/server.go @@ -6,6 +6,7 @@ import ( "github.com/gin-contrib/sessions" "github.com/gin-contrib/sessions/cookie" "github.com/gin-gonic/gin" + "io" "io/fs" "net/http" logger2 "openai/logger" @@ -71,6 +72,7 @@ func NewServer(configPath string) (*Server, error) { func (s *Server) Run(webRoot embed.FS, path string, debug bool) { s.DebugMode = debug gin.SetMode(gin.ReleaseMode) + gin.DefaultWriter = io.Discard engine := gin.Default() if debug { engine.Use(corsMiddleware()) diff --git a/web/src/views/Chat.vue b/web/src/views/Chat.vue index 2d35f862..f5fd5939 100644 --- a/web/src/views/Chat.vue +++ b/web/src/views/Chat.vue @@ -134,7 +134,7 @@ export default defineComponent({ toolBoxHeight: 61 + 42, // 工具框的高度 inputBoxWidth: window.innerWidth - 20, sending: false, - loading: false + loading: true } },