memory

Documentation

Index

• Constants
• func BuildPrompt(text string, opts SummarizeOptions) string
• func EstimateTokens(text string) int
• type LLMSummarizer
  • func (s *LLMSummarizer) Generate(ctx context.Context, prompt string) (rw string, e error)
  • func (s *LLMSummarizer) Summarize(ctx context.Context, text string, opts SummarizeOptions) (string, error)
• type StaticSummarizer
  • func (s *StaticSummarizer) Summarize(_ context.Context, _ string, _ SummarizeOptions) (string, error)
• type SummarizeOptions

Constants

const (
	SummarizePolicyNormal     = "normal"
	SummarizePolicyAggressive = "aggressive"
)

SummarizePolicy constants control how aggressively content is compressed.

Functions

func BuildPrompt

func BuildPrompt(text string, opts SummarizeOptions) string

BuildPrompt constructs the appropriate summarization prompt.

func EstimateTokens

func EstimateTokens(text string) int

EstimateTokens returns a rough token count (~4 chars per token).

Types

type LLMSummarizer

type LLMSummarizer struct {
	// Generate calls the LLM with the given prompt and returns the response text.
	AIChat types.OpenAIChatClient
}

LLMSummarizer generates summaries using an LLM via a callback function.

func (*LLMSummarizer) Generate

func (s *LLMSummarizer) Generate(ctx context.Context, prompt string) (rw string, e error)

func (*LLMSummarizer) Summarize

func (s *LLMSummarizer) Summarize(ctx context.Context, text string, opts SummarizeOptions) (string, error)

Summarize generates a summary with escalation: normal -> aggressive -> deterministic fallback.

type StaticSummarizer

type StaticSummarizer struct {
	Response string
	Err      error
}

StaticSummarizer always returns a fixed response (for testing).

func (*StaticSummarizer) Summarize

func (s *StaticSummarizer) Summarize(_ context.Context, _ string, _ SummarizeOptions) (string, error)

Summarize returns the static response.

type SummarizeOptions

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

SummarizeOptions controls summarization behavior.