From 8900ea6fbd021d7f1d79fcce4d8183eca74a91c2 Mon Sep 17 00:00:00 2001
From: renzhiyuan <465386466@qq.com>
Date: Sun, 28 Sep 2025 10:19:45 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BB=93=E6=9E=84=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/biz/router.go            | 9 +++++----
 internal/data/error/error_code.go | 6 ++++++
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/internal/biz/router.go b/internal/biz/router.go
index f52a040..b1d4726 100644
--- a/internal/biz/router.go
+++ b/internal/biz/router.go
@@ -149,17 +149,17 @@ func (r *AiRouterBiz) RouteWithSocket(c *websocket.Conn, req *entitys.ChatSockRe
 
 	sysInfo, err := r.getSysInfo(key)
 	if err != nil {
-		return errors.SysNotFound
+		return errors.SysErr("获取系统信息失败：%v", err.Error())
 	}
 
 	history, err := r.getSessionChatHis(session)
 	if err != nil {
-		return errors.SystemError
+		return errors.SysErr("获取历史记录失败：%v", err.Error())
 	}
 
 	task, err := r.getTasks(sysInfo.SysID)
 	if err != nil {
-		return errors.SystemError
+		return errors.SysErr("获取任务列表失败：%v", err.Error())
 	}
 
 	prompt := r.getPromptLLM(sysInfo, history, req.Text, task)
@@ -197,7 +197,8 @@ func (r *AiRouterBiz) RouteWithSocket(c *websocket.Conn, req *entitys.ChatSockRe
 	}
 	var matchJson entitys.Match
 	if err := json.Unmarshal([]byte(resMsg), &matchJson); err != nil {
-		return errors.SystemError
+		return errors.SysErr("数据结构错误：%v", err.Error())
+
 	}
 
 	if err := r.handleMatch(ctx, c, ch, &matchJson, task, sysInfo); err != nil {
diff --git a/internal/data/error/error_code.go b/internal/data/error/error_code.go
index a1e1f4a..e6c7230 100644
--- a/internal/data/error/error_code.go
+++ b/internal/data/error/error_code.go
@@ -1,5 +1,7 @@
 package errorcode
 
+import "fmt"
+
 var (
 	Success       = &BusinessErr{code: 200, message: "成功"}
 	ParamError    = &BusinessErr{code: 401, message: "参数错误"}
@@ -40,6 +42,10 @@ func NewBusinessErr(code int, message string) *BusinessErr {
 	return &BusinessErr{code: code, message: message}
 }
 
+func SysErr(message string, arg ...any) *BusinessErr {
+	return &BusinessErr{code: SystemError.code, message: fmt.Sprintf(message, arg)}
+}
+
 func (e *BusinessErr) Wrap(err error) *BusinessErr {
 	return NewBusinessErr(e.code, err.Error())
 }