133 lines
4.8 KiB
Go
133 lines
4.8 KiB
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)
|
|
}
|