fix:调整目录结构,优化hfc prompt,移除日程,移除动态和llm判断willing模式,
This commit is contained in:
64
src/chat/memory_system/debug_memory.py
Normal file
64
src/chat/memory_system/debug_memory.py
Normal file
@@ -0,0 +1,64 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import asyncio
|
||||
import time
|
||||
import sys
|
||||
import os
|
||||
|
||||
# 添加项目根目录到系统路径
|
||||
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__)))))
|
||||
from src.chat.memory_system.Hippocampus import HippocampusManager
|
||||
from src.config.config import global_config
|
||||
from rich.traceback import install
|
||||
|
||||
install(extra_lines=3)
|
||||
|
||||
|
||||
async def test_memory_system():
|
||||
"""测试记忆系统的主要功能"""
|
||||
try:
|
||||
# 初始化记忆系统
|
||||
print("开始初始化记忆系统...")
|
||||
hippocampus_manager = HippocampusManager.get_instance()
|
||||
hippocampus_manager.initialize(global_config=global_config)
|
||||
print("记忆系统初始化完成")
|
||||
|
||||
# 测试记忆构建
|
||||
# print("开始测试记忆构建...")
|
||||
# await hippocampus_manager.build_memory()
|
||||
# print("记忆构建完成")
|
||||
|
||||
# 测试记忆检索
|
||||
test_text = "千石可乐在群里聊天"
|
||||
|
||||
# test_text = '''千石可乐:分不清AI的陪伴和人类的陪伴,是这样吗?'''
|
||||
print(f"开始测试记忆检索,测试文本: {test_text}\n")
|
||||
memories = await hippocampus_manager.get_memory_from_text(
|
||||
text=test_text, max_memory_num=3, max_memory_length=2, max_depth=3, fast_retrieval=False
|
||||
)
|
||||
|
||||
await asyncio.sleep(1)
|
||||
|
||||
print("检索到的记忆:")
|
||||
for topic, memory_items in memories:
|
||||
print(f"主题: {topic}")
|
||||
print(f"- {memory_items}")
|
||||
|
||||
except Exception as e:
|
||||
print(f"测试过程中出现错误: {e}")
|
||||
raise
|
||||
|
||||
|
||||
async def main():
|
||||
"""主函数"""
|
||||
try:
|
||||
start_time = time.time()
|
||||
await test_memory_system()
|
||||
end_time = time.time()
|
||||
print(f"测试完成,总耗时: {end_time - start_time:.2f} 秒")
|
||||
except Exception as e:
|
||||
print(f"程序执行出错: {e}")
|
||||
raise
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
Reference in New Issue
Block a user