memory

type BatchVectorStore interface {
	rag.LowLevelVectorStore
	BatchStore(ctx context.Context, items []VectorItem) error
}

type CompletionFunc func(ctx context.Context, systemPrompt, userPrompt string) (string, error)

type ConsolidationStrategy interface {
	// 判断是否应该整合
	ShouldConsolidate(ctx context.Context, memory any) bool

	// 执行整合
	Consolidate(ctx context.Context, memories []any) error
}

type DBClient interface {
	Exec(ctx context.Context, query string, args ...any) error
	Query(ctx context.Context, query string, args ...any) (DBRows, error)
}

type DBRows interface {
	Next() bool
	Scan(dest ...any) error
	Close() error
}

type DecayConfig struct {
	RecencyWeight   float64       `json:"recency_weight"`   // Weight for recency score (0-1)
	RelevanceWeight float64       `json:"relevance_weight"` // Weight for relevance score (0-1)
	UtilityWeight   float64       `json:"utility_weight"`   // Weight for utility score (0-1)
	DecayThreshold  float64       `json:"decay_threshold"`  // Score below which memories are pruned
	DecayInterval   time.Duration `json:"decay_interval"`   // How often to run decay
	MaxMemories     int           `json:"max_memories"`     // Maximum memories to retain
	HalfLife        time.Duration `json:"half_life"`        // Time for recency score to halve
}

type DecayResult struct {
	Timestamp   time.Time `json:"timestamp"`
	TotalBefore int       `json:"total_before"`
	TotalAfter  int       `json:"total_after"`
	PrunedCount int       `json:"pruned_count"`
	PrunedIDs   []string  `json:"pruned_ids,omitempty"`
}

type DecayStats struct {
	TotalMemories    int     `json:"total_memories"`
	AverageScore     float64 `json:"average_score"`
	AverageRecency   float64 `json:"average_recency"`
	AverageRelevance float64 `json:"average_relevance"`
	AverageUtility   float64 `json:"average_utility"`
}

type Embedder interface {
	Embed(ctx context.Context, text string) ([]float32, error)
}

type EnhancedMemoryConfig struct {
	// 短期记忆配置
	ShortTermTTL     time.Duration `json:"short_term_ttl"`      // 短期记忆 TTL
	ShortTermMaxSize int           `json:"short_term_max_size"` // 最大条目数

	// 工作记忆配置
	WorkingMemorySize int `json:"working_memory_size"` // 工作记忆容量

	// 长期记忆配置
	LongTermEnabled bool `json:"long_term_enabled"` // 是否启用长期记忆
	VectorDimension int  `json:"vector_dimension"`  // 向量维度

	// 情节记忆配置
	EpisodicEnabled   bool          `json:"episodic_enabled"`   // 是否启用情节记忆
	EpisodicRetention time.Duration `json:"episodic_retention"` // 情节记忆保留时间

	// 语义记忆配置
	SemanticEnabled bool `json:"semantic_enabled"` // 是否启用语义记忆

	// 观测记忆配置
	ObservationEnabled bool           `json:"observation_enabled"` // 是否启用观测记忆
	ObserverConfig     ObserverConfig `json:"observer_config"`     // Observer 配置

	// 记忆整合配置
	ConsolidationEnabled  bool          `json:"consolidation_enabled"`  // 是否启用记忆整合
	ConsolidationInterval time.Duration `json:"consolidation_interval"` // 整合间隔
}

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

type Entity struct {
	ID         string         `json:"id"`
	Type       string         `json:"type"`
	Name       string         `json:"name"`
	Properties map[string]any `json:"properties"`
	CreatedAt  time.Time      `json:"created_at"`
	UpdatedAt  time.Time      `json:"updated_at"`
}

type Episode struct {
	ID           string         `json:"id"`
	Timestamp    time.Time      `json:"timestamp"`
	Context      string         `json:"context"`
	Action       string         `json:"action"`
	Result       string         `json:"result"`
	Emotion      string         `json:"emotion,omitempty"`
	Importance   float64        `json:"importance"`
	Participants []string       `json:"participants,omitempty"`
	Metadata     map[string]any `json:"metadata,omitempty"`
}

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

type EpisodicQuery struct {
	AgentID   string
	Type      string
	StartTime time.Time
	EndTime   time.Time
	Limit     int
}

type EpisodicStore interface {
	// 记录事件
	RecordEvent(ctx context.Context, event *types.EpisodicEvent) error

	// 查询事件
	QueryEvents(ctx context.Context, query EpisodicQuery) ([]types.EpisodicEvent, error)

	// 获取时间线
	GetTimeline(ctx context.Context, agentID string, start, end time.Time) ([]types.EpisodicEvent, error)
}

type Fact struct {
	ID         string    `json:"id"`
	Subject    string    `json:"subject"`
	Predicate  string    `json:"predicate"`
	Object     string    `json:"object"`
	Confidence float64   `json:"confidence"`
	Source     string    `json:"source,omitempty"`
	Embedding  []float32 `json:"embedding,omitempty"`
	CreatedAt  time.Time `json:"created_at"`
	UpdatedAt  time.Time `json:"updated_at"`
}

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

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

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

type InMemoryMemoryStoreConfig struct {
	// Max Entries是本商店条目的全球封顶.
	// 0表示无限.
	MaxEntries int

	// 现在用于测试。 默认时间 。 现在。
	Now func() time.Time
}

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

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

type InMemoryVectorStoreConfig struct {
	// 尺寸在 > 0时验证存储/搜索向量。
	Dimension int

	// MaxEntries limits the number of stored vectors. 0 means unlimited.
	// When the limit is reached, the oldest entry is evicted.
	MaxEntries int

	// 现在用于测试。 默认时间 。 现在。
	Now func() time.Time
}

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

type KnowledgeGraph interface {
	// 添加实体
	AddEntity(ctx context.Context, entity *Entity) error

	// 添加关系
	AddRelation(ctx context.Context, relation *Relation) error

	// 查询实体
	QueryEntity(ctx context.Context, id string) (*Entity, error)

	// 查询关系
	QueryRelations(ctx context.Context, entityID string, relationType string) ([]Relation, error)

	// 路径查询
	FindPath(ctx context.Context, fromID, toID string, maxDepth int) ([][]string, error)
}

type LayeredMemory struct {
	Episodic   *EpisodicMemory
	Semantic   *SemanticMemory
	Working    *WorkingMemory
	Procedural *ProceduralMemory
	// contains filtered or unexported fields
}

type LayeredMemoryConfig struct {
	EpisodicMaxSize  int
	WorkingCapacity  int
	WorkingTTL       time.Duration
	Embedder         Embedder
	ProceduralConfig ProceduralConfig
}

type LongTermRetriever interface {
	Retrieve(ctx context.Context, query string, queryEmbedding []float64) ([]rag.RetrievalResult, error)
}

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

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

type MemoryEntry struct {
	ID          string                `json:"id"`
	Type        memorycore.MemoryKind `json:"type"`
	Content     string                `json:"content"`
	Embedding   []float32             `json:"embedding,omitempty"`
	Importance  float64               `json:"importance"`
	AccessCount int                   `json:"access_count"`
	CreatedAt   time.Time             `json:"created_at"`
	LastAccess  time.Time             `json:"last_access"`
	ExpiresAt   *time.Time            `json:"expires_at,omitempty"`
	Metadata    map[string]any        `json:"metadata,omitempty"`
	Relations   []string              `json:"relations,omitempty"`
}

type MemoryItem struct {
	ID           string         `json:"id"`
	Content      string         `json:"content"`
	Vector       []float64      `json:"vector,omitempty"`
	CreatedAt    time.Time      `json:"created_at"`
	LastAccessed time.Time      `json:"last_accessed"`
	AccessCount  int            `json:"access_count"`
	Relevance    float64        `json:"relevance"` // User-defined relevance (0-1)
	Utility      float64        `json:"utility"`   // Computed utility based on usage
	Tags         []string       `json:"tags,omitempty"`
	Metadata     map[string]any `json:"metadata,omitempty"`
}

type MemoryStore interface {
	Save(ctx context.Context, key string, value any, ttl time.Duration) error
	Load(ctx context.Context, key string) (any, error)
	Delete(ctx context.Context, key string) error
	List(ctx context.Context, pattern string, limit int) ([]any, error)
	Clear(ctx context.Context) error
}

type Observation struct {
	ID        string         `json:"id"`
	AgentID   string         `json:"agent_id"`
	Date      string         `json:"date"`
	Content   string         `json:"content"`
	CreatedAt time.Time      `json:"created_at"`
	Metadata  map[string]any `json:"metadata,omitempty"`
}

type ObservationStore interface {
	Save(ctx context.Context, obs Observation) error
	LoadRecent(ctx context.Context, agentID string, limit int) ([]Observation, error)
	LoadByDateRange(ctx context.Context, agentID string, start, end time.Time) ([]Observation, error)
}

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

type ObserverConfig struct {
	MaxMessagesPerBatch       int           `json:"max_messages_per_batch"`
	MinMessagesForObservation int           `json:"min_messages_for_observation"`
	ObservationInterval       time.Duration `json:"observation_interval"`
}

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

type ProceduralConfig struct {
	MaxProcedures int `json:"max_procedures"`
}

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

type Procedure struct {
	ID          string   `json:"id"`
	Name        string   `json:"name"`
	Description string   `json:"description"`
	Steps       []string `json:"steps"`
	Triggers    []string `json:"triggers"`
	SuccessRate float64  `json:"success_rate"`
	Executions  int      `json:"executions"`
}

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

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

type Relation struct {
	ID         string         `json:"id"`
	FromID     string         `json:"from_id"`
	ToID       string         `json:"to_id"`
	Type       string         `json:"type"`
	Properties map[string]any `json:"properties"`
	Weight     float64        `json:"weight"`
	CreatedAt  time.Time      `json:"created_at"`
}

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

type VectorItem struct {
	ID       string
	Vector   []float64
	Metadata map[string]any
}

type WorkingItem struct {
	Key       string    `json:"key"`
	Value     any       `json:"value"`
	Priority  int       `json:"priority"`
	CreatedAt time.Time `json:"created_at"`
	ExpiresAt time.Time `json:"expires_at"`
}

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