Refactor AI skill routing and introduce reply handling services

- Moved the skill routing logic from matcher.go to a new router.go file for better organization. - Implemented replyCommitService to handle sending AI replies and managing reply rounds. - Added replyInterruptService to manage conversation interrupts and resume handling. - Created replyRunLogService to log AI reply actions and their outcomes. - Introduced helper functions for building conversation interrupts and resolving prompts. - Added unit tests for the new services and functions to ensure correctness. - Removed unused code and optimized imports in matcher.go.
2026-04-13 17:17:13 +08:00
parent 93cda6ae82
commit 3c0abaaedc
9 changed files with 942 additions and 650 deletions
@@ -0,0 +1,161 @@
+package runtime
+
+import (
+	"testing"
+	"time"
+
+	"cs-agent/internal/models"
+	"cs-agent/internal/pkg/enums"
+	"cs-agent/internal/pkg/toolx"
+)
+
+func TestReplyEligibilityCanReply(t *testing.T) {
+	eligibility := newReplyEligibility()
+	conversation := newConversationFixture()
+	message := newCustomerMessageFixture("hello")
+	aiAgent := newAIAgentFixture()
+
+	if !eligibility.CanReply(conversation, message, aiAgent) {
+		t.Fatalf("expected customer message to be replyable")
+	}
+
+	message.SenderType = enums.IMSenderTypeAgent
+	if eligibility.CanReply(conversation, message, aiAgent) {
+		t.Fatalf("expected non-customer message to be rejected")
+	}
+
+	message = newCustomerMessageFixture("hello")
+	conversation.HandoffAt = ptrTime(time.Now())
+	if eligibility.CanReply(conversation, message, aiAgent) {
+		t.Fatalf("expected handed-off conversation to be rejected")
+	}
+
+	conversation = newConversationFixture()
+	conversation.CurrentAssigneeID = 1
+	if eligibility.CanReply(conversation, message, aiAgent) {
+		t.Fatalf("expected assigned conversation to be rejected")
+	}
+
+	conversation = newConversationFixture()
+	aiAgent.ServiceMode = enums.IMConversationServiceModeHumanOnly
+	if eligibility.CanReply(conversation, message, aiAgent) {
+		t.Fatalf("expected human-only agent to be rejected")
+	}
+
+	aiAgent = newAIAgentFixture()
+	message.Content = "   "
+	if eligibility.CanReply(conversation, message, aiAgent) {
+		t.Fatalf("expected blank message to be rejected")
+	}
+}
+
+func TestResolveReplyTimeout(t *testing.T) {
+	service := newAIReplyService()
+	aiAgent := newAIAgentFixture()
+
+	if got := service.resolveReplyTimeout(aiAgent); got != 180*time.Second {
+		t.Fatalf("expected default timeout, got %v", got)
+	}
+
+	aiAgent.ReplyTimeoutSeconds = 30
+	if got := service.resolveReplyTimeout(aiAgent); got != 30*time.Second {
+		t.Fatalf("expected exact timeout, got %v", got)
+	}
+
+	aiAgent.ReplyTimeoutSeconds = 999
+	if got := service.resolveReplyTimeout(aiAgent); got != 600*time.Second {
+		t.Fatalf("expected clamped timeout, got %v", got)
+	}
+}
+
+func TestBuildRunLogPlan(t *testing.T) {
+	summary := &Summary{
+		PlannedSkillCode: "faq_router",
+		PlanReason:       "manual",
+	}
+	action, toolCode, reason := buildRunLogPlan(summary)
+	if action != "skill" || toolCode != "" || reason != "manual" {
+		t.Fatalf("unexpected skill plan result: action=%q toolCode=%q reason=%q", action, toolCode, reason)
+	}
+
+	summary = &Summary{
+		Interrupted: true,
+		TraceData: `{
+			"graphTools": {
+				"items": [
+					{
+						"toolCode": "` + toolx.GraphTriageServiceRequest.Code + `",
+						"recommendedAction": "create_ticket",
+						"ticketDraftReady": true
+					}
+				]
+			}
+		}`,
+	}
+	action, toolCode, reason = buildRunLogPlan(summary)
+	if action != "graph" || toolCode != toolx.GraphTriageServiceRequest.Code || reason == "" {
+		t.Fatalf("unexpected graph interrupt result: action=%q toolCode=%q reason=%q", action, toolCode, reason)
+	}
+
+	summary = &Summary{
+		InvokedToolCodes: []string{toolx.BuiltinToolSearch.Code},
+		TraceData: `{
+			"toolSearch": {
+				"items": [
+					{
+						"targetToolCode": "mcp/test/search"
+					}
+				]
+			}
+		}`,
+	}
+	action, toolCode, reason = buildRunLogPlan(summary)
+	if action != "tool" || toolCode != "mcp/test/search" || reason != "agent invoked dynamic tool via tool_search" {
+		t.Fatalf("unexpected dynamic tool result: action=%q toolCode=%q reason=%q", action, toolCode, reason)
+	}
+
+	summary = &Summary{ReplyText: "done"}
+	action, toolCode, reason = buildRunLogPlan(summary)
+	if action != "reply" || toolCode != "" || reason != "agent replied directly" {
+		t.Fatalf("unexpected reply result: action=%q toolCode=%q reason=%q", action, toolCode, reason)
+	}
+}
+
+func TestResolveInterruptPrompt(t *testing.T) {
+	summary := &Summary{
+		Interrupts: []InterruptContextSummary{
+			{
+				ID:          "interrupt-1",
+				Type:        "question",
+				InfoPreview: `{"message":"请补充订单号"}`,
+			},
+		},
+	}
+	if got := resolveInterruptPrompt(summary); got != "请补充订单号" {
+		t.Fatalf("unexpected interrupt prompt: %q", got)
+	}
+
+	summary.Interrupts[0].InfoPreview = "直接补充手机号"
+	if got := resolveInterruptPrompt(summary); got != "直接补充手机号" {
+		t.Fatalf("unexpected raw interrupt prompt: %q", got)
+	}
+}
+
+func newConversationFixture() models.Conversation {
+	return models.Conversation{}
+}
+
+func newCustomerMessageFixture(content string) models.Message {
+	return models.Message{
+		SenderType: enums.IMSenderTypeCustomer,
+		Content:    content,
+	}
+}
+
+func newAIAgentFixture() models.AIAgent {
+	return models.AIAgent{}
+}
+
+func ptrTime(v time.Time) *time.Time {
+	return &v
+}