worker

type AgentOpsConfig struct {
	// Enabled controls whether AgentOps tracing is active.
	Enabled bool

	// Store is the AgentOps store for persisting traces.
	// If nil and Enabled is true, a no-op store is used.
	Store agentops.Store
}

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

type ClientOption func(*WorkerClient)

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

type CoordinatorConfig struct {
	// ID is the unique identifier for this coordinator.
	ID string

	// Workflow is the workflow executor for orchestration.
	// If nil, a simple sequential executor is used.
	Workflow WorkflowExecutor

	// AgentOps configures observability.
	AgentOps *AgentOpsConfig

	// Logger is the logger for this coordinator.
	Logger *slog.Logger

	// Timeout is the maximum execution time for a workflow.
	// Default: 5 minutes.
	Timeout time.Duration

	// MaxRetries is the maximum number of retries for failed tasks.
	// Default: 3.
	MaxRetries int
}

type CoordinatorRequest struct {
	// WorkflowID is a unique identifier for this workflow instance.
	// If empty, one will be generated.
	WorkflowID string `json:"workflow_id,omitempty"`

	// Input contains workflow-specific input data.
	Input map[string]any `json:"input"`
}

type CoordinatorResponse struct {
	// WorkflowID echoes back the workflow ID.
	WorkflowID string `json:"workflow_id"`

	// Output contains workflow output data.
	Output map[string]any `json:"output"`

	// Stats contains execution statistics.
	Stats WorkflowStats `json:"stats"`

	// Error contains error information if the workflow failed.
	Error *WorkerError `json:"error,omitempty"`
}

type HealthStatus struct {
	// Status is one of "healthy", "degraded", or "unhealthy".
	Status string `json:"status"`

	// Details contains additional health information.
	Details map[string]string `json:"details,omitempty"`
}

type LLMConfig struct {
	// Provider is the LLM provider (e.g., "anthropic", "openai", "gemini").
	Provider string

	// Model is the model name (e.g., "claude-sonnet-4-20250514").
	Model string

	// APIKey is the API key for the provider.
	// If empty, will be read from environment.
	APIKey string

	// BaseURL is an optional custom base URL for the provider.
	BaseURL string

	// Temperature controls randomness (0.0-1.0).
	Temperature float64

	// MaxTokens is the maximum tokens in the response.
	MaxTokens int
}

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

type Request struct {
	// ID is a unique identifier for this request.
	ID string `json:"id"`

	// TaskID is the AgentOps task ID for tracing.
	TaskID string `json:"task_id,omitempty"`

	// WorkflowID is the workflow this request belongs to.
	WorkflowID string `json:"workflow_id,omitempty"`

	// Input contains worker-specific input data.
	Input map[string]any `json:"input"`
}

type Response struct {
	// RequestID echoes back the request ID.
	RequestID string `json:"request_id"`

	// Output contains worker-specific output data.
	Output map[string]any `json:"output"`

	// Error contains structured error information if the request failed.
	Error *WorkerError `json:"error,omitempty"`
}

type SequentialWorkflow struct {
	// WorkerIDs is the ordered list of workers to execute.
	WorkerIDs []string

	// InputMapper maps workflow input to worker input.
	// If nil, the workflow input is passed directly to each worker.
	InputMapper func(workerID string, workflowInput, previousOutput map[string]any) map[string]any
}

type Worker interface {
	// ID returns the unique identifier for this worker instance.
	ID() string

	// Type returns the worker type (e.g., "research", "synthesis").
	// Multiple workers can have the same type.
	Type() string

	// Version returns the worker implementation version.
	Version() string

	// Init initializes the worker. Called once before Execute.
	Init(ctx context.Context) error

	// Shutdown gracefully shuts down the worker.
	Shutdown(ctx context.Context) error

	// Health returns the current health status of the worker.
	Health(ctx context.Context) HealthStatus

	// Execute performs the worker's task.
	Execute(ctx context.Context, req *Request) (*Response, error)
}

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

type WorkerConfig struct {
	// ID is the unique identifier for this worker instance.
	ID string

	// Type is the worker type (e.g., "research", "synthesis").
	Type string

	// Version is the worker implementation version.
	Version string

	// LLM configures the LLM client for workers that need it.
	// Optional - not all workers require LLM.
	LLM *LLMConfig

	// AgentOps configures observability.
	AgentOps *AgentOpsConfig

	// Logger is the logger for this worker.
	// If nil, a default logger is used.
	Logger *slog.Logger

	// Timeout is the maximum execution time for a single request.
	// Default: 60 seconds.
	Timeout time.Duration
}

type WorkerDispatcher interface {
	// Dispatch calls a worker by ID with the given request.
	Dispatch(ctx context.Context, workerID string, req *Request) (*Response, error)
}

type WorkerError struct {
	// Code is a machine-readable error code.
	Code string `json:"code"`

	// Message is a human-readable error message.
	Message string `json:"message"`

	// Details contains additional error context.
	Details any `json:"details,omitempty"`

	// Cause is the underlying error, if any.
	Cause error `json:"-"`
}

type WorkflowExecutor interface {
	// Execute runs the workflow with the given input.
	// The dispatcher is used to call workers within the workflow.
	Execute(ctx context.Context, input map[string]any, dispatcher WorkerDispatcher) (map[string]any, error)
}

type WorkflowFunc func(ctx context.Context, input map[string]any, dispatcher WorkerDispatcher) (map[string]any, error)

type WorkflowStats struct {
	// Duration is the total workflow execution time.
	Duration time.Duration `json:"duration"`

	// TaskCount is the number of tasks executed.
	TaskCount int `json:"task_count"`

	// SuccessCount is the number of successful tasks.
	SuccessCount int `json:"success_count"`

	// FailureCount is the number of failed tasks.
	FailureCount int `json:"failure_count"`

	// RetryCount is the number of retried tasks.
	RetryCount int `json:"retry_count"`
}