ctxmgr

type BeforeCompactEvent struct {
	Preparation   *CompactionPreparation
	BranchEntries []session.TreeEntry
	Reason        CompactReason
	WillRetry     bool
}

type BeforeCompactFn func(ctx context.Context, event BeforeCompactEvent) (*BeforeCompactOutcome, error)

type BeforeCompactOutcome struct {
	Cancel     bool
	Compaction *CompactionResult
}

type BeforeTreeEvent struct {
	TargetEntryID      string
	OldLeafID          string
	CommonAncestorID   string
	EntriesToSummarize []session.TreeEntry
	// UserWantsSummary is true when the caller did not supply a summary and the
	// manager would generate one; false when the caller already provided summary text.
	UserWantsSummary bool
}

type BeforeTreeFn func(ctx context.Context, event BeforeTreeEvent) (*BeforeTreeOutcome, error)

type BeforeTreeOutcome struct {
	Cancel  bool
	Summary string
}

type BranchSummaryResult struct {
	Summary       string
	ReadFiles     []string
	ModifiedFiles []string
}

type CompactOpts struct {
	// Force enables overflow-style re-compaction after a prior compaction leaf.
	Force bool
	// Reason classifies why compaction ran (manual, threshold, overflow).
	Reason CompactReason
	// WillRetry reports whether the aborted turn retries after compaction (overflow recovery).
	WillRetry bool
}

type CompactReason string

type CompactReport struct {
	Pruned     bool
	Summarized bool
}

type CompactionPreparation struct {
	FirstKeptEntryID    string
	MessagesToSummarize []msg.AgentMessage
	TurnPrefixMessages  []msg.AgentMessage
	IsSplitTurn         bool
	TokensBefore        int
	PreviousSummary     string
	FileOps             FileOperations
	Settings            Settings
	SystemPrompt        string
	Tools               []llms.Tool
	ThinkingLevel       string
}

type CompactionResult struct {
	Summary          string
	FirstKeptEntryID string
	TokensBefore     int
	ReadFiles        []string
	ModifiedFiles    []string
}

type ContextUsage struct {
	Tokens        int
	ContextWindow int
	Percent       float64
}

type ContextUsageEstimate struct {
	Tokens         int
	UsageTokens    int
	TrailingTokens int
	LastUsageIndex int
}

type CutPointResult struct {
	FirstKeptEntryIndex int
	TurnStartIndex      int
	IsSplitTurn         bool
}

type FileOperations struct {
	Read    map[string]struct{}
	Written map[string]struct{}
	Edited  map[string]struct{}
}

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

type Options struct {
	Model         llms.Model
	ModelName     string
	ContextWindow int
	Settings      Settings
	SystemPrompt  string
	Tools         []llms.Tool
	ThinkingLevel string
	BeforeCompact BeforeCompactFn
	BeforeTree    BeforeTreeFn
}

type PrepareOptions struct {
	// Force allows re-compaction when the session leaf is already a compaction node.
	Force bool
	// ExtraMessages are unpersisted agent messages appended to the summarization set.
	ExtraMessages []msg.AgentMessage
}

type Settings struct {
	Enabled bool
	// PruneToolOutputs rewrites oversized current tool results before summarization.
	PruneToolOutputs bool
	ReserveTokens    int
	KeepRecentTokens int
}

type StatefulAgent interface {
	State() *msg.Context
	ApplyState(fn func(*msg.Context))
}