feat: add AI workflow DSL validator

2026-06-22 00:08:31 +08:00
parent a8a622dd68
commit fa99c5bed9
5 changed files with 456 additions and 0 deletions
@@ -0,0 +1,33 @@
+package registry
+
+const (
+	NodeTypeStart               = "start"
+	NodeTypeKnowledgeRetrieve   = "knowledge_retrieve"
+	NodeTypeAnswerabilityGate   = "answerability_gate"
+	NodeTypeLLMReply            = "llm_reply"
+	NodeTypeCondition           = "condition"
+	NodeTypeAnalyzeConversation = "analyze_conversation"
+	NodeTypePrepareTicketDraft  = "prepare_ticket_draft"
+	NodeTypeHumanConfirm        = "human_confirm"
+	NodeTypeCreateTicket        = "create_ticket"
+	NodeTypeHandoffToHuman      = "handoff_to_human"
+	NodeTypeSendReply           = "send_reply"
+	NodeTypeEnd                 = "end"
+)
+
+func DefaultRegistry() *Registry {
+	return NewRegistry(
+		NodeSpec{Type: NodeTypeStart, Title: "Start", Description: "Conversation workflow entry.", RiskLevel: NodeRiskLevelLow},
+		NodeSpec{Type: NodeTypeKnowledgeRetrieve, Title: "Knowledge Retrieve", Description: "Retrieve knowledge for the current user message.", RiskLevel: NodeRiskLevelLow},
+		NodeSpec{Type: NodeTypeAnswerabilityGate, Title: "Answerability Gate", Description: "Decide whether retrieved knowledge is enough to answer.", RiskLevel: NodeRiskLevelLow},
+		NodeSpec{Type: NodeTypeLLMReply, Title: "LLM Reply", Description: "Generate a reply or structured analysis with the configured model.", RiskLevel: NodeRiskLevelMedium},
+		NodeSpec{Type: NodeTypeCondition, Title: "Condition", Description: "Route by controlled workflow variables.", RiskLevel: NodeRiskLevelLow},
+		NodeSpec{Type: NodeTypeAnalyzeConversation, Title: "Analyze Conversation", Description: "Analyze intent, risk, and recommended next action.", RiskLevel: NodeRiskLevelLow},
+		NodeSpec{Type: NodeTypePrepareTicketDraft, Title: "Prepare Ticket Draft", Description: "Build a ticket draft from conversation context.", RiskLevel: NodeRiskLevelMedium},
+		NodeSpec{Type: NodeTypeHumanConfirm, Title: "Human Confirm", Description: "Interrupt and wait for explicit user confirmation.", RiskLevel: NodeRiskLevelMedium, Interruptible: true},
+		NodeSpec{Type: NodeTypeCreateTicket, Title: "Create Ticket", Description: "Create a ticket from a confirmed draft.", RiskLevel: NodeRiskLevelHigh, RequiresConfirmationPredecessor: true},
+		NodeSpec{Type: NodeTypeHandoffToHuman, Title: "Handoff To Human", Description: "Transfer the conversation to human support.", RiskLevel: NodeRiskLevelHigh, RequiresConfirmationPredecessor: true},
+		NodeSpec{Type: NodeTypeSendReply, Title: "Send Reply", Description: "Return or commit customer-visible reply text.", RiskLevel: NodeRiskLevelLow},
+		NodeSpec{Type: NodeTypeEnd, Title: "End", Description: "End workflow execution.", RiskLevel: NodeRiskLevelLow},
+	)
+}
@@ -0,0 +1,53 @@
+package registry
+
+type NodeRiskLevel string
+
+const (
+	NodeRiskLevelLow    NodeRiskLevel = "low"
+	NodeRiskLevelMedium NodeRiskLevel = "medium"
+	NodeRiskLevelHigh   NodeRiskLevel = "high"
+)
+
+type NodeSpec struct {
+	Type                            string        `json:"type"`
+	Title                           string        `json:"title"`
+	Description                     string        `json:"description"`
+	RiskLevel                       NodeRiskLevel `json:"riskLevel"`
+	Interruptible                   bool          `json:"interruptible"`
+	RequiresConfirmationPredecessor bool          `json:"requiresConfirmationPredecessor"`
+}
+
+type Registry struct {
+	specsByType map[string]NodeSpec
+	specs       []NodeSpec
+}
+
+func NewRegistry(specs ...NodeSpec) *Registry {
+	ret := &Registry{
+		specsByType: make(map[string]NodeSpec, len(specs)),
+		specs:       make([]NodeSpec, 0, len(specs)),
+	}
+	for _, spec := range specs {
+		if spec.Type == "" {
+			continue
+		}
+		ret.specsByType[spec.Type] = spec
+		ret.specs = append(ret.specs, spec)
+	}
+	return ret
+}
+
+func (r *Registry) Get(nodeType string) (NodeSpec, bool) {
+	if r == nil {
+		return NodeSpec{}, false
+	}
+	spec, ok := r.specsByType[nodeType]
+	return spec, ok
+}
+
+func (r *Registry) List() []NodeSpec {
+	if r == nil {
+		return nil
+	}
+	return append([]NodeSpec(nil), r.specs...)
+}