executor

type AgentSessionHandler interface {
	SetAgentSession(*ir.AgentSession)
	GetAgentSession() *ir.AgentSession
}

type ChatMessageHandler interface {
	SetContext([]ir.LLMMessage)
	GetMessages() []ir.LLMMessage
}

type DAGExecutor interface {
	Executor

	// SetParams sets the parameters for running a sub DAG.
	SetParams(RunParams)
}

type DeclaredOutputsProvider interface {
	OutputsProvider
	PublishesDeclaredOutputs() bool
}

type EnqueueRequest struct {
	DAG            *ir.DAG
	RootDAGRun     ir.DAGRunRef
	RunID          string
	QueueName      string
	ProfileName    string
	TriggerActor   string
	ParallelItem   string
	WorkerSelector map[string]string
}

type EnqueueResult struct {
	Status        ir.Status
	AlreadyExists bool
}

type Enqueuer interface {
	Enqueue(ctx context.Context, req EnqueueRequest) (EnqueueResult, error)
}

type Executor interface {
	SetStdout(out io.Writer)
	SetStderr(out io.Writer)
	Kill(sig os.Signal) error
	Run(ctx context.Context) error
}

type ExecutorFactory func(ctx context.Context, step ir.Step) (Executor, error)

type ExitCoder interface {
	ExitCode() int
}

type NodeStatusDeterminer interface {
	DetermineNodeStatus() (ir.NodeStatus, error)
}

type OutputsProvider interface {
	GetOutputs() map[string]any
}

type OutputsValueProvider interface {
	GetOutputsValue() any
}

type ParallelExecutor interface {
	Executor

	// SetParamsList sets the parameters for running multiple sub DAGs in parallel.
	SetParamsList([]RunParams)
}

type ProgressCallbackAware interface {
	SetProgressCallback(func())
}

type PushBackAware interface {
	SetPushBackContext(inputs map[string]string, iteration int)
}

type PushBackPreviousStdoutAware interface {
	SetPushBackPreviousStdout(path string)
}

type RunParams struct {
	RunID          string
	Params         string
	ParallelItem   string
	DAGName        string
	WorkerSelector map[string]string
	// PassedEnv holds resolved "KEY=value" pairs the parent opted to share
	// with the child run via the step's pass_env field.
	PassedEnv []string
}

type StatusDetailsProvider interface {
	GetStatusDetails() []ir.NodeStatusDetail
}

type Stopper interface {
	Stop(cmdutil.TerminationIntent) error
}

type SubDAGExecutor struct {
	// DAG is the sub DAG to execute.
	// For local DAGs, this DAG's Location will be set to a temporary file.
	DAG *ir.DAG
	// contains filtered or unexported fields
}

type SubRunProvider interface {
	GetSubRuns() []ir.SubDAGRun
}

type SubWorkflowCancelIntent struct {
	Mode   SubWorkflowCancelMode
	Signal os.Signal
}

type SubWorkflowCancelMode string

type SubWorkflowCancelRequest struct {
	DAG        *ir.DAG
	RootDAGRun ir.DAGRunRef
	RunID      string
	Intent     SubWorkflowCancelIntent
}

type SubWorkflowRequest struct {
	DAG               *ir.DAG
	ParentDAG         *ir.DAG
	RootDAGRun        ir.DAGRunRef
	ParentDAGRun      ir.DAGRunRef
	RunID             string
	Params            string
	ParallelItem      string
	ProfileName       string
	TriggerActor      string
	WorkDir           string
	WorkerSelector    map[string]string
	ExternalStepRetry bool
	Reuse             bool
	RetryPath         dagrun.RetryPath
	Workspace         *SubWorkflowWorkspace
	// PassedEnv holds resolved "KEY=value" pairs the parent opted to share
	// with the child run via the step's pass_env field.
	PassedEnv []string
}

type SubWorkflowRetryRequest struct {
	SubWorkflowRequest
	StepName            string
	IncludeDownstream   bool
	BypassPreconditions bool
}

type SubWorkflowRunner interface {
	ShouldRun(ctx context.Context, req SubWorkflowRequest) bool
	Run(ctx context.Context, req SubWorkflowRequest) (*ir.RunStatus, error)
	Retry(ctx context.Context, req SubWorkflowRetryRequest) (*ir.RunStatus, error)
	Cancel(ctx context.Context, req SubWorkflowCancelRequest) error
}

type SubWorkflowWorkspace struct {
	Descriptor workspacebundle.Descriptor
	Archive    []byte
}

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

type TaskOption func(*dispatch.DispatchTask)

type ToolDefinitionProvider interface {
	GetToolDefinitions() []ir.ToolDefinition
}

type WorkspaceSeed struct {
	Descriptor workspacebundle.Descriptor
	Archive    []byte
}