From 4333cf1dd3ac8ac8d650d0362e0fe1109bcef8a4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=85=AC=E6=98=8E?=
 <83812544+Ed1s0nZ@users.noreply.github.com>
Date: Wed, 22 Jul 2026 21:07:21 +0800
Subject: [PATCH] Add files via upload

---
 internal/handler/multi_agent_prepare.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/internal/handler/multi_agent_prepare.go b/internal/handler/multi_agent_prepare.go
index 4becd55c..660d5f04 100644
--- a/internal/handler/multi_agent_prepare.go
+++ b/internal/handler/multi_agent_prepare.go
@@ -56,6 +56,7 @@ func (h *AgentHandler) prepareMultiAgentSession(req *ChatRequest, c *gin.Context
 		var err error
 		meta := audit.ConversationCreateMetaFromGin(c, source)
 		meta.ProjectID = projectID
+		meta.RoleName = req.Role
 		if webshellID != "" {
 			meta.Source = source + "_webshell"
 			meta.WebShellConnectionID = webshellID
@@ -80,6 +81,9 @@ func (h *AgentHandler) prepareMultiAgentSession(req *ChatRequest, c *gin.Context
 			return nil, fmt.Errorf("无权访问该对话")
 		}
 	}
+	if err := h.db.SetConversationRoleName(conversationID, req.Role); err != nil {
+		h.logger.Warn("更新对话角色失败", zap.String("conversationId", conversationID), zap.String("role", req.Role), zap.Error(err))
+	}
 
 	agentHistoryMessages, err := h.loadHistoryFromAgentTrace(conversationID)
 	if err != nil {