From 5d663347766c631e04d513d090cce4260d57e8e4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=98=A5=E6=B2=B3=E6=99=B4?= Date: Mon, 21 Apr 2025 14:21:08 +0900 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=9C=AA=E8=A7=A3?= =?UTF-8?q?=E6=9E=90=E7=9A=84=E9=A5=AE=E7=94=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../memory_system/manually_alter_memory.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) diff --git a/src/plugins/memory_system/manually_alter_memory.py b/src/plugins/memory_system/manually_alter_memory.py index 818742113..1452d3d56 100644 --- a/src/plugins/memory_system/manually_alter_memory.py +++ b/src/plugins/memory_system/manually_alter_memory.py @@ -5,7 +5,8 @@ import time from pathlib import Path import datetime from rich.console import Console -from memory_manual_build import Memory_graph, Hippocampus # 海马体和记忆图 +from Hippocampus import Hippocampus # 海马体和记忆图 + from dotenv import load_dotenv @@ -45,13 +46,13 @@ else: # 查询节点信息 -def query_mem_info(memory_graph: Memory_graph): +def query_mem_info(hippocampus: Hippocampus): while True: query = input("\n请输入新的查询概念（输入'退出'以结束）：") if query.lower() == "退出": break - items_list = memory_graph.get_related_item(query) + items_list = hippocampus.memory_graph.get_related_item(query) if items_list: have_memory = False first_layer, second_layer = items_list @@ -312,14 +313,11 @@ def alter_mem_edge(hippocampus: Hippocampus): async def main(): start_time = time.time() - # 创建记忆图 - memory_graph = Memory_graph() - # 创建海马体 - hippocampus = Hippocampus(memory_graph) + hippocampus = Hippocampus() # 从数据库同步数据 - hippocampus.sync_memory_from_db() + hippocampus.entorhinal_cortex.sync_memory_from_db() end_time = time.time() logger.info(f"\033[32m[加载海马体耗时: {end_time - start_time:.2f} 秒]\033[0m") @@ -338,7 +336,7 @@ async def main(): query = -1 if query == 0: - query_mem_info(memory_graph) + query_mem_info(hippocampus.memory_graph) elif query == 1: add_mem_node(hippocampus) elif query == 2: @@ -355,7 +353,7 @@ async def main(): print("已结束操作") break - hippocampus.sync_memory_to_db() + hippocampus.entorhinal_cortex.sync_memory_to_db() if __name__ == "__main__":