mlogclub
135 lines
4.2 KiB
Go
135 lines
4.2 KiB
Go
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...)
|
|
}
|