preload

type Activity struct {
	// Timestamp when the activity occurred.
	Timestamp time.Time

	// Query is the search query (if any).
	Query string

	// AgentType used for the query.
	AgentType string

	// SessionDuration is the duration of the session.
	SessionDuration time.Duration

	// TokensUsed is the number of tokens consumed.
	TokensUsed int

	// Topics are topics related to this activity.
	Topics []string
}

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

type AnalyzerStats struct {
	CachedPatterns int
	MaxCacheSize   int
	TotalSamples   int64
	TotalQueries   int
}

type Config struct {
	// Store is the pattern store backend.
	Store PatternStore

	// MaxCacheSize is the maximum patterns to keep in memory.
	MaxCacheSize int

	// TTL is how long cached patterns remain valid.
	TTL time.Duration

	// MinSamples is the minIntimum samples before a pattern is considered valid.
	MinSamples int
}

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

type MockPreloader struct {
	Calls []struct {
		UserID int64
		Query  string
	}
	// contains filtered or unexported fields
}

type PatternStore interface {
	// SavePattern saves a user pattern.
	SavePattern(ctx context.Context, userID int64, pattern *UserPattern) error

	// LoadPattern loads a user pattern.
	LoadPattern(ctx context.Context, userID int64) (*UserPattern, error)

	// UpdatePattern updates an existing pattern.
	UpdatePattern(ctx context.Context, userID int64, pattern *UserPattern) error
}

type PreloadSuggestions struct {
	UserID         int64
	Timestamp      time.Time
	Queries        []string
	IsPeakHour     bool
	ExpectedTokens int
}

type Preloader interface {
	// PreloadQuery preloads data for a query.
	PreloadQuery(ctx context.Context, userID int64, query string) error

	// PreloadUserContext preloads user context data.
	PreloadUserContext(ctx context.Context, userID int64) error
}

type QueryPattern struct {
	// Query is the search query (may be normalized).
	Query string

	// Frequency is how often this query is used.
	Frequency int

	// LastUsed is when this query was last used.
	LastUsed time.Time

	// AvgTokens is the average tokens used for this query.
	AvgTokens int
}

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

type SchedulerConfig struct {
	// AnalysisInterval is how often to analyze patterns.
	AnalysisInterval time.Duration

	// PreloadInterval is how often to perform preloading.
	PreloadInterval time.Duration

	// QuietHours are hours when preloading is paused.
	QuietHours []int // 0-23

	// MaxCPUPercent limits CPU usage for preloading.
	MaxCPUPercent int

	// MaxMemoryMB limits memory usage for preloading.
	MaxMemoryMB int

	// ConcurrentPreloads is the max concurrent preload operations.
	ConcurrentPreloads int
}

type SchedulerStats struct {
	ScheduledUsers int
	SuccessCount   int64
	ErrorCount     int64
	Running        bool
	QuietHours     []int
	MaxCPUPercent  int
	MaxMemoryMB    int
}

type UserPattern struct {
	// UserID is the user's ID.
	UserID int64

	// ActiveHours shows when the user is most active.
	// Each bucket represents an hour of the day (0-23).
	ActiveHours [24]int

	// ActiveDays shows which days of the week the user is active.
	// Each bucket represents a day (0=Sunday, 6=Saturday).
	ActiveDays [7]int

	// FrequentQueries are the user's most common search queries.
	FrequentQueries []*QueryPattern

	// CommonTopics are topics the user frequently searches for.
	CommonTopics []string

	// LastUpdate is when this pattern was last updated.
	LastUpdate time.Time

	// SampleCount is the number of data points used to build this pattern.
	SampleCount int64

	// AvgSessionDuration is the user's average session duration.
	AvgSessionDuration time.Duration

	// PeakHours are the user's peak activity hours.
	PeakHours []int

	// QuietHours are hours when the user is rarely active.
	QuietHours []int
}