orchestration

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

type AgentCommandParams struct {
	Message           string              `json:"message"`
	AgentID           *string             `json:"agent_id,omitempty"`
	HookID            *uint               `json:"hook_id,omitempty"`
	SessionID         string              `json:"session_id"`
	SessionKey        *string             `json:"session_key,omitempty"`
	Deliver           *bool               `json:"deliver,omitempty"`
	TimeoutMs         *int                `json:"timeout_ms,omitempty"`
	Policy            *RunExecutionPolicy `json:"policy,omitempty"`
	Lane              *string             `json:"lane,omitempty"`
	ExtraSystemPrompt *string             `json:"extra_system_prompt,omitempty"`
	InputProvenance   *InputProvenance    `json:"input_provenance,omitempty"`
	Routing           RoutingContext      `json:"routing"`
	Attachments       []map[string]any    `json:"attachments,omitempty"`
	IdempotencyKey    string              `json:"idempotency_key"`
	Label             *string             `json:"label,omitempty"`
	SpawnedBy         *string             `json:"spawned_by,omitempty"`
	Priority          string              `json:"priority,omitempty"`
	BudgetUSD         *float64            `json:"budget_usd,omitempty"`
	ParentTaskID      *string             `json:"parent_task_id,omitempty"`
	SourceViewID      *string             `json:"source_view_id,omitempty"`
	DispatchDelay     time.Duration       `json:"-"`
}

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

type AttemptEvent struct {
	AttemptID string
	RunID     string
	Change    int
	Status    string
	Ts        int64
}

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

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

type DispatchEnvelope struct {
	TaskID       string
	Prompt       string
	Reason       string
	RetryDelayMs int
	RetryAttempt int
	Resume       ResumeDirective
	Wake         WakeDirective
}

type ExecAsk string

type ExecHost string

type ExecSecurity string

type ExecutionInstance struct {
	Ctx        context.Context
	CancelFunc context.CancelFunc

	AttemptEventChan  chan AttemptEvent
	DispatchEventChan chan int

	InstanceLockKey        string
	FailedAttemptThreshold int
	// contains filtered or unexported fields
}

type ExecutionInstanceConfig struct {
	InstanceKey            string
	WorkspaceID            uint
	AgentID                *string
	Lane                   *string
	ExecutionClassKey      string
	FailedAttemptThreshold int
	InstanceLockKey        string
}

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

type ExecutionInstanceState struct {
	RunningAttempts  int
	PendingAttempts  int
	StoppingAttempts int
	FailedAttempts   []string
}

type ExecutionInstanceStore interface {
	GetOrCreateExecutionInstance(ctx context.Context, instance *types.AgentExecutionInstance) (*types.AgentExecutionInstance, error)
	GetExecutionInstanceByKey(ctx context.Context, instanceKey string) (*types.AgentExecutionInstance, error)
	UpdateExecutionInstanceState(
		ctx context.Context,
		instanceKey string,
		runningAttempts int,
		pendingAttempts int,
		stoppingAttempts int,
		desiredDispatchConcurrency int,
		status types.AgentExecutionInstanceStatus,
		lastSyncAt *time.Time,
	) error
}

type IExecutionInstance interface {
	ConsumeAttemptEvent(AttemptEvent)
	ConsumeDispatchTarget(int)
	HandleDispatchEvent(int) error
	Sync() error
}

type InputProvenance struct {
	Source        *string `json:"source,omitempty"`
	MessageID     *string `json:"message_id,omitempty"`
	Channel       *string `json:"channel,omitempty"`
	ToolCallID    *string `json:"tool_call_id,omitempty"`
	CorrelationID *string `json:"correlation_id,omitempty"`
}

type InstanceDispatchLocker interface {
	WithInstanceLock(ctx context.Context, lockKey string, fn func() error) error
}

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

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

type ResumeDirective struct {
	Enabled         bool
	ExcludeRunID    string
	CheckpointRunID string
}

type RoutingContext struct {
	To             *string `json:"to,omitempty"`
	ReplyTo        *string `json:"reply_to,omitempty"`
	Channel        *string `json:"channel,omitempty"`
	ReplyChannel   *string `json:"reply_channel,omitempty"`
	AccountID      *string `json:"account_id,omitempty"`
	ReplyAccountID *string `json:"reply_account_id,omitempty"`
	ThreadID       *string `json:"thread_id,omitempty"`
	GroupID        *string `json:"group_id,omitempty"`
	GroupChannel   *string `json:"group_channel,omitempty"`
	GroupSpace     *string `json:"group_space,omitempty"`
}

type RunExecutionPolicy struct {
	Host            ExecHost        `json:"host"`
	Security        ExecSecurity    `json:"security"`
	Ask             ExecAsk         `json:"ask"`
	RuntimeType     string          `json:"runtime_type"`
	WorkspaceAccess string          `json:"workspace_access"`
	NetworkEnabled  bool            `json:"network_enabled"`
	Interactive     bool            `json:"interactive"`
	Resources       map[string]any  `json:"resources,omitempty"`
	Retry           *RunRetryPolicy `json:"retry,omitempty"`
}

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

type RunFactoryConfig struct {
	Backend           repository.BackendRepository
	TaskQueue         repository.TaskQueue
	TerminalIO        repository.TerminalIORepository
	S2                *common.S2Client
	Lifecycle         *TaskLifecycle
	ResumeBarrier     *ResumeBarrier
	DefaultImage      string
	PublishTaskUpdate func(context.Context, uint, string)
}

type RunResultEnvelope struct {
	TaskID              string
	AttemptID           string
	ExitCode            int
	ErrorText           string
	ResultKey           string
	RetryAttempt        int
	PostRun             *types.RunExecutionPostRun
	WaitingForInput     bool
	Wake                WakeDirective
	SubtaskRequests     []*types.SubtaskRequest
	SourceWatchRequests []*types.SourceWatchRequest
}

type RunRetryPolicy struct {
	MaxAttempts int `json:"max_attempts,omitempty"`
	DelayMs     int `json:"delay_ms,omitempty"`
}

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

type SettleOpts struct {
	WaitingForInput bool
	WakeSignal      *types.RunExecutionWakeSignal
	Blocker         *types.TaskBlockerSpec
}

type SourceWatchRegistrar interface {
	RegisterTaskSourceWatches(ctx context.Context, task *types.AgentTask, wakeSignal *types.RunExecutionWakeSignal, requests []*types.SourceWatchRequest) (*types.TaskBlockerSpec, error)
	CleanupTaskSourceWatches(ctx context.Context, task *types.AgentTask) error
	HasTaskSourceWatches(ctx context.Context, task *types.AgentTask) bool
}

type TaskCommandPayload struct {
	Message           string
	Prompt            string
	OriginalMessage   string
	SessionID         string
	SessionKey        *string
	AgentID           *string
	HookID            *uint
	TimeoutMs         int
	Policy            RunExecutionPolicy
	Lane              *string
	ExtraSystemPrompt *string
	InputProvenance   *InputProvenance
	Deliver           *bool
	Attachments       []map[string]any
	InstanceKey       string
	Label             *string
	SpawnedBy         *string
	Priority          string
	Provider          *string
	Model             *string
	AgentConfig       map[string]any
	SourceViewID      *string
	Resume            ResumeDirective
}

type TaskEventBatch struct {
	TaskID             string                 `json:"task_id"`
	RunID              *string                `json:"run_id,omitempty"`
	Task               *types.AgentTask       `json:"task,omitempty"`
	Run                *types.AgentRun        `json:"run,omitempty"`
	Interaction        *types.RunInteraction  `json:"interaction,omitempty"`
	Blocker            *types.ResolvedBlocker `json:"blocker"`
	Logs               []common.TaskLogEntry  `json:"logs"`
	RunEvents          []map[string]any       `json:"run_events"`
	Outputs            []*types.TaskOutput    `json:"outputs,omitempty"`
	NextLogCursor      int64                  `json:"next_log_cursor"`
	NextRunEventCursor int                    `json:"next_run_event_cursor"`
}

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

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

type TaskUpdateParams struct {
	Priority    *string        `json:"priority,omitempty"`
	BudgetUSD   *float64       `json:"budget_usd,omitempty"`
	PayloadJSON map[string]any `json:"payload_json,omitempty"`
	RoutingJSON map[string]any `json:"routing_json,omitempty"`
}

type ToolExecutor interface {
	ExecuteDeferred(ctx context.Context, workspaceID, memberID uint, toolName string, args []string) (stdout, stderr string, exitCode int, err error)
	RecordToolRejection(ctx context.Context, taskID, tool, command string) error
	GrantWritePreapproval(ctx context.Context, taskID string) error
}

type WakeDirective struct {
	DelayMinutes   int
	Reason         string
	FollowUpPrompt string
	Agenda         []*types.TaskWakeAgendaItem
}

type WorkspaceLiveBatch struct {
	Tasks   []*types.AgentTask  `json:"tasks"`
	Outputs []*types.TaskOutput `json:"outputs"`
}