adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[fix] Remove the limit on the number of output items.

Browse Source
This commit is contained in:
lanceyq
2026-03-30 11:35:09 +08:00
parent 9f6026492d
commit 8f216db353
1 changed files with 4 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
api/app/services/memory_forget_service.py
View File

@@ -203,8 +203,7 @@ class MemoryForgetService:
connector: Neo4jConnector, connector: Neo4jConnector,
end_user_id: str, end_user_id: str,
forgetting_threshold: float, forgetting_threshold: float,
min_days_since_access: int, min_days_since_access: int
limit: int = 20
) -> list[Dict[str, Any]]: ) -> list[Dict[str, Any]]:
""" """
获取待遗忘节点列表 获取待遗忘节点列表
@@ -216,7 +215,6 @@ class MemoryForgetService:
end_user_id: 组ID end_user_id: 组ID
forgetting_threshold: 遗忘阈值 forgetting_threshold: 遗忘阈值
min_days_since_access: 最小未访问天数 min_days_since_access: 最小未访问天数
limit: 返回节点数量限制
Returns: Returns:
list: 待遗忘节点列表 list: 待遗忘节点列表
@@ -247,14 +245,12 @@ class MemoryForgetService:
n.activation_value as activation_value, n.activation_value as activation_value,
n.last_access_time as last_access_time n.last_access_time as last_access_time
ORDER BY n.activation_value ASC ORDER BY n.activation_value ASC
LIMIT $limit
""" """
params = { params = {
'end_user_id': end_user_id, 'end_user_id': end_user_id,
'threshold': forgetting_threshold, 'threshold': forgetting_threshold,
'min_access_time_str': min_access_time_str, 'min_access_time_str': min_access_time_str
'limit': limit
} }
results = await connector.execute_query(query, **params) results = await connector.execute_query(query, **params)
@@ -636,7 +632,7 @@ class MemoryForgetService:
api_logger.error(f"获取历史趋势数据失败: {str(e)}") api_logger.error(f"获取历史趋势数据失败: {str(e)}")
# 失败时返回空列表，不影响主流程 # 失败时返回空列表，不影响主流程
# 获取待遗忘节点列表（前20个满足遗忘条件的节点） # 获取待遗忘节点列表
pending_nodes = [] pending_nodes = []
try: try:
if end_user_id: if end_user_id:
@@ -652,8 +648,7 @@ class MemoryForgetService:
connector=connector, connector=connector,
end_user_id=end_user_id, end_user_id=end_user_id,
forgetting_threshold=forgetting_threshold, forgetting_threshold=forgetting_threshold,
min_days_since_access=int(min_days), min_days_since_access=int(min_days)
limit=20
) )
api_logger.info(f"成功获取 {len(pending_nodes)} 个待遗忘节点") api_logger.info(f"成功获取 {len(pending_nodes)} 个待遗忘节点")