engine

type BudgetMetrics struct {
	Type  string     `json:"type"`
	Used  float64    `json:"used"`
	Limit float64    `json:"limit"`
	Run   RunMetrics `json:"run"`
}

type CompactionManager interface {
	Compact(context.Context, CompactionRequest) (compaction.Result, []session.Message, error)
}

type CompactionRequest struct {
	RunID                string
	SessionID            string
	TraceID              string
	Step                 int
	History              []session.Message
	Policy               contextpolicy.Policy
	Trigger              compaction.Trigger
	Reason               compaction.Reason
	Phase                compaction.Phase
	Provider             provider.Provider
	ProviderName         string
	Model                string
	PreviousCompactionID string
	PreviousSummary      string
	ContextUsage         contextpolicy.Usage
	Details              map[string]string
}

type CompletionPolicy string

type CompletionReason string

type Config struct {
	Provider provider.Provider
	Tools    *tools.Registry
	Store    session.Store
	Prompt   cache.Store
	// Artifacts is required when a tool output policy preserves truncated full
	// output. Runtime and harness callers provide a default store; direct engine
	// callers may leave it nil only if their tools do not need preservation.
	Artifacts    artifact.Store
	SystemPrompt string
	Sink         event.Sink
	Approver     tools.Approver
	StopHook     StopHook
	Compactor    CompactionManager
	Options      Options
}

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

type ContinuationReason string

type ControlDisposition string

type ControlSignal struct {
	Disposition ControlDisposition
	Name        string
	CallID      string
	Payload     map[string]any
	// OutputText is the human-readable control result. For ControlContinue it is
	// provider-visible; host-only details must stay in Payload.
	OutputText string
	ArgsHash   string
	Labels     map[string]string
}

type ControlSpec struct {
	Definitions []provider.ToolDefinition
	Project     func(provider.ToolCall) (ControlSignal, bool, error)
}

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

type LocalCompactionManager struct {
	Generator compaction.SummaryGenerator
	Now       func() time.Time
}

type Options struct {
	RunID                    string
	SessionID                string
	TraceID                  string
	ProviderName             string
	Model                    string
	Labels                   RunLabels
	CacheNamespace           string
	CacheRetention           cache.Retention
	ContextPolicy            contextpolicy.Policy
	MaxEmptyProviderRetries  int
	NoProgressLimit          int
	DuplicateToolLimit       int
	WallTime                 time.Duration
	MaxTotalTokens           int64
	MaxCostUSD               float64
	MaxToolCalls             int
	HostedToolDefinitions    []provider.HostedToolDefinition
	CompletionPolicy         CompletionPolicy
	ControlSpec              ControlSpec
	PreviousProviderState    *provider.State
	MaxLengthContinuations   int
	MaxStopHookContinuations int
	// contains filtered or unexported fields
}

type PressureAnchorState = cache.PressureAnchorState

type ProviderUsageContextStatus struct {
	Phase                string                        `json:"phase"`
	RequestID            string                        `json:"request_id,omitempty"`
	LogicalRequestID     string                        `json:"logical_request_id,omitempty"`
	Attempt              int                           `json:"attempt,omitempty"`
	Usage                provider.Usage                `json:"usage"`
	RequestEstimate      contextpolicy.RequestEstimate `json:"request_estimate"`
	ContextPressure      contextpolicy.ContextPressure `json:"context_pressure"`
	UsedRatio            float64                       `json:"used_ratio,omitempty"`
	ThresholdRatio       float64                       `json:"threshold_ratio,omitempty"`
	Status               string                        `json:"status"`
	CompactionGeneration int                           `json:"compaction_generation,omitempty"`
	CompactionWindowID   string                        `json:"compaction_window_id,omitempty"`
}

type RequestShapeHashes = cache.RequestShapeHashes

type Result struct {
	Status             Status
	Output             string
	Err                error
	Metrics            RunMetrics
	Messages           []session.Message
	CompletionReason   CompletionReason
	ContinuationReason ContinuationReason
	FinishReason       provider.FinishReason
	RawFinishReason    string
	FinishInferred     bool
	ControlSignal      *ControlSignal
	ProviderState      *provider.State
}

type RunDecision struct {
	CompletionReason   CompletionReason
	ContinuationReason ContinuationReason
	FinishReason       provider.FinishReason
	RawFinishReason    string
	FinishInferred     bool
	Detail             string
	ControlSignal      *ControlSignal
	ProviderState      *provider.State
}

type RunInput struct {
	RunID                 string
	SessionID             string
	TraceID               string
	Labels                RunLabels
	PreviousProviderState *provider.State
	History               []session.Message
}

type RunLabels struct {
	Correlation map[string]string
	Host        map[string]string
}

type RunMetrics struct {
	Usage       provider.Usage `json:"usage"`
	Steps       int            `json:"steps"`
	LLMRequests int            `json:"llm_requests"`
	ToolCalls   int            `json:"tool_calls"`
	Compactions int            `json:"compactions"`
	Retries     int            `json:"retries"`
	WallTimeMS  int64          `json:"wall_time_ms,omitempty"`
}

type Status string

type StepMetrics struct {
	Step               int            `json:"step"`
	Provider           string         `json:"provider,omitempty"`
	Model              string         `json:"model,omitempty"`
	Usage              provider.Usage `json:"usage"`
	ProviderLatencyMS  int64          `json:"provider_latency_ms,omitempty"`
	ToolLatencyMS      int64          `json:"tool_latency_ms,omitempty"`
	ToolCalls          int            `json:"tool_calls,omitempty"`
	Retries            int            `json:"retries,omitempty"`
	FinishReason       string         `json:"finish_reason,omitempty"`
	RawFinishReason    string         `json:"raw_finish_reason,omitempty"`
	FinishInferred     bool           `json:"finish_inferred,omitempty"`
	CompletionReason   string         `json:"completion_reason,omitempty"`
	ContinuationReason string         `json:"continuation_reason,omitempty"`
}

type StepOutput struct {
	Text            string
	Reasoning       string
	Calls           []provider.ToolCall
	Usage           provider.Usage
	ResponseID      string
	Retry           bool
	Truncated       bool
	FinishReason    provider.FinishReason
	RawFinishReason string
	FinishInferred  bool
	ResponseState   *provider.State
}

type StopHook func(context.Context, StopHookContext) (StopHookResult, error)

type StopHookContext struct {
	RunID           string
	SessionID       string
	TraceID         string
	Step            int
	LastAssistant   session.Message
	Messages        []session.Message
	FinishReason    provider.FinishReason
	RawFinishReason string
	FinishInferred  bool
	Metrics         RunMetrics
}

type StopHookResult struct {
	Continue bool
	Prompt   string
	Reason   string
}