feat: add AI workflow editor

2026-06-22 00:26:16 +08:00
parent df7b199c18
commit 14f50e064f
19 changed files with 1588 additions and 1 deletions
@@ -0,0 +1,39 @@
+package compiler
+
+import (
+	"testing"
+
+	"agent-desk/internal/ai/workflow/dsl"
+	workflowregistry "agent-desk/internal/ai/workflow/registry"
+	"agent-desk/internal/pkg/toolx"
+)
+
+func TestCompileMapsWorkflowNodesToGraphTools(t *testing.T) {
+	result := Compile(dsl.Definition{
+		EntryNodeID: "start",
+		Nodes: []dsl.Node{
+			{ID: "start", Type: workflowregistry.NodeTypeStart, Name: "Start"},
+			{ID: "analyze", Type: workflowregistry.NodeTypeAnalyzeConversation, Name: "Analyze"},
+			{ID: "draft", Type: workflowregistry.NodeTypePrepareTicketDraft, Name: "Draft"},
+			{ID: "create", Type: workflowregistry.NodeTypeCreateTicket, Name: "Create"},
+			{ID: "handoff", Type: workflowregistry.NodeTypeHandoffToHuman, Name: "Handoff"},
+		},
+	})
+	want := []string{
+		toolx.GraphAnalyzeConversation.Code,
+		toolx.GraphPrepareTicketDraft.Code,
+		toolx.GraphCreateTicketConfirm.Code,
+		toolx.GraphHandoffConversation.Code,
+	}
+	if len(result.ToolCodes) != len(want) {
+		t.Fatalf("expected %d tool codes, got %d: %#v", len(want), len(result.ToolCodes), result.ToolCodes)
+	}
+	for i, item := range want {
+		if result.ToolCodes[i] != item {
+			t.Fatalf("tool code[%d] = %s, want %s", i, result.ToolCodes[i], item)
+		}
+	}
+	if result.Appendix == "" {
+		t.Fatalf("expected workflow appendix")
+	}
+}