Alice 记忆系统：AI 长期记忆开源方案

┌─────────────────────────────────────────┐
│           工作记忆 (Working)             │  ← 当前对话上下文
│         大小: 100K-200K tokens          │
├─────────────────────────────────────────┤
│           短期记忆 (Short-term)          │  ← 近期事实、临时偏好
│         保留: 几天到几周                │
├─────────────────────────────────────────┤
│           长期记忆 (Long-term)           │  ← 核心事实、关系、历史
│         保留: 永久                      │
├─────────────────────────────────────────┤
│           程序记忆 (Procedural)          │  ← 行为模式、技能、规则
│         更新: 版本迭代                  │
└─────────────────────────────────────────┘

对话输入
    ↓
重要性评估 ──→ 低 ──→ 丢弃
    ↓ 高
提取关键信息
    ↓
┌─────────────┐
│  工作记忆   │ ← 立即使用
└─────────────┘
    ↓
向量化 + 结构化
    ↓
┌─────────────┐
│  短期记忆   │ ← 7-30天
└─────────────┘
    ↓ (定期整理)
┌─────────────┐
│  长期记忆   │ ← 永久保存
└─────────────┘

# 检索评分公式
score = semantic_similarity * 0.4 + \
        time_decay * 0.3 + \
        importance * 0.2 + \
        access_frequency * 0.1

用户提问: "帮我订那家餐厅"
    ↓
提取关键词: ["订餐厅", "那家"]
    ↓
向量化查询
    ↓
┌─────────────────┐
│ 向量相似度搜索   │
│ 召回 Top-K      │
└─────────────────┘
    ↓
┌─────────────────┐
│ 重排序 (Rerank) │
│ 结合时间、重要性 │
└─────────────────┘
    ↓
┌─────────────────┐
│ 注入上下文       │
│ 生成回答         │
└─────────────────┘

# 核心类设计
class MemorySystem:
    def __init__(self):
        self.vector_db = ChromaDB()
        self.graph_db = Neo4j()
        self.embedder = OpenAIEmbedding()
    
    def store(self, content: str, metadata: dict):
        """存储记忆"""
        embedding = self.embedder.encode(content)
        self.vector_db.add(embedding, content, metadata)
        
        # 提取实体关系存入图数据库
        entities = self.extract_entities(content)
        self.graph_db.add_relations(entities)
    
    def retrieve(self, query: str, top_k: int = 5):
        """检索记忆"""
        query_embedding = self.embedder.encode(query)
        candidates = self.vector_db.search(query_embedding, top_k * 2)
        
        # 重排序
        ranked = self.rerank(candidates, query)
        return ranked[:top_k]

def should_forget(memory: Memory) -> bool:
    """判断是否应该遗忘"""
    if memory.importance > 0.8:
        return False  # 重要记忆永不遗忘
    
    if memory.last_access > 90_days:
        return True  # 90天未访问
    
    if memory.confidence < 0.3:
        return True  # 低置信度
    
    return False