workflow

type BranchFunc func(result *ExecutionResult, state ContextMap) string

type BranchMode string

type ChannelSink struct {
	// contains filtered or unexported fields
}

type ComposerDocument struct {
	Definition *WorkflowDefinition `json:"definition"`
	Layout     json.RawMessage     `json:"layout,omitempty"`
}

type ContextMap map[string]*ExecutionResult

type ControlSignal string

type ControlSignalError struct {
	Signal ControlSignal
	Target string
}

type Duration struct {
	time.Duration
}

type EdgeEnv struct {
	State  map[string]*ExecutionResult
	Result *ExecutionResult
	Node   string
}

type EdgeSpec struct {
	From  string `json:"from"`
	To    string `json:"to"`
	When  string `json:"when,omitempty"`
	Label string `json:"label,omitempty"`
	Order int    `json:"order,omitempty"`
}

type Engine struct {
	Registry    *Registry
	Store       StateStore
	Concurrency int
	Logger      *slog.Logger
	FailFast    bool
}

type Event struct {
	Type       EventType `json:"type,omitempty"`
	WorkflowID string    `json:"workflow_id,omitempty"`
	RunID      string    `json:"run_id,omitempty"`
	NodeID     string    `json:"node_id,omitempty"`
	Time       time.Time `json:"time,omitempty"`
	Data       any       `json:"data,omitempty"`
	Error      string    `json:"error,omitempty"`
}

type EventSink interface {
	Emit(ctx context.Context, e Event)
}

type EventType string

type ExecutableUnit interface {
	GetUnitMeta() *Unit
	Execute(ctx context.Context, state ContextMap, self *Node) (*ExecutionResult, error)
}

type ExecutionResult struct {
	NodeName      string        `json:"node_name,omitempty"`
	Data          any           `json:"data,omitempty"`
	Stream        bool          `json:"stream,omitempty"`
	Raw           any           `json:"raw,omitempty"`
	Error         string        `json:"error,omitempty"`
	Control       ControlSignal `json:"control,omitempty"`
	ControlTarget string        `json:"control_target,omitempty"`
}

type ExecutionState struct {
	WorkflowID string                `json:"workflow_id,omitempty"`
	RunID      string                `json:"run_id,omitempty"`
	Status     RunStatus             `json:"status,omitempty"`
	StartedAt  time.Time             `json:"started_at,omitempty"`
	UpdatedAt  time.Time             `json:"updated_at,omitempty"`
	Nodes      map[string]*NodeState `json:"nodes,omitempty"`
}

type Graph struct {
	Nodes map[string]*Node
	Edges map[string][]string
	Hooks struct {
		Before func(name string, state ContextMap)
		After  func(name string, result any, err error, state ContextMap)
	}
	// contains filtered or unexported fields
}

type GraphJSON struct {
	Nodes map[string]NodeJSON `json:"nodes"`
	Edges map[string][]string `json:"edges"`
}

type Input struct {
	Data      any    `json:"data,omitempty"`      //最终输出
	DataType  string `json:"data_type,omitempty"` // plaintext, json, json_array，socket
	Slottable bool   `json:"slottable,omitempty"` // 是否是可插槽的
}

type JoinPolicy string

type LayoutValidator interface {
	Validate(def *WorkflowDefinition, layout map[string]any) error
}

type LayoutValidatorFunc func(def *WorkflowDefinition, layout map[string]any) error

type LoopCondFunc func(state ContextMap) bool

type Node struct {
	ID           string
	Name         string
	Input        *Input
	Params       map[string]any
	Execute      NodeFunc
	Branch       BranchFunc            // 可选分支函数
	Parallel     bool                  // 是否并行节点
	LoopCond     func(ContextMap) bool // 可选循环条件
	ExportFields []string              // 导出字段，用于供下游引用
}

type NodeFunc func(ctx context.Context, state ContextMap, self *Node) (*ExecutionResult, error)

type NodeJSON struct {
	UnitID string         `json:"unit_id"`
	Name   string         `json:"name"`
	Input  *Input         `json:"input,omitempty"`
	Params map[string]any `json:"params,omitempty"`
}

type NodeOptions struct {
	Timeout         Duration   `json:"timeout,omitempty"`
	Retries         int        `json:"retries,omitempty"`
	RetryBackoff    Duration   `json:"retry_backoff,omitempty"`
	RetryBackoffMax Duration   `json:"retry_backoff_max,omitempty"`
	Join            JoinPolicy `json:"join,omitempty"`
	ContinueOnError bool       `json:"continue_on_error,omitempty"`
	BranchMode      BranchMode `json:"branch_mode,omitempty"`
}

type NodeRole string

type NodeSpec struct {
	ID           string         `json:"id,omitempty"`
	Name         string         `json:"name,omitempty"`
	Unit         string         `json:"unit,omitempty"`
	UnitID       string         `json:"unit_id,omitempty"`
	Input        *Input         `json:"input,omitempty"`
	Params       map[string]any `json:"params,omitempty"`
	ExportFields []string       `json:"export_fields,omitempty"`
	Options      *NodeOptions   `json:"options,omitempty"`
	Role         NodeRole       `json:"role,omitempty"`
	RunMode      RunMode        `json:"run_mode,omitempty"`
	DependsOn    []string       `json:"depends_on,omitempty"`
}

type NodeState struct {
	ID         string           `json:"id,omitempty"`
	Name       string           `json:"name,omitempty"`
	Status     NodeStatus       `json:"status,omitempty"`
	Attempts   int              `json:"attempts,omitempty"`
	StartedAt  time.Time        `json:"started_at,omitempty"`
	FinishedAt time.Time        `json:"finished_at,omitempty"`
	Result     *ExecutionResult `json:"result,omitempty"`
	Error      string           `json:"error,omitempty"`
}

type NodeStatus string

type Registry struct {
	// contains filtered or unexported fields
}

type RunMode string

type RunOptions struct {
	RunID         string
	Start         []string
	Concurrency   int
	Store         StateStore
	Sink          EventSink
	FailFast      *bool
	AllowCycles   bool
	StopOnControl bool
	SeedState     ContextMap
	SeedExports   map[string][]string
	ResumeState   *ExecutionState
}

type RunStatus string

type StateStore interface {
	Save(ctx context.Context, state *ExecutionState) error
	Load(ctx context.Context, workflowID, runID string) (*ExecutionState, error)
}

type Unit struct {
	ID          string `json:"id"`
	UnitName    string `json:"unit_name,omitempty"`
	DisplayName string `json:"display_name,omitempty"` // UI 显示名（如“调用接口”）
	Status      string `json:"status,omitempty"`
	ErrMsg      string `json:"err_msg,omitempty"`    // 如果失败，存错误
	OutputRef   string `json:"output_ref,omitempty"` // 输出导出的字段名（用于 UI 显示）
}

type UnitFactory func() ExecutableUnit

type WorkflowDefinition struct {
	ID    string               `json:"id,omitempty"`
	Name  string               `json:"name,omitempty"`
	Start []string             `json:"start,omitempty"`
	Nodes map[string]*NodeSpec `json:"nodes,omitempty"`
	Edges []EdgeSpec           `json:"edges,omitempty"`
}