context

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

type BudgetProfile struct {
	Name        string
	Description string

	// Token allocation ratios for the remaining budget (after SystemPrompt + UserPrefs)
	// Must sum to 1.0
	ShortTermRatio float64
	LongTermRatio  float64
	RetrievalRatio float64

	// Fixed ratios
	UserPrefsRatio float64 // Default: 0.10
}

type CacheProvider interface {
	Get(ctx context.Context, key string) ([]byte, bool)
	Set(ctx context.Context, key string, value []byte, ttl time.Duration) error
}

type Config struct {
	MaxTurns    int           // Max conversation turns (default: 10)
	MaxEpisodes int           // Max episodic memories (default: 3)
	MaxTokens   int           // Default max tokens (default: 4096)
	CacheTTL    time.Duration // Cache TTL (default: 5 minutes)
}

type ContextBuilder interface {
	// Build constructs the full context from various sources.
	Build(ctx context.Context, req *ContextRequest) (*ContextResult, error)

	// GetStats returns context building statistics.
	GetStats() *ContextStats
}

type ContextPriority int

type ContextRequest struct {
	SessionID        string
	CurrentQuery     string
	AgentType        string
	RetrievalResults []*RetrievalItem
	MaxTokens        int
	UserID           int32
}

type ContextRequestWithMessages struct {
	ContextRequest
	Messages []*Message
}

type ContextResult struct {
	TokenBreakdown      *TokenBreakdown
	SystemPrompt        string
	ConversationContext string
	RetrievalContext    string
	UserPreferences     string
	TotalTokens         int
	BuildTime           time.Duration
}

type ContextSegment struct {
	Content   string
	Source    string
	Priority  ContextPriority
	TokenCost int
}

type ContextSnapshot struct {
	// Messages in the conversation.
	Messages []*Message

	// Retrieval results.
	RetrievalResults []*RetrievalItem

	// System prompt hash.
	SystemPromptHash string

	// User preferences hash.
	UserPrefsHash string

	// Episodic memories hash.
	EpisodicHash string

	// Timestamp of the snapshot.
	Timestamp time.Time

	// Token count at time of snapshot.
	TokenCount int

	// Query that was processed.
	Query string

	// Agent type used.
	AgentType string
}

type ContextStats struct {
	TotalBuilds      int64
	AverageTokens    float64
	CacheHits        int64
	AverageBuildTime time.Duration
}

type Delta struct {
	// NewMessages are messages added since last build.
	NewMessages []*Message

	// ModifiedSections are sections that changed.
	ModifiedSections []string

	// RemovedSections are sections that were removed.
	RemovedSections []string

	// NewRetrievalItems are new retrieval results.
	NewRetrievalItems []*RetrievalItem

	// PreviousHash is the hash of the previous context.
	PreviousHash string

	// CurrentHash is the hash of the current context.
	CurrentHash string

	// Strategy used for this delta.
	Strategy UpdateStrategy
}

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

type DeltaStats struct {
	TotalDeltas          int64
	DeltaHits            int64
	FullRebuilds         int64
	AppendOnlyHits       int64
	ConversationOnlyHits int64
	AverageDeltaMs       int64
	SavedTokens          int64
	HitRate              float64
}

type EpisodicMemory struct {
	Timestamp time.Time
	Summary   string
	AgentType string
	Outcome   string
	ID        int64
}

type EpisodicProvider interface {
	SearchEpisodes(ctx context.Context, userID int32, query string, limit int) ([]*EpisodicMemory, error)
}

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

type IntentResolver struct {
}

type LongTermContext struct {
	Preferences *UserPreferences
	Episodes    []*EpisodicMemory
}

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

type Message struct {
	Timestamp time.Time
	Role      string
	Content   string
}

type MessageProvider interface {
	GetRecentMessages(ctx context.Context, sessionID string, limit int) ([]*Message, error)
}

type PreferenceProvider interface {
	GetPreferences(ctx context.Context, userID int32) (*UserPreferences, error)
}

type PriorityRanker struct{}

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

type RetrievalItem struct {
	ID      string
	Content string
	Source  string
	Score   float32
}

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

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

type TokenBreakdown struct {
	SystemPrompt    int
	ShortTermMemory int
	LongTermMemory  int
	Retrieval       int
	UserPrefs       int
}

type TokenBudget struct {
	Total           int
	SystemPrompt    int
	ShortTermMemory int
	LongTermMemory  int
	Retrieval       int
	UserPrefs       int
}

type UpdateStrategy int

type UserPreferences struct {
	Timezone           string
	CommunicationStyle string
	PreferredTimes     []string
	DefaultDuration    int
}