ai-agent/internal/ai/workflow/registry/spec.go

package registry

import "code.tczkiot.com/wlw/ai-agent/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...)
}