logic

type ConsolidationConfig struct {
	// SimilarityThreshold 相似度阈值（0.0-1.0）
	// 超过此阈值的 Memory 将被合并
	SimilarityThreshold float64

	// MinGroupSize 最小合并组大小
	// 只有当相似 Memory 数量 >= 此值时才进行合并
	MinGroupSize int

	// MaxMergeCount 单次合并的最大数量
	// 限制单次操作的 Memory 数量，避免长时间阻塞
	MaxMergeCount int

	// PreserveHighConfidence 保留高置信度的 Memory
	// 高置信度的 Memory 不会被合并到其他 Memory
	PreserveHighConfidenceThreshold float64

	// MergeStrategy 合并策略
	MergeStrategy MergeStrategy
}

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

type ConsolidationResult struct {
	// TotalMemories 处理的 Memory 总数
	TotalMemories int

	// MergedGroups 合并的组数
	MergedGroups int

	// DeletedMemories 删除的 Memory 数量
	DeletedMemories int

	// MergedMemories 合并后的 Memory 列表
	MergedMemories []*LogicMemory

	// StartTime 开始时间
	StartTime time.Time

	// EndTime 结束时间
	EndTime time.Time
}

type DefaultSimilarityCalculator struct{}

type Event struct {
	// Type 事件类型（如 "user_message", "tool_result", "user_feedback" 等）
	Type string `json:"type"`

	// Source 来源（agent_id, user_id 等）
	Source string `json:"source"`

	// Data 事件数据（结构化）
	Data map[string]any `json:"data"`

	// Timestamp 时间戳
	Timestamp time.Time `json:"timestamp"`
}

type Filter func(*FilterOptions)

type FilterOptions struct {
	// Type 过滤类型
	Type string

	// Scope 过滤作用域
	Scope MemoryScope

	// MinConfidence 最低置信度
	MinConfidence float64

	// MaxResults TopK 限制
	MaxResults int

	// OrderBy 排序字段
	OrderBy OrderBy

	// SinceLastAccess 最后访问时间过滤
	SinceLastAccess time.Duration
}

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

type LogicMemory struct {

	// ID 唯一标识符
	ID string `json:"id"`

	// Namespace 租户隔离（如 user:123, team:456, global）
	Namespace string `json:"namespace"`

	// Scope 作用域（Session/User/Global）
	Scope MemoryScope `json:"scope"`

	// Type Memory 类型（如 "user_preference", "behavior_pattern" 等）
	Type string `json:"type"`

	// Category 分类（可选，用于进一步分组）
	Category string `json:"category,omitempty"`

	// Key 唯一标识（如 "writing_tone_preference"）
	// Namespace + Key 构成全局唯一键
	Key string `json:"key"`

	// Value 值（结构化数据）
	// 应用层可以存储任意 JSON 可序列化的数据
	Value any `json:"value"`

	// Description 人类可读描述（用于 Prompt 注入）
	// 例如："用户偏好口语化表达，避免使用书面语"
	Description string `json:"description"`

	// Provenance 记忆溯源信息
	Provenance *memory.MemoryProvenance `json:"provenance"`

	// AccessCount 访问次数（用于 LRU 淘汰）
	AccessCount int `json:"access_count"`

	// LastAccessed 最后访问时间
	LastAccessed time.Time `json:"last_accessed"`

	// Metadata 扩展字段（应用层自定义）
	Metadata map[string]any `json:"metadata,omitempty"`

	// CreatedAt 创建时间
	CreatedAt time.Time `json:"created_at"`

	// UpdatedAt 更新时间
	UpdatedAt time.Time `json:"updated_at"`
}

type LogicMemoryStore interface {

	// Save 保存或更新 Memory
	// 如果 Namespace + Key 已存在，则更新；否则创建新记录
	Save(ctx context.Context, memory *LogicMemory) error

	// Get 获取单个 Memory
	// 返回 ErrMemoryNotFound 如果不存在
	Get(ctx context.Context, namespace, key string) (*LogicMemory, error)

	// Delete 删除 Memory
	// 如果不存在不返回错误
	Delete(ctx context.Context, namespace, key string) error

	// List 列出符合条件的 Memory
	// filters 用于过滤和排序
	List(ctx context.Context, namespace string, filters ...Filter) ([]*LogicMemory, error)

	// SearchByType 按类型搜索
	SearchByType(ctx context.Context, namespace, memoryType string) ([]*LogicMemory, error)

	// SearchByScope 按作用域搜索
	SearchByScope(ctx context.Context, namespace string, scope MemoryScope) ([]*LogicMemory, error)

	// GetTopK 获取 TopK Memory
	// 按 orderBy 排序后返回前 k 个
	GetTopK(ctx context.Context, namespace string, k int, orderBy OrderBy) ([]*LogicMemory, error)

	// IncrementAccessCount 增加访问计数
	// 同时更新 LastAccessed 时间
	IncrementAccessCount(ctx context.Context, namespace, key string) error

	// GetStats 获取统计信息
	GetStats(ctx context.Context, namespace string) (*MemoryStats, error)

	// Prune 清理低价值 Memory
	// 返回清理的数量
	Prune(ctx context.Context, criteria PruneCriteria) (int, error)

	// Close 关闭连接
	Close() error
}

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

type ManagerConfig struct {
	// Store 存储后端（必需）
	Store LogicMemoryStore

	// Matchers 模式匹配器列表（可选，至少提供一个才能自动捕获）
	Matchers []PatternMatcher

	// DefaultProvenance 默认溯源信息（可选）
	DefaultProvenance *memory.MemoryProvenance

	// AutoConsolidate 是否自动合并相似 Memory（默认 false）
	AutoConsolidate bool

	// ConsolidationThreshold 相似度阈值（用于自动合并，默认 0.85）
	ConsolidationThreshold float64

	// ConfidenceBoost 每次重复出现的置信度提升（默认 0.05）
	ConfidenceBoost float64
}

type MemoryScope string

type MemoryStats struct {
	// TotalCount 总记忆数
	TotalCount int

	// CountByType 按类型统计
	CountByType map[string]int

	// CountByScope 按作用域统计
	CountByScope map[MemoryScope]int

	// AverageConfidence 平均置信度
	AverageConfidence float64

	// LastUpdated 最后更新时间
	LastUpdated time.Time
}

type MergeStrategy string

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

type MetricsSnapshot struct {
	Timestamp time.Time

	// 计数器
	MemorySaveTotal    int64
	MemorySaveErrors   int64
	MemoryGetTotal     int64
	MemoryGetErrors    int64
	MemoryDeleteTotal  int64
	EventProcessTotal  int64
	EventProcessErrors int64
	ConsolidationTotal int64
	PruneTotal         int64

	// 分布
	MemoryByNamespace map[string]int64
	MemoryByType      map[string]int64
	MemoryByScope     map[MemoryScope]int64

	// 耗时统计
	AvgSaveDuration         time.Duration
	AvgGetDuration          time.Duration
	AvgEventProcessDuration time.Duration
	P99SaveDuration         time.Duration
	P99GetDuration          time.Duration
	P99EventProcessDuration time.Duration
}

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

type MySQLStoreConfig struct {
	// DB 数据库连接（必需）
	DB *sql.DB

	// TableName 表名（默认 "logic_memories"）
	TableName string

	// AutoMigrate 是否自动创建表（默认 true）
	AutoMigrate bool
}

type NoopMatcher struct{}

type OrderBy string

type PatternMatcher interface {
	// MatchEvent 从事件中识别 Memory
	// 返回识别出的 Memory 列表（可能为空）
	MatchEvent(ctx context.Context, event Event) ([]*LogicMemory, error)

	// SupportedEventTypes 返回支持的事件类型列表
	// 用于 Manager 筛选事件，避免不必要的调用
	SupportedEventTypes() []string
}

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

type PostgreSQLStoreConfig struct {
	// DB 数据库连接（必需）
	DB *sql.DB

	// TableName 表名（默认 "logic_memories"）
	TableName string

	// AutoMigrate 是否自动创建表（默认 true）
	AutoMigrate bool
}

type PrometheusExporter interface {
	// ExportGauge 导出 Gauge 指标
	ExportGauge(name string, value float64, labels map[string]string)

	// ExportCounter 导出 Counter 指标
	ExportCounter(name string, value float64, labels map[string]string)

	// ExportHistogram 导出 Histogram 指标
	ExportHistogram(name string, value float64, labels map[string]string)
}

type PruneCriteria struct {
	// MinConfidence 最低置信度（低于此值将被清理）
	MinConfidence float64

	// MaxAge 最大年龄（超过此时长将被清理）
	MaxAge time.Duration

	// MinAccessCount 最少访问次数（低于此值将被清理）
	MinAccessCount int

	// SinceLastAccess 最后访问时间（超过此时长未访问将被清理）
	SinceLastAccess time.Duration
}

type SimilarityCalculator interface {
	// Calculate 计算两个 Memory 的相似度（0.0-1.0）
	Calculate(a, b *LogicMemory) float64
}

type StoreConfig struct {
	// Type 存储类型（"postgres", "redis", "inmemory"）
	Type string

	// ConnectionString 连接字符串（PostgreSQL/Redis）
	ConnectionString string

	// TableName 表名（PostgreSQL，可选，默认 "logic_memories"）
	TableName string

	// KeyPrefix 键前缀（Redis，可选）
	KeyPrefix string

	// MaxConnections 最大连接数（可选）
	MaxConnections int
}

type StoreError struct {
	Code    string
	Message string
	Err     error
}