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,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...)
+}