context

type CompressionEvent struct {
	Timestamp        time.Time
	BeforeMessages   int
	AfterMessages    int
	BeforeTokens     int
	AfterTokens      int
	CompressionRatio float64
	Strategy         string
}

type CompressionStrategy interface {
	// Name 返回策略名称
	Name() string

	// Compress 压缩消息列表
	// 返回压缩后的消息列表
	Compress(ctx context.Context, messages []Message, config WindowManagerConfig) ([]Message, error)
}

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

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

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

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

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

type InjectorConfig struct {
	EnableDateTime     bool   // 注入日期时间
	EnableLocation     bool   // 注入位置信息
	EnableSessionState bool   // 注入 Session 状态
	EnableUserInfo     bool   // 注入用户信息
	EnableSystemInfo   bool   // 注入系统信息
	CustomContext      string // 自定义上下文

	// 可选的数据提供者
	LocationProvider     LocationProvider
	SessionStateProvider SessionStateProvider
	UserInfoProvider     UserInfoProvider
}

type LocationInfo struct {
	City      string
	Country   string
	Timezone  string
	Language  string
	Latitude  float64
	Longitude float64
}

type LocationProvider interface {
	GetLocation(ctx context.Context) (*LocationInfo, error)
}

type Message struct {
	Role    string // "system", "user", "assistant"
	Content string
}

type MessagePriority int

type MessageWithPriority struct {
	Message  Message
	Priority MessagePriority
	Score    float64 // 优先级分数（用于排序）
}

type ModelConfig struct {
	Name              string  // 模型名称
	CharsPerToken     float64 // 平均每个 Token 的字符数
	BaseTokenOverhead int     // 每条消息的基础 Token 开销
	RoleTokenCost     int     // Role 标签的 Token 开销
}

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

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

type PriorityCalculator interface {
	// CalculatePriority 计算消息的优先级和分数
	CalculatePriority(ctx context.Context, msg Message, position int, totalMessages int) (MessagePriority, float64)
}

type SessionState struct {
	SessionID    string
	MessageCount int
	Duration     time.Duration
	CustomState  map[string]any
}

type SessionStateProvider interface {
	GetSessionState(ctx context.Context) (*SessionState, error)
}

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

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

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

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

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

type TikTokenCounter struct {
}

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

type TokenBudget struct {
	MaxTokens        int     // 最大 Token 数
	ReservedTokens   int     // 预留 Token 数（用于输出）
	WarningThreshold float64 // 警告阈值（0.0-1.0）
}

type TokenCounter interface {
	// Count 计算单个文本的 Token 数量
	Count(ctx context.Context, text string) (int, error)

	// CountBatch 批量计算多个文本的 Token 数量
	CountBatch(ctx context.Context, texts []string) ([]int, error)

	// EstimateMessages 估算消息列表的总 Token 数
	EstimateMessages(ctx context.Context, messages []Message) (int, error)

	// ModelName 返回此计数器对应的模型名称
	ModelName() string
}

type TokenEstimate struct {
	TotalTokens   int            // 总 Token 数
	MessageTokens []int          // 每条消息的 Token 数
	Breakdown     map[string]int // 详细分解（content, overhead, role 等）
}

type UserInfo struct {
	UserID            string
	Name              string
	PreferredLanguage string
	Timezone          string
	Preferences       map[string]any
}

type UserInfoProvider interface {
	GetUserInfo(ctx context.Context) (*UserInfo, error)
}

type WindowManagerConfig struct {
	// Token 预算
	Budget TokenBudget

	// 自动压缩配置
	AutoCompress         bool    // 是否启用自动压缩
	CompressionThreshold float64 // 触发压缩的使用率阈值 (0.0-1.0)

	// 消息保留策略
	MinMessagesToKeep int  // 最少保留的消息数
	AlwaysKeepSystem  bool // 始终保留 system 消息
	AlwaysKeepRecent  int  // 始终保留最近的 N 条消息

	// 优先级配置
	EnablePrioritization bool // 是否启用消息优先级
}

type WindowStats struct {
	CurrentMessages  int     // 当前消息数
	TotalMessages    int     // 总共添加的消息数
	CurrentTokens    int     // 当前 Token 使用量
	RemainingTokens  int     // 剩余可用 Token
	UsagePercentage  float64 // 使用百分比
	CompressionCount int     // 压缩次数
	IsWithinBudget   bool    // 是否在预算内
	ShouldWarn       bool    // 是否应该警告
}