From d515e88d9741d209aa954cfb5b5f81afb2a3cbb3 Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Wed, 29 Mar 2023 21:44:44 +0800
Subject: [PATCH] =?UTF-8?q?=E7=A6=81=E7=94=A8=20gin=20=E6=89=93=E5=8D=B0?=
 =?UTF-8?q?=E6=97=A5=E5=BF=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/chat_handler.go | 2 ++
 server/server.go       | 2 ++
 web/src/views/Chat.vue | 2 +-
 3 files changed, 5 insertions(+), 1 deletion(-)

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
     }
   },