myagent-ai 1.13.3 → 1.13.4

package/memory/manager.py

@@ -633,6 +633,62 @@ class MemoryManager:
     # 记忆搜索
     # ==========================================================================
 
+    # ==========================================================================
+    # 时间衰减权重 (遗忘曲线)
+    # ==========================================================================
+
+    @staticmethod
+    def _compute_time_weight(created_at: str, half_life_days: float = 30.0) -> float:
+        """
+        计算记忆的时间衰减权重（模拟遗忘曲线）。
+
+        使用指数衰减模型:  weight = e^(-ln(2) * age_days / half_life_days)
+        - 刚创建的记忆权重为 1.0
+        - 经过 half_life_days 后权重降至 0.5
+        - 经过 2 * half_life_days 后权重降至 0.25
+        - 权重永远不会降到 0（长期记忆仍可被召回）
+
+        Args:
+            created_at: 记忆创建时间（ISO 8601 格式，如 "2025-01-15T08:30:00+00:00"）
+            half_life_days: 半衰期天数（默认 30 天）
+                - 7 天: 适合短期会话记忆，快速遗忘
+                - 30 天: 默认值，平衡近期与长期记忆
+                - 90 天: 适合知识型场景，长期保留
+
+        Returns:
+            时间权重值 (0.0 ~ 1.0)
+
+        Examples:
+            >>> _compute_time_weight("2025-04-12T00:00:00+00:00", half_life_days=30)
+            1.0  # 今天的记忆
+            >>> _compute_time_weight("2025-03-13T00:00:00+00:00", half_life_days=30)
+            0.5  # 30天前的记忆
+            >>> _compute_time_weight("2025-02-11T00:00:00+00:00", half_life_days=30)
+            0.25  # 60天前的记忆
+        """
+        if not created_at:
+            return 0.5  # 无时间信息的记忆给中等权重
+
+        try:
+            # 解析 ISO 8601 时间戳
+            created_dt = datetime.fromisoformat(created_at.replace("Z", "+00:00"))
+            now_dt = datetime.now(created_dt.tzinfo) if created_dt.tzinfo else datetime.now()
+
+            age_seconds = (now_dt - created_dt).total_seconds()
+            if age_seconds <= 0:
+                return 1.0  # 未来的时间戳或零延迟
+
+            age_days = age_seconds / 86400.0
+
+            # 指数衰减: e^(-ln(2) * age / half_life)
+            decay = math.exp(-0.693147 * age_days / half_life_days)
+
+            # 限制最低权重为 0.05，确保极老的记忆仍有一丝被召回的可能
+            return max(decay, 0.05)
+        except (ValueError, TypeError, OSError) as e:
+            logger.debug(f"时间衰减计算失败 ({created_at}): {e}")
+            return 0.5  # 解析失败时给中等权重
+
     # ==========================================================================
     # TF-IDF 语义搜索 (无外部依赖)
     # ==========================================================================
@@ -753,21 +809,31 @@ class MemoryManager:
         category: str = "",
         limit: int = 10,
         mode: str = "hybrid",
+        time_decay: bool = True,
+        half_life_days: float = 30.0,
     ) -> List[MemoryEntry]:
         """
-        搜索记忆。
+        搜索记忆（内置遗忘曲线时间衰减）。
 
         支持三种搜索模式:
           - "keyword": 传统 LIKE 关键词匹配（快速）
           - "semantic": TF-IDF 语义搜索（理解语义相似性）
           - "hybrid": 混合搜索（默认）= 0.4 * keyword_score + 0.6 * tfidf_score
 
+        所有模式均支持时间衰减权重（遗忘曲线），越久远的记忆权重越低。
+        最终得分 = 相关性得分 × 时间权重。
+
         Args:
             query: 搜索查询
             session_id: 会话 ID（空=所有会话）
             category: 记忆类别（空=所有类别）
             limit: 返回数量
             mode: 搜索模式 "keyword" | "semantic" | "hybrid"
+            time_decay: 是否启用时间衰减（默认 True）
+            half_life_days: 遗忘曲线半衰期天数（默认 30 天）
+                - 7 天: 短期记忆场景，快速遗忘
+                - 30 天: 默认，平衡近期与长期
+                - 90 天: 知识型场景，长期保留
         """
         conn = self._get_conn()
         conditions = ["1=1"]
@@ -787,26 +853,38 @@ class MemoryManager:
         where = " AND ".join(conditions)
 
         if mode == "keyword":
-            return self._search_keyword(conn, query, where, params, limit)
+            return self._search_keyword(conn, query, where, params, limit,
+                                         time_decay=time_decay, half_life_days=half_life_days)
         elif mode == "semantic":
-            return self._search_semantic(conn, query, where, params, limit)
+            return self._search_semantic(conn, query, where, params, limit,
+                                         time_decay=time_decay, half_life_days=half_life_days)
         else:
-            # 混合模式：取两种搜索结果的加权和
-            keyword_results = self._search_keyword(conn, query, where, params, limit * 2)
-            semantic_results = self._search_semantic(conn, query, where, params, limit * 2)
+            # 混合模式：取两种搜索结果的加权和 + 时间衰减
+            keyword_results = self._search_keyword(conn, query, where, params, limit * 2,
+                                                   time_decay=False, half_life_days=half_life_days)
+            semantic_results = self._search_semantic(conn, query, where, params, limit * 2,
+                                                     time_decay=False, half_life_days=half_life_days)
 
-            # 合并评分
+            # 合并评分（相关性 + 时间衰减）
             combined: Dict[str, Tuple[MemoryEntry, float]] = {}
             for i, entry in enumerate(keyword_results):
-                score = 1.0 - (i / max(len(keyword_results), 1))
-                combined[entry.id] = (entry, score * 0.4)
+                relevance = 1.0 - (i / max(len(keyword_results), 1))
+                score = relevance * 0.4
+                combined[entry.id] = (entry, score)
 
             for i, entry in enumerate(semantic_results):
-                score = 1.0 - (i / max(len(semantic_results), 1))
+                relevance = 1.0 - (i / max(len(semantic_results), 1))
+                sem_score = relevance * 0.6
                 if entry.id in combined:
-                    combined[entry.id] = (entry, combined[entry.id][1] + score * 0.6)
+                    combined[entry.id] = (entry, combined[entry.id][1] + sem_score)
                 else:
-                    combined[entry.id] = (entry, score * 0.6)
+                    combined[entry.id] = (entry, sem_score)
+
+            # 应用时间衰减权重: final_score = relevance_score × time_weight
+            if time_decay:
+                for mem_id, (entry, relevance_score) in combined.items():
+                    tw = self._compute_time_weight(entry.created_at, half_life_days)
+                    combined[mem_id] = (entry, relevance_score * tw)
 
             # 按综合得分排序
             sorted_results = sorted(
@@ -832,28 +910,45 @@ class MemoryManager:
         where: str,
         params: list,
         limit: int,
+        time_decay: bool = False,
+        half_life_days: float = 30.0,
     ) -> List[MemoryEntry]:
-        """关键词 LIKE 搜索"""
+        """关键词 LIKE 搜索（支持时间衰减重排序）"""
         like_pattern = f"%{query}%"
         conditions = f"{where} AND (content LIKE ? OR summary LIKE ? OR key LIKE ?)"
         search_params = params + [like_pattern, like_pattern, like_pattern]
 
+        # 取更多候选（后续按时间衰减重排序）
+        fetch_limit = limit * 3 if time_decay else limit
         sql = f"""
             SELECT * FROM memories WHERE {conditions}
             ORDER BY importance DESC, access_count DESC
             LIMIT ?
         """
-        search_params.append(limit)
+        search_params.append(fetch_limit)
         rows = conn.execute(sql, search_params).fetchall()
 
-        for row in rows:
+        entries = [MemoryEntry.from_row(row) for row in rows]
+
+        # 应用时间衰减重排序
+        if time_decay and entries:
+            scored = []
+            for entry in entries:
+                base_score = entry.importance + entry.access_count * 0.01
+                tw = self._compute_time_weight(entry.created_at, half_life_days)
+                scored.append((entry, base_score * tw))
+            scored.sort(key=lambda x: x[1], reverse=True)
+            entries = [e for e, _ in scored[:limit]]
+
+        # 更新访问计数
+        for entry in entries:
             conn.execute(
                 "UPDATE memories SET access_count = access_count + 1 WHERE id = ?",
-                (row["id"],),
+                (entry.id,),
             )
         conn.commit()
 
-        return [MemoryEntry.from_row(row) for row in rows]
+        return entries
 
     def _search_semantic(
         self,
@@ -862,8 +957,10 @@ class MemoryManager:
         where: str,
         params: list,
         limit: int,
+        time_decay: bool = False,
+        half_life_days: float = 30.0,
     ) -> List[MemoryEntry]:
-        """TF-IDF 语义搜索"""
+        """TF-IDF 语义搜索（支持时间衰减重排序）"""
         # 先取一批候选文档
         candidate_sql = f"SELECT * FROM memories WHERE {where} ORDER BY created_at DESC LIMIT 200"
         rows = conn.execute(candidate_sql, params).fetchall()
@@ -883,6 +980,14 @@ class MemoryManager:
         # 计算 TF-IDF 得分
         scores = self._compute_tfidf(query, documents)
 
+        # 应用时间衰减重排序
+        if time_decay:
+            for doc_id, tfidf_score in scores.items():
+                row = row_map.get(doc_id)
+                if row:
+                    tw = self._compute_time_weight(row["created_at"], half_life_days)
+                    scores[doc_id] = tfidf_score * tw
+
         # 按得分排序，取 top N
         top_ids = list(scores.keys())[:limit]
         result = [MemoryEntry.from_row(row_map[doc_id]) for doc_id in top_ids if doc_id in row_map]
@@ -903,9 +1008,12 @@ class MemoryManager:
         category: str = "",
         limit: int = 20,
         mode: str = "hybrid",
+        time_decay: bool = True,
+        half_life_days: float = 30.0,
     ) -> List[MemoryEntry]:
         """跨会话搜索"""
-        return self.search(query, session_id="", category=category, limit=limit, mode=mode)
+        return self.search(query, session_id="", category=category, limit=limit, mode=mode,
+                          time_decay=time_decay, half_life_days=half_life_days)
 
     # ==========================================================================
     # 记忆总结与维护

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "myagent-ai",
-  "version": "1.13.3",
+  "version": "1.13.4",
   "description": "本地桌面端执行型AI助手 - Open Interpreter 风格 | Local Desktop Execution-Oriented AI Assistant",
   "main": "main.py",
   "bin": {