CalvinBackup/CyberStrikeAI
1
0
Fork 0
mirror of https://github.com/Ed1s0nZ/CyberStrikeAI.git synced 2026-08-05 18:38:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
公明
2026-07-22 21:07:21 +08:00
committed by GitHub
parent 577c97aab0
commit 4333cf1dd3
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/handler/multi_agent_prepare.go
+4
View File
@@ -56,6 +56,7 @@ func (h *AgentHandler) prepareMultiAgentSession(req *ChatRequest, c *gin.Context
var err error var err error
meta := audit.ConversationCreateMetaFromGin(c, source) meta := audit.ConversationCreateMetaFromGin(c, source)
meta.ProjectID = projectID meta.ProjectID = projectID
meta.RoleName = req.Role
if webshellID != "" { if webshellID != "" {
meta.Source = source + "_webshell" meta.Source = source + "_webshell"
meta.WebShellConnectionID = webshellID meta.WebShellConnectionID = webshellID
@@ -80,6 +81,9 @@ func (h *AgentHandler) prepareMultiAgentSession(req *ChatRequest, c *gin.Context
return nil, fmt.Errorf("无权访问该对话") 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) agentHistoryMessages, err := h.loadHistoryFromAgentTrace(conversationID)
if err != nil { if err != nil {