better 稍微优化了一下记忆prompt

This commit is contained in:
SengokuCola
2025-03-19 14:44:46 +08:00
parent b97d62d365
commit b187c8a21b
2 changed files with 9 additions and 8 deletions

View File

@@ -85,13 +85,13 @@ class PromptBuilder:
# 调用 hippocampus 的 get_relevant_memories 方法 # 调用 hippocampus 的 get_relevant_memories 方法
relevant_memories = await hippocampus.get_relevant_memories( relevant_memories = await hippocampus.get_relevant_memories(
text=message_txt, max_topics=5, similarity_threshold=0.4, max_memory_num=5 text=message_txt, max_topics=3, similarity_threshold=0.5, max_memory_num=4
) )
if relevant_memories: if relevant_memories:
# 格式化记忆内容 # 格式化记忆内容
memory_str = '\n'.join(f"关于「{m['topic']}」的记忆:{m['content']}" for m in relevant_memories) memory_str = '\n'.join(m['content'] for m in relevant_memories)
memory_prompt = f"看到这些聊天,你想起来\n{memory_str}\n" memory_prompt = f"你回忆起\n{memory_str}\n"
# 打印调试信息 # 打印调试信息
logger.debug("[记忆检索]找到以下相关记忆:") logger.debug("[记忆检索]找到以下相关记忆:")
@@ -155,13 +155,14 @@ class PromptBuilder:
prompt = f""" prompt = f"""
今天是{current_date},现在是{current_time},你今天的日程是:\ 今天是{current_date},现在是{current_time},你今天的日程是:\
`<schedule>` `<schedule>`\n
{bot_schedule.today_schedule} {bot_schedule.today_schedule}\n
`</schedule>`\ `</schedule>`\n
{prompt_info} {prompt_info}\n
{memory_prompt}\n
{chat_target}\n {chat_target}\n
{chat_talking_prompt}\n {chat_talking_prompt}\n
{memory_prompt} 现在"{sender_name}"说的:\n 现在"{sender_name}"说的:\n
`<UserMessage>`\n `<UserMessage>`\n
{message_txt}\n {message_txt}\n
`</UserMessage>`\n `</UserMessage>`\n

View File