hooks

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

type CompactionMissEvent struct {
	ToolName       string
	OriginalSize   int
	CompressedSize int
}

type CompactionTracker interface {
	MarkCompressed(toolName string, originalSize, compressedSize int)
	// GetChunkBoost returns a multiplier (>1) for max_chunks on a tool that
	// previously caused a compaction miss, or 0 if no boost is needed.
	GetChunkBoost(toolName string) int
}

type ContextBudgetEvent struct {
	PersonaTokens    int
	ToolSchemaTokens int
	HistoryTokens    int
	TotalTokens      int
	MaxTokens        int
	UtilizationPct   float64
}

type DryRunEvent struct {
	PlannedSteps  int
	ToolsUsed     []string
	EstimatedCost string
}

type ExecutionHook interface {
	// OnIterationStart is called before each adaptive loop iteration executes.
	OnIterationStart(ctx context.Context, event IterationStartEvent)

	// OnIterationEnd is called after each adaptive loop iteration completes,
	// with the iteration's result status and tool usage.
	OnIterationEnd(ctx context.Context, event IterationEndEvent)

	// OnReflection is called after a RAR reflection step completes.
	OnReflection(ctx context.Context, event ReflectionEvent)

	// OnToolValidation is called when the critic middleware evaluates a tool call.
	OnToolValidation(ctx context.Context, event ToolValidationEvent)

	// OnDryRun is called when a dry run simulation completes.
	OnDryRun(ctx context.Context, event DryRunEvent)

	// OnPlanExecution is called when the orchestrator starts executing a plan.
	OnPlanExecution(ctx context.Context, event PlanExecutionEvent)

	// OnContextBudget is called before each model execution to report token utilization.
	OnContextBudget(ctx context.Context, event ContextBudgetEvent)

	// OnCompactionMiss is called when the runner detects a likely infinite loop caused by output compaction.
	OnCompactionMiss(ctx context.Context, event CompactionMissEvent)
}

type IterationEndEvent struct {
	Iteration      int
	Status         string         // "running", "success", "failure"
	ToolCallCounts map[string]int // per-tool call counts for this iteration
	TaskCompleted  bool
	Output         string
}

type IterationStartEvent struct {
	Goal          string
	Iteration     int
	MaxIterations int
}

type NoOpHook struct{}

type PlanExecutionEvent struct {
	Flow      string // "sequence", "parallel", "fallback"
	StepCount int
	StepNames []string
}

type ReflectionEvent struct {
	Iteration     int
	Monologue     string
	ShouldProceed bool
}

type ToolValidationEvent struct {
	ToolName string
	Allowed  bool
	Reason   string // non-empty when Allowed=false
}