Implement skill debugging functionality and refactor skill execution flow

2026-04-09 11:58:04 +08:00
parent eb9cd78646
commit 28d18ab6c0
12 changed files with 321 additions and 233 deletions
@@ -0,0 +1,83 @@
+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.ToolCodes = append([]string(nil), summary.ToolCodes...)
+	resp.InvokedToolCodes = append([]string(nil), summary.InvokedToolCodes...)
+	resp.CheckPointID = summary.CheckPointID
+	resp.Interrupted = summary.Interrupted
+	resp.TraceData = summary.TraceData
+	resp.ErrorMessage = summary.ErrorMessage
+	return resp
+}