flux

type Asset interface {
	// contains filtered or unexported methods
}

type Backend interface {
	// PersistNode 持久化单个节点的状态和输出。
	PersistNode(ctx context.Context, taskID string, node string, state NodeState, output map[string]any) error

	// CreateAwait 为异步节点创建一个外部等待凭证。
	// 返回 bindingID：Flux 用它关联后续的 Notify。
	CreateAwait(ctx context.Context, taskID string, node string, providerTaskID string, input map[string]any) (bindingID string, err error)

	// CompleteAwait 原子地将 binding 标记为完成（waiting→completing）。
	// 返回 (claimed, error)。claimed=false 表示已被其他线程完成（幂等安全）。
	CompleteAwait(ctx context.Context, bindingID string) (claimed bool, err error)

	// Lock 获取分布式锁。用于 Resume 并发控制。
	// 返回 unlock 函数。获取失败时阻塞等待。
	Lock(ctx context.Context, key string) (unlock func(), err error)

	// LoadState 加载任务的所有节点状态（crash 恢复用）。
	LoadState(ctx context.Context, taskID string) (*TaskState, error)
}

type Config struct {
	Backend Backend // 宿主提供的持久化 + 异步基础设施（向后兼容）

	// ── v3 Store 接口（可选）──
	// 当 Store 接口非 nil 时，Engine 优先使用 Store 接口（而非 Backend）进行持久化。
	// 当 Store 接口为 nil 时，Engine 回退到 Backend（向后兼容）。
	WorkflowStore store.WorkflowStore // nil → 回退到 Backend
	AwaitStore    store.AwaitStore    // nil → 回退到 Backend
	TraceStore    store.TraceStore    // nil → 不记录 trace
}

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

type Event struct {
	Provider       string         // 外部服务标识（如 "tts"、"aliyun"）
	ProviderTaskID string         // 外部 Provider 的任务 ID
	Output         map[string]any // 外部任务完成的产出
	Error          string         // 外部任务失败的信息（为空表示成功）
}

type LLMConfig struct {
	APIKey  string `yaml:"api_key"`
	BaseURL string `yaml:"base_url"`
	Model   string `yaml:"model"`
}

type NodeSnapshot struct {
	State  NodeState      `json:"state"`
	Output map[string]any `json:"output,omitempty"`
}

type NodeState int

type RunRequest struct {
	Asset  string         // 已注册的 Asset 名
	Input  map[string]any // 任务输入
	TaskID string         // 可选：宿主已有的 task ID（如 DreamAI 的 Task.ID）。为空则自动生成
}

type RunResult struct {
	Status RunStatus      // Completed / Suspended / Failed
	Output map[string]any // 完成时的最终产出
	TaskID string         // 内部 task ID（供 Notify 定位）
	Err    error          // 失败时的错误
}

type RunStatus int

type TaskState struct {
	Input map[string]any          `json:"input"`
	Nodes map[string]NodeSnapshot `json:"nodes"`
}

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

type WorkflowToolConfig struct {
	// Provider 是 DAGPlanner 使用的 LLM provider。必填。
	// 类型为 model.Completer 接口——*OpenAICompatibleProvider 满足它，
	// 任意 OpenAI 兼容端点/网关（DeepSeek/Qwen/Claude-via-gateway）皆可注入。
	Provider model.Completer

	// ModelName 是传给 LLM 的模型名。默认 "deepseek-chat"。
	ModelName string

	// ToolReg 是 Workflow 执行时可用的工具注册表。必填。
	// 这些工具是 Workflow DAG 中的节点——图片/视频/TTS Provider 等，不是 Agent 的对话工具。
	ToolReg *tool.Registry

	// ── v3 Store 接口（可选）──
	WFStore    store.WorkflowStore // nil → 内存模式（仅 sync workflow）
	AwaitStore store.AwaitStore    // nil → async 节点会失败
	TraceStore store.TraceStore    // nil → 不记录 trace

	// MaxRepairs 是 DAGPlanner 的 validate→repair 最大轮数。默认 3。
	MaxRepairs int
}