ai-agent/workflow/consts/node/node_template.go

package node

// ======================== 【常量定义：所有中文文案放这里！】 ========================
// 分组名称
const (
	NodeGroupNameComponent = "组件"
	NodeGroupNameBase      = "基础"
	NodeGroupNameCustom    = "自定义"
)

// 节点名称
const (
	NodeNameTextModel      = "生成文案"
	NodeNameImageModel     = "生成图片"
	NodeNameVideoModel     = "生成视频"
	NodeNameSenseOptimize  = "语义优化"
	NodeNameStoryOptimize  = "分镜优化"
	NodeNameScriptOptimize = "剧本优化"
	NodeNameAudioModel     = "音频"
	NodeNameModel          = "模型"
	NodeNameMerge          = "结果合并"
	NodeNameJudge          = "条件判断"
	NodeNameForm           = "表单"
	NodeNameHttp           = "HTTP(S)接口"
	NodeNameCustomNode     = "自定义节点"
)

// 表单字段 Label
const (
	FormLabelApiKey = "API Key"
	FormLabelModel  = "模型名称"

	FormLabelCondition = "判断条件"
)

// ======================== 枚举类型 ========================
type NodeGroup string

const (
	NodeGroupComponent NodeGroup = "component"
	NodeGroupBase      NodeGroup = "base"
	NodeGroupCustom    NodeGroup = "custom"
)

type NodeType string

const (
	// 组件
	NodeTypeTextModel      NodeType = "text_model"
	NodeTypeImageModel     NodeType = "image_model"
	NodeTypeVideoModel     NodeType = "video_model"
	NodeTypeSenseOptimize  NodeType = "sense_optimize"
	NodeTypeStoryOptimize  NodeType = "story_optimize"
	NodeTypeScriptOptimize NodeType = "script_optimize"
	NodeTypeAudioModel     NodeType = "audio_model"

	// 基础
	NodeTypeModel  NodeType = "model"
	NodeTypeMerge  NodeType = "merge"
	NodeTypeJudge  NodeType = "judge"
	NodeTypeForm   NodeType = "form"
	NodeTypeIntent NodeType = "intent"
	NodeTypeHttp   NodeType = "http"
	// 自定义
	NodeTypeCustomNode NodeType = "custom_node"
)

const (
	ModelTypeText  = 1
	ModelTypeImage = 2
)

// ======================== 结构定义 ========================
type NodeFormField struct {
	Value    string         `json:"value"`
	Field    string         `json:"field"`
	Label    string         `json:"label"` // 从常量来
	Type     string         `json:"type"`
	Required bool           `json:"required"`
	Default  any            `json:"default,omitempty"`
	Options  []SelectOption `json:"options"`
	Expand   any            `json:"expand"`
}

type SelectOption struct {
	Label string `json:"label"`
	Value string `json:"value"`
}

type ModelItem struct {
	ModelApiKey   string         `json:"modelApiKey"`
	ModelName     string         `json:"modelName"`
	ModelForm     map[string]any `json:"modelForm"`
	ModelResponse map[string]any `json:"modelResponse"`
}

type NodeItem struct {
	NodeId      string          `json:"nodeId"`
	NodeCode    NodeType        `json:"nodeCode"`
	ModelType   int             `json:"modelType"`
	NodeName    string          `json:"nodeName"` // 从常量来
	SkillOption bool            `json:"skillOption"`
	FormConfig  []NodeFormField `json:"formConfig"`
	ModelConfig []ModelItem     `json:"modelConfig"`
}

type NodeGroupItem struct {
	Group NodeGroup  `json:"group"`
	Label string     `json:"label"` // 从常量来
	Items []NodeItem `json:"items"`
}