feat: add AI workflow editor
This commit is contained in:
@@ -0,0 +1,60 @@
|
||||
package compiler
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"agent-desk/internal/ai/workflow/dsl"
|
||||
workflowregistry "agent-desk/internal/ai/workflow/registry"
|
||||
"agent-desk/internal/pkg/toolx"
|
||||
)
|
||||
|
||||
type Result struct {
|
||||
ToolCodes []string
|
||||
Appendix string
|
||||
}
|
||||
|
||||
func Compile(def dsl.Definition) Result {
|
||||
toolCodes := make([]string, 0)
|
||||
lines := make([]string, 0, len(def.Nodes)+2)
|
||||
if strings.TrimSpace(def.EntryNodeID) != "" {
|
||||
lines = append(lines, fmt.Sprintf("Workflow entry node: %s.", strings.TrimSpace(def.EntryNodeID)))
|
||||
}
|
||||
for _, node := range def.Nodes {
|
||||
nodeType := strings.TrimSpace(node.Type)
|
||||
if code := graphToolCodeForNodeType(nodeType); code != "" {
|
||||
toolCodes = append(toolCodes, code)
|
||||
}
|
||||
nodeName := strings.TrimSpace(node.Name)
|
||||
if nodeName == "" {
|
||||
nodeName = strings.TrimSpace(node.ID)
|
||||
}
|
||||
if nodeName == "" {
|
||||
continue
|
||||
}
|
||||
lines = append(lines, fmt.Sprintf("- %s: %s", nodeName, nodeType))
|
||||
}
|
||||
appendix := ""
|
||||
if len(lines) > 0 {
|
||||
appendix = "Published customer-service workflow:\n" + strings.Join(lines, "\n")
|
||||
}
|
||||
return Result{
|
||||
ToolCodes: toolx.NormalizeToolCodes(toolCodes),
|
||||
Appendix: appendix,
|
||||
}
|
||||
}
|
||||
|
||||
func graphToolCodeForNodeType(nodeType string) string {
|
||||
switch strings.TrimSpace(nodeType) {
|
||||
case workflowregistry.NodeTypeAnalyzeConversation:
|
||||
return toolx.GraphAnalyzeConversation.Code
|
||||
case workflowregistry.NodeTypePrepareTicketDraft:
|
||||
return toolx.GraphPrepareTicketDraft.Code
|
||||
case workflowregistry.NodeTypeCreateTicket:
|
||||
return toolx.GraphCreateTicketConfirm.Code
|
||||
case workflowregistry.NodeTypeHandoffToHuman:
|
||||
return toolx.GraphHandoffConversation.Code
|
||||
default:
|
||||
return ""
|
||||
}
|
||||
}
|
||||
@@ -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")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user