ai-agent/internal/services/agent_run_log_service.go

package services

import (
	"cs-agent/internal/models"
	"cs-agent/internal/pkg/toolx"
	"cs-agent/internal/repositories"
	"strings"

	"github.com/mlogclub/simple/sqls"
	"github.com/mlogclub/simple/web/params"
)

var AgentRunLogService = newAgentRunLogService()

func newAgentRunLogService() *agentRunLogService {
	return &agentRunLogService{}
}

type agentRunLogService struct{}

type AgentRunGraphSummary struct {
	TriageCount              int64
	TriagePrepareTicket      int64
	TriagePrepareTicketReady int64
	TriageHandoff            int64
	TriageContinueAnswering  int64
	AnalyzeCount             int64
	PrepareDraftCount        int64
	CreateTicketCount        int64
	HandoffCount             int64
}

func (s *agentRunLogService) Get(id int64) *models.AgentRunLog {
	return repositories.AgentRunLogRepository.Get(sqls.DB(), id)
}

func (s *agentRunLogService) Take(where ...interface{}) *models.AgentRunLog {
	return repositories.AgentRunLogRepository.Take(sqls.DB(), where...)
}

func (s *agentRunLogService) Find(cnd *sqls.Cnd) []models.AgentRunLog {
	return repositories.AgentRunLogRepository.Find(sqls.DB(), cnd)
}

func (s *agentRunLogService) FindOne(cnd *sqls.Cnd) *models.AgentRunLog {
	return repositories.AgentRunLogRepository.FindOne(sqls.DB(), cnd)
}

func (s *agentRunLogService) FindPageByParams(params *params.QueryParams) (list []models.AgentRunLog, paging *sqls.Paging) {
	return repositories.AgentRunLogRepository.FindPageByParams(sqls.DB(), params)
}

func (s *agentRunLogService) FindPageByCnd(cnd *sqls.Cnd) (list []models.AgentRunLog, paging *sqls.Paging) {
	return repositories.AgentRunLogRepository.FindPageByCnd(sqls.DB(), cnd)
}

func (s *agentRunLogService) Count(cnd *sqls.Cnd) int64 {
	return repositories.AgentRunLogRepository.Count(sqls.DB(), cnd)
}

func (s *agentRunLogService) Create(t *models.AgentRunLog) error {
	return repositories.AgentRunLogRepository.Create(sqls.DB(), t)
}

func (s *agentRunLogService) Update(t *models.AgentRunLog) error {
	return repositories.AgentRunLogRepository.Update(sqls.DB(), t)
}

func (s *agentRunLogService) Updates(id int64, columns map[string]interface{}) error {
	return repositories.AgentRunLogRepository.Updates(sqls.DB(), id, columns)
}

func (s *agentRunLogService) UpdateColumn(id int64, name string, value interface{}) error {
	return repositories.AgentRunLogRepository.UpdateColumn(sqls.DB(), id, name, value)
}

func (s *agentRunLogService) Delete(id int64) {
	repositories.AgentRunLogRepository.Delete(sqls.DB(), id)
}

func (s *agentRunLogService) ApplyHITLStatusFilter(cnd *sqls.Cnd, hitlStatus string) *sqls.Cnd {
	if cnd == nil {
		cnd = sqls.NewCnd()
	}
	switch strings.TrimSpace(hitlStatus) {
	case "pending":
		cnd.Eq("final_status", "interrupted")
	case "expired":
		cnd.Eq("final_status", "expired")
	case "cancelled":
		cnd.Where("(reply_text LIKE ? OR reply_text LIKE ?)", "%已取消本次工单创建%", "%已取消本次转人工%")
	case "confirmed":
		cnd.Where("resume_source <> ''")
	case "triggered":
		cnd.Where("interrupt_type <> ''")
	}
	return cnd
}

func (s *agentRunLogService) BuildGraphSummary(aiAgentID int64) *AgentRunGraphSummary {
	buildBaseCnd := func() *sqls.Cnd {
		cnd := sqls.NewCnd()
		if aiAgentID > 0 {
			cnd.Eq("ai_agent_id", aiAgentID)
		}
		return cnd
	}
	return &AgentRunGraphSummary{
		TriageCount:              s.countByGraphTool(buildBaseCnd(), toolx.GraphTriageServiceRequest.Code),
		TriagePrepareTicket:      s.countByGraphToolAndPlanReason(buildBaseCnd(), toolx.GraphTriageServiceRequest.Code, "%prepare_ticket%"),
		TriagePrepareTicketReady: s.countByGraphToolAndPlanReason(buildBaseCnd(), toolx.GraphTriageServiceRequest.Code, "%prepare_ticket with ready ticket draft%"),
		TriageHandoff:            s.countByGraphToolAndPlanReason(buildBaseCnd(), toolx.GraphTriageServiceRequest.Code, "%handoff_to_human%"),
		TriageContinueAnswering:  s.countByGraphToolAndPlanReason(buildBaseCnd(), toolx.GraphTriageServiceRequest.Code, "%continue_answering%"),
		AnalyzeCount:             s.countByGraphTool(buildBaseCnd(), toolx.GraphAnalyzeConversation.Code),
		PrepareDraftCount:        s.countByGraphTool(buildBaseCnd(), toolx.GraphPrepareTicketDraft.Code),
		CreateTicketCount:        s.countByGraphTool(buildBaseCnd(), toolx.GraphCreateTicketConfirm.Code),
		HandoffCount:             s.countByGraphTool(buildBaseCnd(), toolx.GraphHandoffConversation.Code),
	}
}

func (s *agentRunLogService) countByGraphTool(baseCnd *sqls.Cnd, graphToolCode string) int64 {
	cnd := baseCnd
	cnd.Eq("graph_tool_code", strings.TrimSpace(graphToolCode))
	return s.Count(cnd)
}

func (s *agentRunLogService) countByGraphToolAndPlanReason(baseCnd *sqls.Cnd, graphToolCode string, planReasonLike string) int64 {
	cnd := baseCnd
	cnd.Eq("graph_tool_code", strings.TrimSpace(graphToolCode))
	cnd.Where("plan_reason LIKE ?", strings.TrimSpace(planReasonLike))
	return s.Count(cnd)
}