memory

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

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

type CompactionMode int

type CompactionResult struct {
	LeafSummariesCreated      int
	CondensedSummariesCreated int
	MessagesCompacted         int
	TokensBefore              int
	TokensAfter               int
	Duration                  time.Duration
}

type DescribeResult struct {
	SummaryID       string
	Kind            string
	Depth           int
	Content         string
	EarliestAt      *time.Time
	LatestAt        *time.Time
	DescendantCount int
	ParentIDs       []string
	ChildIDs        []string
}

type Engine interface {
	// Bootstrap reconciles session state on startup.
	Bootstrap(ctx context.Context, sessionID string) error

	// Ingest persists a message and appends to context.
	Ingest(ctx context.Context, sessionID string, msg ai.Message) error

	// IngestBatch persists multiple messages.
	IngestBatch(ctx context.Context, sessionID string, msgs []ai.Message) error

	// Assemble builds context for the model within token budget.
	Assemble(ctx context.Context, sessionID string, budget int, freshTail int) ([]ai.Message, error)

	// Compact runs compaction passes (leaf + optional condensation).
	Compact(ctx context.Context, sessionID string, mode CompactionMode) (*CompactionResult, error)

	// NeedsCompaction checks if compaction should run based on token threshold.
	NeedsCompaction(ctx context.Context, sessionID string, threshold float64) bool

	// SaveInfo persists session metadata (upsert).
	SaveInfo(ctx context.Context, info SessionInfo) error

	// LoadInfo retrieves session metadata by session ID.
	LoadInfo(ctx context.Context, sessionID string) (SessionInfo, error)

	// ListInfo lists session metadata. If includeArchived is false, archived sessions are excluded.
	ListInfo(ctx context.Context, includeArchived bool) ([]SessionInfo, error)

	// Load returns the full event history for a session.
	Load(ctx context.Context, sessionID string) ([]ai.Message, error)

	// Retrieval returns the retrieval engine for tools.
	Retrieval() *RetrievalEngine

	// Close releases database resources.
	Close() error
}

type EngineOption func(*engine)

type ExpandChild struct {
	SummaryID string
	Kind      string
	Depth     int
	Content   string
}

type ExpandMessage struct {
	MessageID int64
	Role      string
	Content   string
	CreatedAt time.Time
}

type ExpandResult struct {
	SummaryID string
	Children  []ExpandChild
	Messages  []ExpandMessage
}

type GrepResult struct {
	SourceType string // "message" or "summary"
	SourceID   string
	Content    string
	Relevance  float64 // reserved for future scoring; currently 0
	Timestamp  time.Time
}

type LLMSummarizer struct {
	// Generate calls the LLM with the given prompt and returns the response text.
	Generate func(ctx context.Context, prompt string) (string, error)
}

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

type SessionInfo struct {
	ID         string
	Channel    string
	Title      string
	CreatedAt  time.Time
	LastActive time.Time
	Archived   bool
}

type StaticSummarizer struct {
	Response string
	Err      error
}

type SummarizeOptions struct {
	IsCondensed  bool
	Depth        int
	Aggressive   bool
	Previous     string // previous summary for continuity
	TargetTokens int
}

type Summarizer interface {
	Summarize(ctx context.Context, text string, opts SummarizeOptions) (string, error)
}