chore: import clean Agent Desk snapshot (2/7)

internal/ai/workflow/registry/spec.go

@@ -0,0 +1,134 @@
+package registry
+
+import "agent-desk/internal/ai/workflow/dsl"
+
+type NodeRiskLevel string
+
+const (
+	NodeRiskLevelLow    NodeRiskLevel = "low"
+	NodeRiskLevelMedium NodeRiskLevel = "medium"
+	NodeRiskLevelHigh   NodeRiskLevel = "high"
+)
+
+type VariableType string
+
+const (
+	VariableTypeString       VariableType = "string"
+	VariableTypeNumber       VariableType = "number"
+	VariableTypeInteger      VariableType = "integer"
+	VariableTypeBoolean      VariableType = "boolean"
+	VariableTypeObject       VariableType = "object"
+	VariableTypeStringArray  VariableType = "array<string>"
+	VariableTypeIntegerArray VariableType = "array<int>"
+	VariableTypeObjectArray  VariableType = "array<object>"
+	VariableTypeAny          VariableType = "any"
+)
+
+type VariableSpec struct {
+	Name         string                `json:"name"`
+	Label        string                `json:"label,omitempty"`
+	Type         VariableType          `json:"type"`
+	Required     bool                  `json:"required,omitempty"`
+	Description  string                `json:"description"`
+	Operators    []string              `json:"operators,omitempty"`
+	ValueOptions []VariableValueOption `json:"valueOptions,omitempty"`
+}
+
+type VariableValueOption struct {
+	Value       any    `json:"value"`
+	Label       string `json:"label"`
+	Description string `json:"description,omitempty"`
+}
+
+type NodeSpec struct {
+	Type                            string               `json:"type"`
+	Title                           string               `json:"title"`
+	Description                     string               `json:"description"`
+	Icon                            string               `json:"icon"`
+	Category                        string               `json:"category"`
+	Executable                      bool                 `json:"executable"`
+	RiskLevel                       NodeRiskLevel        `json:"riskLevel"`
+	Interruptible                   bool                 `json:"interruptible"`
+	RequiresConfirmationPredecessor bool                 `json:"requiresConfirmationPredecessor"`
+	ConfigSchema                    any                  `json:"configSchema,omitempty"`
+	InputSchema                     []VariableSpec       `json:"inputSchema,omitempty"`
+	OutputSchema                    []VariableSpec       `json:"outputSchema,omitempty"`
+	DefaultInputs                   map[string]dsl.Value `json:"defaultInputs,omitempty"`
+}
+
+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
+		}
+		spec.Executable = IsExecutableNodeType(spec.Type)
+		if spec.Category == "" {
+			spec.Category = NodeCategory(spec.Type)
+		}
+		ret.specsByType[spec.Type] = spec
+		ret.specs = append(ret.specs, spec)
+	}
+	return ret
+}
+
+func IsExecutableNodeType(nodeType string) bool {
+	switch nodeType {
+	case NodeTypeStart,
+		NodeTypeConversationUnderstanding,
+		NodeTypeReplyPolicy,
+		NodeTypeKnowledgeRetrieve,
+		NodeTypeAnswerabilityGate,
+		NodeTypeCondition,
+		NodeTypeAnalyzeConversation,
+		NodeTypePrepareTicketDraft,
+		NodeTypeHumanConfirm,
+		NodeTypeCreateTicket,
+		NodeTypeLLMReply,
+		NodeTypeLLM,
+		NodeTypeSendReply,
+		NodeTypeHandoffToHuman,
+		NodeTypeEnd:
+		return true
+	default:
+		return false
+	}
+}
+
+func NodeCategory(nodeType string) string {
+	switch nodeType {
+	case NodeTypeStart, NodeTypeEnd:
+		return "trigger"
+	case NodeTypeCondition, NodeTypeMultiCondition, NodeTypeLoop, NodeTypeBlockStart, NodeTypeBlockEnd, NodeTypeContinue, NodeTypeBreak:
+		return "control"
+	case NodeTypeConversationUnderstanding, NodeTypeReplyPolicy, NodeTypeAnswerabilityGate, NodeTypeAnalyzeConversation, NodeTypeLLMReply, NodeTypeLLM, NodeTypeKnowledgeRetrieve:
+		return "ai"
+	case NodeTypePrepareTicketDraft, NodeTypeHumanConfirm, NodeTypeCreateTicket, NodeTypeHandoffToHuman, NodeTypeSendReply:
+		return "business"
+	default:
+		return "utility"
+	}
+}
+
+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...)
+}