mlogclub
89 lines
3.0 KiB
Go
89 lines
3.0 KiB
Go
package runtime
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"cs-agent/internal/models"
|
|
"cs-agent/internal/pkg/dto/request"
|
|
"cs-agent/internal/pkg/dto/response"
|
|
"cs-agent/internal/pkg/enums"
|
|
"cs-agent/internal/pkg/errorsx"
|
|
svc "cs-agent/internal/services"
|
|
)
|
|
|
|
func init() {
|
|
svc.SkillDebugRunHook = DebugRunSkill
|
|
}
|
|
|
|
func DebugRunSkill(ctx context.Context, req request.SkillDebugRunRequest) (*response.SkillDebugRunResponse, error) {
|
|
aiAgent := svc.AIAgentService.Get(req.AIAgentID)
|
|
if aiAgent == nil || aiAgent.Status != enums.StatusOk {
|
|
return nil, errorsx.InvalidParam("AI Agent不存在或未启用")
|
|
}
|
|
aiConfig := svc.AIConfigService.Get(aiAgent.AIConfigID)
|
|
if aiConfig == nil {
|
|
return nil, errorsx.InvalidParam("AI Agent关联的AI配置不存在")
|
|
}
|
|
conversation := &models.Conversation{ID: req.ConversationID, AIAgentID: req.AIAgentID}
|
|
if req.ConversationID > 0 {
|
|
conversation = svc.ConversationService.Get(req.ConversationID)
|
|
if conversation == nil {
|
|
return nil, errorsx.InvalidParam("会话不存在")
|
|
}
|
|
}
|
|
message := &models.Message{
|
|
ConversationID: req.ConversationID,
|
|
SenderType: enums.IMSenderTypeCustomer,
|
|
MessageType: enums.IMMessageTypeText,
|
|
Content: strings.TrimSpace(req.UserMessage),
|
|
}
|
|
summary, err := Service.Run(ctx, Request{
|
|
Conversation: conversation,
|
|
UserMessage: message,
|
|
AIAgent: aiAgent,
|
|
AIConfig: aiConfig,
|
|
ManualSkillCode: strings.TrimSpace(req.SkillCode),
|
|
})
|
|
if err != nil {
|
|
return buildSkillDebugRunResponse(req, summary, nil), err
|
|
}
|
|
selectedSkill := svc.SkillDefinitionService.GetByCode(strings.TrimSpace(req.SkillCode))
|
|
if summary == nil || strings.TrimSpace(summary.PlannedSkillCode) == "" {
|
|
return nil, errorsx.InvalidParam("Skill 未命中")
|
|
}
|
|
return buildSkillDebugRunResponse(req, summary, selectedSkill), nil
|
|
}
|
|
|
|
func buildSkillDebugRunResponse(req request.SkillDebugRunRequest, summary *Summary, skill *models.SkillDefinition) *response.SkillDebugRunResponse {
|
|
resp := &response.SkillDebugRunResponse{
|
|
ConversationID: req.ConversationID,
|
|
AIAgentID: req.AIAgentID,
|
|
}
|
|
if skill != nil {
|
|
resp.SkillCode = skill.Code
|
|
resp.SkillName = skill.Name
|
|
}
|
|
if summary == nil {
|
|
return resp
|
|
}
|
|
if resp.SkillCode == "" {
|
|
resp.SkillCode = strings.TrimSpace(summary.PlannedSkillCode)
|
|
}
|
|
resp.ReplyText = summary.ReplyText
|
|
resp.PlanReason = summary.PlanReason
|
|
resp.SkillRouteTrace = summary.SkillRouteTrace
|
|
resp.SkillAllowedToolCodes = append([]string(nil), summary.SkillAllowedToolCodes...)
|
|
resp.ToolCodes = append([]string(nil), summary.ToolCodes...)
|
|
resp.InvokedToolCodes = append([]string(nil), summary.InvokedToolCodes...)
|
|
resp.ToolSearchTrace = extractToolSearchTrace(summary)
|
|
resp.GraphToolTrace = extractGraphToolTrace(summary)
|
|
resp.GraphToolCode = firstGraphToolCode(summary)
|
|
resp.InterruptType = firstInterruptType(summary)
|
|
resp.CheckPointID = summary.CheckPointID
|
|
resp.Interrupted = summary.Interrupted
|
|
resp.TraceData = summary.TraceData
|
|
resp.ErrorMessage = summary.ErrorMessage
|
|
return resp
|
|
}
|