修复BUG可能造成记忆被清空问题

pull/940/head
Process Xie 2025-05-09 22:24:39 +08:00 committed by GitHub
parent db39318ffb
commit a948a0e534
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 2 deletions

View File

@ -1046,8 +1046,14 @@ class EntorhinalCortex:
# 清空数据库
clear_start = time.time()
db.graph_data.nodes.delete_many({})
db.graph_data.edges.delete_many({})
# 备份节点和边集合
db.graph_data.nodes.rename("nodes_backup", dropTarget=True)
db.graph_data.edges.rename("edges_backup", dropTarget=True)
# 创建新的空集合
db.create_collection("nodes")
db.create_collection("edges")
clear_end = time.time()
logger.info(f"[数据库] 清空数据库耗时: {clear_end - clear_start:.2f}")