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"`
}

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 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 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 RunRetryPolicy struct {
	MaxAttempts int `json:"max_attempts,omitempty"`
	DelayMs     int `json:"delay_ms,omitempty"`
}

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"`
	Logs               []common.TaskLogEntry `json:"logs"`
	RunEvents          []map[string]any      `json:"run_events"`
	PendingInputs      []types.PendingInput  `json:"pending_inputs"`
	NextLogCursor      int64                 `json:"next_log_cursor"`
	NextRunEventCursor int                   `json:"next_run_event_cursor"`
}

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

type TaskQueueStore interface {
	UpdateTaskState(ctx context.Context, taskID string, state types.AgentTaskState, dropReason *string, targetRunID *string) error
	PushQueueToken(ctx context.Context, token string) error
	PopQueueToken(ctx context.Context, timeout time.Duration) (string, error)
	GetModeTaskID(ctx context.Context, modeKey string) (string, error)
	SetModeTaskID(ctx context.Context, modeKey string, taskID string, ttl time.Duration) error
	AddModeKey(ctx context.Context, modeKey string) (bool, error)
	RemoveModeKey(ctx context.Context, modeKey string) error
	GetDelModeTaskID(ctx context.Context, modeKey string) (string, error)
}