contextwindow

type CompressionPolicy struct {
	Enabled      bool    `json:"enabled" yaml:"enabled"`
	TriggerRatio float64 `json:"trigger_ratio,omitempty" yaml:"trigger_ratio,omitempty"`
}

type InsertPosition string

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

type Message struct {
	Role       Role   `json:"role"`
	Content    string `json:"content,omitempty"`
	Name       string `json:"name,omitempty"`
	ToolCallID string `json:"tool_call_id,omitempty"`
	// ToolCallIDs lists tool_call ids issued by an assistant turn. Used for
	// tool-pair-safe trimming (assistant + matching tool results stay atomic).
	ToolCallIDs []string          `json:"tool_call_ids,omitempty"`
	Metadata    map[string]string `json:"metadata,omitempty"`
}

type NormalizeResult struct {
	Policy       Policy
	PolicySource string // profile | context_policy | default_8192
	Fallback8192 bool
}

type Policy struct {
	Strategy             Strategy `json:"strategy,omitempty" yaml:"strategy,omitempty"`
	ContextWindowTokens  int      `json:"context_window_tokens,omitempty" yaml:"context_window_tokens,omitempty"`
	MaxInputTokens       int      `json:"max_input_tokens,omitempty" yaml:"max_input_tokens,omitempty"`
	ReservedOutputTokens int      `json:"reserved_output_tokens,omitempty" yaml:"reserved_output_tokens,omitempty"`
	SummaryTokens        int      `json:"summary_tokens,omitempty" yaml:"summary_tokens,omitempty"`
	ToolResultMaxTokens  int      `json:"tool_result_max_tokens,omitempty" yaml:"tool_result_max_tokens,omitempty"`
	// ToolOutputMaxBytes caps tool message bytes written to session memory.
	// Zero disables the write-side byte cap (LLM-side ToolResultMaxTokens still applies).
	ToolOutputMaxBytes     int               `json:"tool_output_max_bytes,omitempty" yaml:"tool_output_max_bytes,omitempty"`
	MemoryRecallLimit      int               `json:"memory_recall_limit,omitempty" yaml:"memory_recall_limit,omitempty"`
	SystemPromptProtection bool              `json:"system_prompt_protection,omitempty" yaml:"system_prompt_protection,omitempty"`
	Compression            CompressionPolicy `json:"compression,omitempty" yaml:"compression,omitempty"`
	RoleBudgets            RoleBudgets       `json:"role_budgets,omitempty" yaml:"role_budgets,omitempty"`
	SummaryMode            string            `json:"summary_mode,omitempty" yaml:"summary_mode,omitempty"`
	ToolSchemaPruning      bool              `json:"tool_schema_pruning,omitempty" yaml:"tool_schema_pruning,omitempty"`
	StaleToolTurns         int               `json:"stale_tool_turns,omitempty" yaml:"stale_tool_turns,omitempty"`
	// ExcludeFromStaleWindow lists tool result classes that do not consume a
	// StaleToolTurns slot. When nil, denied and empty are excluded by default.
	ExcludeFromStaleWindow []ToolResultClass `json:"exclude_from_stale_window,omitempty" yaml:"exclude_from_stale_window,omitempty"`
	PinUserMessages        *bool             `json:"pin_user_messages,omitempty" yaml:"pin_user_messages,omitempty"`
	// MemoryStoreLimit caps how many messages are retained in the persistent
	// flat memory store. When exceeded, the oldest messages are dropped on
	// write (keeping the most recent MemoryStoreLimit). Zero (the default)
	// disables the write-side cap, preserving unbounded append behavior.
	MemoryStoreLimit int `json:"memory_store_limit,omitempty" yaml:"memory_store_limit,omitempty"`
	// StripAssistantPatterns is an optional list of substrings; when non-empty,
	// matching assistant messages may be stripped on memory write by integrators
	// that opt in. The framework does not strip by default.
	StripAssistantPatterns []string `json:"strip_assistant_patterns,omitempty" yaml:"strip_assistant_patterns,omitempty"`
	// InjectCompactReminder asks the runtime to reinject a system reminder with
	// active plan/TODO state after compaction drops or summarizes history.
	// Placement is before the last user message (see InsertBeforeLastUserMessage).
	InjectCompactReminder bool `json:"inject_compact_reminder,omitempty" yaml:"inject_compact_reminder,omitempty"`
	// ObservationMaskAfterTurns replaces tool-role messages older than this many
	// assistant turns with a short placeholder before summarization/truncation.
	ObservationMaskAfterTurns int `json:"observation_mask_after_turns,omitempty" yaml:"observation_mask_after_turns,omitempty"`
}

type Result struct {
	Messages []Message `json:"messages"`
	Stats    Stats     `json:"stats"`
}

type Role string

type RoleBudgets struct {
	System    int `json:"system,omitempty" yaml:"system,omitempty"`
	User      int `json:"user,omitempty" yaml:"user,omitempty"`
	Assistant int `json:"assistant,omitempty" yaml:"assistant,omitempty"`
	Tool      int `json:"tool,omitempty" yaml:"tool,omitempty"`
}

type Stats struct {
	Strategy                 Strategy `json:"strategy"`
	BeforeTokens             int      `json:"before_tokens"`
	AfterTokens              int      `json:"after_tokens"`
	MaxInputTokens           int      `json:"max_input_tokens"`
	DroppedMessages          int      `json:"dropped_messages"`
	DroppedUserMessages      int      `json:"dropped_user_messages,omitempty"`
	DroppedAssistantMessages int      `json:"dropped_assistant_messages,omitempty"`
	DroppedToolMessages      int      `json:"dropped_tool_messages,omitempty"`
	ContextIncomplete        bool     `json:"context_incomplete,omitempty"`
	Summarized               bool     `json:"summarized"`
	SummaryTokens            int      `json:"summary_tokens,omitempty"`
	PolicySource             string   `json:"policy_source,omitempty"`
	FallbackApplied          bool     `json:"fallback_applied,omitempty"`
	StaleDroppedToolTurns    int      `json:"stale_dropped_tool_turns,omitempty"`
	DenialOccupiedSlots      int      `json:"denial_occupied_slots,omitempty"`
	StaleExcludedTurns       int      `json:"stale_excluded_turns,omitempty"`
	CompactedToolDenials     int      `json:"compacted_tool_denials,omitempty"`
	// NeedsReminder is true when compaction dropped or summarized history and
	// hosts should re-inject active plan/TODO state.
	NeedsReminder bool `json:"needs_reminder,omitempty"`
}

type Strategy string

type Summarizer func(messages []Message, budget int) string

type ToolOutputTransform func(tool string, raw []byte, limit int) (out []byte, meta TransformMeta)

type ToolResultClass string

type TransformMeta struct {
	Truncated      bool   `json:"truncated"`
	OriginalBytes  int    `json:"original_bytes"`
	TruncatedBytes int    `json:"truncated_bytes"`
	Strategy       string `json:"strategy"` // integrator | json_aware | byte_cut | none
}