fix:优化激活值,优化logger显示

This commit is contained in:
SengokuCola
2025-03-28 09:09:30 +08:00
parent 4a72fe104a
commit de8d2aba68
8 changed files with 102 additions and 34 deletions

View File

@@ -130,9 +130,9 @@ class ChatBot:
# 根据话题计算激活度
topic = ""
interested_rate = await HippocampusManager.get_instance().get_activate_from_text(
message.processed_plain_text)*300
message.processed_plain_text,fast_retrieval=True)
# interested_rate = 0.1
logger.info(f"{message.processed_plain_text}的激活度:{interested_rate}")
# logger.info(f"对{message.processed_plain_text}的激活度:{interested_rate}")
# logger.info(f"\033[1;32m[主题识别]\033[0m 使用{global_config.topic_extract}主题: {topic}")
await self.storage.store_message(message, chat, topic[0] if topic else None)

View File

@@ -81,15 +81,15 @@ class PromptBuilder:
# 调用 hippocampus 的 get_relevant_memories 方法
relevant_memories = await HippocampusManager.get_instance().get_memory_from_text(
text=message_txt,
max_memory_num=4,
max_memory_num=3,
max_memory_length=2,
max_depth=3,
fast_retrieval=True
fast_retrieval=False
)
memory_str = ""
for _topic, memories in relevant_memories:
memory_str += f"{memories}\n"
print(f"memory_str: {memory_str}")
# print(f"memory_str: {memory_str}")
if relevant_memories:
# 格式化记忆内容