compaction

type Agent struct {
	Name  string
	Model *ModelRef
}

type AgentProvider interface {
	GetAgent(ctx context.Context, name string) (Agent, error)
}

type AgentProviderFunc func(ctx context.Context, name string) (Agent, error)

type AutoContinueInput struct {
	SessionID string
	Agent     string
	Model     Model
	Provider  ProviderInfo
	Message   msgmodel.User
	Overflow  bool
}

type CompactingResult struct {
	Context []string
	Prompt  *string
}

type CompactionDecision struct {
	SessionID string  `json:"sessionID"`
	Status    string  `json:"status"`
	Before    float64 `json:"beforeTokens"`
	After     float64 `json:"afterTokens"`
	Capacity  float64 `json:"capacityTokens"`
	Low       float64 `json:"lowTokens"`
	High      float64 `json:"highTokens"`
	// DroppedTail: the watermark rebuild removed the verbatim tail.
	// StubbedSummary: it also had to replace the summary with the capacity
	// stub because the summary block alone did not fit.
	DroppedTail    bool `json:"droppedTail"`
	StubbedSummary bool `json:"stubbedSummary,omitempty"`
	// SummaryStatus is valid, normalized, fallback, no-head (nothing to
	// summarize, no call made), overflow (the summary request itself exceeded
	// the model), summary-error (the call failed), summary-stopped, or
	// capacity-fallback (the watermark rebuild replaced the record). Every
	// status except valid and normalized installs the deterministic record.
	SummaryStatus string `json:"summaryStatus"`
	SummaryClass  string `json:"summaryClass,omitempty"`
	SummaryError  string `json:"summaryError,omitempty"`

	TranscriptMessages  int     `json:"transcriptMessages"`
	TranscriptChars     int     `json:"transcriptChars"`
	TranscriptCutChars  float64 `json:"transcriptCutChars,omitempty"`
	PromptChars         int     `json:"promptChars"`
	SummaryPromptTokens uint64  `json:"summaryPromptTokens"`
	SummaryOutputTokens uint64  `json:"summaryOutputTokens"`
	SummaryWallMs       int64   `json:"summaryWallMs"`

	TailBudget             float64 `json:"tailBudget"`
	TailMessages           int     `json:"tailMessages"`
	TailTokens             float64 `json:"tailTokens"`
	TailTruncatedOutputs   int     `json:"tailTruncatedOutputs"`
	TailTruncatedReasoning int     `json:"tailTruncatedReasoning,omitempty"`
	// PreviousSummaryCarried is set when a deterministic record carried the
	// previous boundary's summary forward verbatim.
	PreviousSummaryCarried bool `json:"previousSummaryCarried,omitempty"`
}

type CompletedCompaction struct {
	UserIndex      int
	AssistantIndex int
	Summary        *string
}

type ConfigProvider interface {
	GetConfig(ctx context.Context) (overflow.Config, error)
}

type ConfigProviderFunc func(ctx context.Context) (overflow.Config, error)

type ContextCapacityError struct {
	After float64
	High  float64
}

type ContextSizer interface {
	EstimateContext(ctx context.Context, messages []msgmodel.WithParts, model Model) (float64, error)
}

type ContextSizerFunc func(
	ctx context.Context, messages []msgmodel.WithParts, model Model,
) (float64, error)

type Controller struct {
	Compaction *Service
}

type CreateInput struct {
	SessionID string
	Agent     string
	Model     ModelRef
	Auto      bool
	Overflow  *bool
}

type DecisionSink interface {
	CompactionDecision(decision CompactionDecision)
}

type DecisionSinkFunc func(decision CompactionDecision)

type Dependencies struct {
	Store      steploop.Store
	Config     ConfigProvider
	Agents     AgentProvider
	Provider   ModelProvider
	Plugin     Plugin
	Processors ProcessorFactory
	Evidence   EvidenceSelector
	Sizer      ContextSizer
	Decisions  DecisionSink
	Events     EventSink
	Instance   InstanceContext
	// ChangedFiles reports the workspace's changed files as preformatted
	// lines, computed by code (a diffstat against the starting tree plus the
	// status). It is pinned beside every summary as a record the model cannot
	// misremember; nil disables the pin.
	ChangedFiles func(ctx context.Context) []string

	NewID func(prefix string) string
	Now   func() uint64
}

type EstimateFunc func(messages []msgmodel.WithParts, model Model) (float64, error)

type EventSink interface {
	CompactionStarted(sessionID string, timestamp uint64, reason string)
	CompactionEnded(sessionID string, timestamp uint64, text string, include *string)
	PublishCompacted(ctx context.Context, sessionID string) error
}

type EvidenceSelector interface {
	SelectEvidence(ctx context.Context, blocks []string) (*string, error)
}

type EvidenceSelectorFunc func(ctx context.Context, blocks []string) (*string, error)

type FallbackEvidenceSelector struct {
	MaxChars float64
}

type InstanceContext struct {
	Directory string
	Worktree  string
}

type Model struct {
	Message  msgmodel.Model
	Overflow overflow.Model
}

type ModelProvider interface {
	GetModel(ctx context.Context, providerID, modelID string) (Model, error)
	GetProvider(ctx context.Context, providerID string) (ProviderInfo, error)
}

type ModelRef struct {
	ProviderID string
	ModelID    string
}

type OverflowHistory struct {
	Messages []msgmodel.WithParts
	Replay   *Replay
}

type Plugin interface {
	Compacting(ctx context.Context, sessionID string) (CompactingResult, error)
	TransformMessages(ctx context.Context, messages []msgmodel.WithParts) error
	AutoContinue(ctx context.Context, input AutoContinueInput) (bool, error)
}

type ProcessInput struct {
	ParentID  string
	Messages  []msgmodel.WithParts
	SessionID string
	Auto      bool
	Overflow  *bool
}

type ProcessorFactory interface {
	Create(
		ctx context.Context,
		assistant *msgmodel.Assistant,
		sessionID string,
		model Model,
	) (SummaryProcessor, error)
}

type ProcessorFactoryFunc func(
	ctx context.Context,
	assistant *msgmodel.Assistant,
	sessionID string,
	model Model,
) (SummaryProcessor, error)

type ProviderInfo struct {
	Source  string
	Options any
}

type Replay struct {
	Info  msgmodel.User
	Parts msgmodel.Parts
}

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

type SummaryProcessor interface {
	Process(ctx context.Context, request SummaryRequest) (steploop.Result, error)
	Message() msgmodel.Assistant
}

type SummaryRequest struct {
	User      msgmodel.User
	Agent     Agent
	SessionID string
	Messages  []msgmodel.ModelMessage
	Model     Model
}

type Turn struct {
	Start int    `json:"start"`
	End   int    `json:"end"`
	ID    string `json:"id"`
}