From 6b9b4e992f17edf4d12a64bc50c323e4ecadf704 Mon Sep 17 00:00:00 2001
From: renzhiyuan <465386466@qq.com>
Date: Mon, 13 Oct 2025 15:17:00 +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                | 5 +++--
 internal/biz/session.go               | 1 +
 internal/data/model/ai_session.gen.go | 3 ++-
 internal/entitys/session.go           | 5 +++--
 4 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/internal/biz/router.go b/internal/biz/router.go
index d26933c..986b63b 100644
--- a/internal/biz/router.go
+++ b/internal/biz/router.go
@@ -73,7 +73,7 @@ func (r *AiRouterBiz) RouteWithSocket(c *websocket.Conn, req *entitys.ChatSockRe
 	requireData.Ch = make(chan entitys.Response)
 	ctx, cancel := context.WithCancel(context.Background())
 	// 启动独立的消息处理协程
-	done := r.startMessageHandler(ctx, c, &requireData)
+	done := r.startMessageHandler(ctx, c, &requireData, req.Text)
 	defer func() {
 		close(requireData.Ch) //关闭主通道
 		<-done                // 等待消息处理完成
@@ -113,6 +113,7 @@ func (r *AiRouterBiz) startMessageHandler(
 	ctx context.Context,
 	c *websocket.Conn,
 	requireData *entitys.RequireData,
+	userInput string,
 ) <-chan struct{} {
 	done := make(chan struct{})
 	var chat []string
@@ -125,7 +126,7 @@ func (r *AiRouterBiz) startMessageHandler(
 				{
 					SessionID: requireData.Session,
 					Role:      "user",
-					Content:   "", // 用户输入在外部处理
+					Content:   userInput, // 用户输入在外部处理
 				},
 			}
 			if len(chat) > 0 {
diff --git a/internal/biz/session.go b/internal/biz/session.go
index 5961c04..e1dd17e 100644
--- a/internal/biz/session.go
+++ b/internal/biz/session.go
@@ -62,6 +62,7 @@ func (s *SessionBiz) SessionInit(ctx context.Context, req *entitys.SessionInitRe
 			SysID:     sysConfig.SysID,
 			SessionID: utils.UUID(),
 			UserID:    req.UserId,
+			UserName:  req.UserName,
 		}
 		err = s.sessionRepo.Create(&session)
 		if err != nil {
diff --git a/internal/data/model/ai_session.gen.go b/internal/data/model/ai_session.gen.go
index 6f236c0..d872adb 100644
--- a/internal/data/model/ai_session.gen.go
+++ b/internal/data/model/ai_session.gen.go
@@ -20,7 +20,8 @@ type AiSession struct {
 	UpdateAt          *time.Time `gorm:"column:update_at;default:CURRENT_TIMESTAMP" json:"update_at"`
 	Status            int32      `gorm:"column:status;not null" json:"status"`
 	DeleteAt          *time.Time `gorm:"column:delete_at" json:"delete_at"`
-	UserID            string     `gorm:"column:user_id;comment:用户id" json:"user_id"` // 用户id
+	UserID            string     `gorm:"column:user_id;comment:用户id" json:"user_id"`     // 用户id
+	UserName          string     `gorm:"column:user_name;comment:用户名称" json:"user_name"` // 用户id
 }
 
 // TableName AiSession's table name
diff --git a/internal/entitys/session.go b/internal/entitys/session.go
index 55e5c54..1b31ac1 100644
--- a/internal/entitys/session.go
+++ b/internal/entitys/session.go
@@ -1,8 +1,9 @@
 package entitys
 
 type SessionInitRequest struct {
-	SysId  string `json:"sys_id"`
-	UserId string `json:"user_id"`
+	SysId    string `json:"sys_id"`
+	UserId   string `json:"user_id"`
+	UserName string `json:"user_name"`
 }
 
 type SessionInitResponse struct {