This commit is contained in:
2025-09-07 14:01:47 +08:00
parent 6d3753176a
commit d3a21ecd81

View File

@@ -54,6 +54,7 @@ def main():
logger.info(f"模型回复: {response}")
# 归档对话
# 这里不需要使用llm_client对象因为它只是存储数据不生成新内容
archive_conversation(
memory_service=memory_service, # 内存记忆服务
collection_name=collection_name, # 集合名称
@@ -65,7 +66,7 @@ def main():
# 等待一段时间确保索引更新
logger.info("等待索引更新...")
time.sleep(5)
time.sleep(10)
# 验证大模型是否记住个人信息
logger.info("验证大模型是否记住个人信息...")
@@ -73,7 +74,7 @@ def main():
test_response = handle_conversation_turn(
memory_service=memory_service, # 内存记忆服务
llm_client=llm_client,
llm_client=llm_client,# 大模型客户端
collection_name=collection_name,# 集合名称
user_id=user_id, # 用户ID
assistant_id=assistant_id, # 助手ID