feat: enhance event consumption logic and add success reply handling

2026-04-14 19:05:05 +08:00
parent 1f894ef7f0
commit 3cdd31786a
6 changed files with 103 additions and 2 deletions
@@ -0,0 +1,41 @@
+package executor
+
+import (
+	"testing"
+
+	"cs-agent/internal/pkg/toolx"
+
+	"github.com/cloudwego/eino/adk"
+	"github.com/cloudwego/eino/schema"
+)
+
+func TestConsumeAgentEventsUsesGraphToolTextAsReplyFallback(t *testing.T) {
+	summary := &RunResult{
+		Status:           "started",
+		InvokedToolCodes: make([]string, 0),
+	}
+	events, gen := adk.NewAsyncIteratorPair[*adk.AgentEvent]()
+	gen.Send(&adk.AgentEvent{
+		Output: &adk.AgentOutput{
+			MessageOutput: &adk.MessageVariant{
+				Role:     schema.Tool,
+				ToolName: toolx.GraphHandoffConversation.Name,
+				Message: &schema.Message{
+					Content: "已为你转接人工客服，请稍候。，请稍候。",
+				},
+			},
+		},
+	})
+	gen.Close()
+
+	consumeAgentEvents(events, summary, nil, map[string]string{
+		toolx.GraphHandoffConversation.Name: toolx.GraphHandoffConversation.Code,
+	})
+
+	if summary.ReplyText != "已为你转接人工客服，请稍候。，请稍候。" {
+		t.Fatalf("unexpected reply text: %q", summary.ReplyText)
+	}
+	if summary.Status != "completed" {
+		t.Fatalf("unexpected summary status: %q", summary.Status)
+	}
+}