From 94a554699e0be0641ae918df71ca1e480a682339 Mon Sep 17 00:00:00 2001
From: SengokuCola <1026294844@qq.com>
Date: Fri, 28 Mar 2025 09:34:21 +0800
Subject: [PATCH] =?UTF-8?q?better=EF=BC=9A=E4=BC=98=E5=8C=96=E7=BB=9F?=
=?UTF-8?q?=E8=AE=A1=E5=92=8C=E5=BF=83=E6=B5=81=E6=8F=90=E7=A4=BA=E8=AF=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
src/common/logger.py | 4 ++--
src/plugins/chat/prompt_builder.py | 2 +-
src/plugins/memory_system/Hippocampus.py | 4 ++--
src/plugins/utils/statistic.py | 14 ++++++++++----
src/think_flow_demo/heartflow.py | 7 ++++---
src/think_flow_demo/sub_heartflow.py | 18 ++++++++++--------
6 files changed, 29 insertions(+), 20 deletions(-)
diff --git a/src/common/logger.py b/src/common/logger.py
index 8556c8058..ef41f87ab 100644
--- a/src/common/logger.py
+++ b/src/common/logger.py
@@ -81,7 +81,7 @@ MEMORY_STYLE_CONFIG = {
"file_format": ("{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 海马体 | {message}"),
},
"simple": {
- "console_format": ("{time:MM-DD HH:mm} | 海马体 | {message}"),
+ "console_format": ("{time:MM-DD HH:mm} | 海马体 | {message}"),
"file_format": ("{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 海马体 | {message}"),
},
}
@@ -240,7 +240,7 @@ SUB_HEARTFLOW_STYLE_CONFIG = {
"file_format": ("{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 麦麦小脑袋 | {message}"),
},
"simple": {
- "console_format": ("{time:MM-DD HH:mm} | 麦麦小脑袋 | {message}"), # noqa: E501
+ "console_format": ("{time:MM-DD HH:mm} | 麦麦小脑袋 | {message}"), # noqa: E501
"file_format": ("{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 麦麦小脑袋 | {message}"),
},
}
diff --git a/src/plugins/chat/prompt_builder.py b/src/plugins/chat/prompt_builder.py
index c527df647..6b33e9881 100644
--- a/src/plugins/chat/prompt_builder.py
+++ b/src/plugins/chat/prompt_builder.py
@@ -83,7 +83,7 @@ class PromptBuilder:
text=message_txt,
max_memory_num=3,
max_memory_length=2,
- max_depth=3,
+ max_depth=4,
fast_retrieval=False
)
memory_str = ""
diff --git a/src/plugins/memory_system/Hippocampus.py b/src/plugins/memory_system/Hippocampus.py
index 7cd8ff744..bdb2a50b1 100644
--- a/src/plugins/memory_system/Hippocampus.py
+++ b/src/plugins/memory_system/Hippocampus.py
@@ -1046,14 +1046,14 @@ class Hippocampus:
# 将选中的节点添加到remember_map
for node, normalized_activation in sorted_nodes:
remember_map[node] = activate_map[node] # 使用原始激活值
- logger.info(
+ logger.debug(
f"节点 '{node}' (归一化激活值: {normalized_activation:.2f}, 激活值: {activate_map[node]:.2f})")
else:
logger.info("没有有效的激活值")
# 从选中的节点中提取记忆
all_memories = []
- logger.info("开始从选中的节点中提取记忆:")
+ # logger.info("开始从选中的节点中提取记忆:")
for node, activation in remember_map.items():
logger.debug(f"处理节点 '{node}' (激活值: {activation:.2f}):")
node_data = self.memory_graph.G.nodes[node]
diff --git a/src/plugins/utils/statistic.py b/src/plugins/utils/statistic.py
index aad33e88c..1071b29b0 100644
--- a/src/plugins/utils/statistic.py
+++ b/src/plugins/utils/statistic.py
@@ -44,13 +44,19 @@ class LLMStatistics:
def _record_online_time(self):
"""记录在线时间"""
- try:
+ current_time = datetime.now()
+ # 检查5分钟内是否已有记录
+ recent_record = db.online_time.find_one({
+ "timestamp": {
+ "$gte": current_time - timedelta(minutes=5)
+ }
+ })
+
+ if not recent_record:
db.online_time.insert_one({
- "timestamp": datetime.now(),
+ "timestamp": current_time,
"duration": 5 # 5分钟
})
- except Exception:
- logger.exception("记录在线时间失败")
def _collect_statistics_for_period(self, start_time: datetime) -> Dict[str, Any]:
"""收集指定时间段的LLM请求统计数据
diff --git a/src/think_flow_demo/heartflow.py b/src/think_flow_demo/heartflow.py
index 724ccfda3..45bf3a852 100644
--- a/src/think_flow_demo/heartflow.py
+++ b/src/think_flow_demo/heartflow.py
@@ -35,6 +35,7 @@ class Heartflow:
self._subheartflows = {}
self.active_subheartflows_nums = 0
+ self.personality_info = " ".join(global_config.PROMPT_PERSONALITY)
async def heartflow_start_working(self):
@@ -46,13 +47,13 @@ class Heartflow:
logger.info("麦麦大脑袋转起来了")
self.current_state.update_current_state_info()
- personality_info = " ".join(global_config.PROMPT_PERSONALITY)
+ personality_info = self.personality_info
current_thinking_info = self.current_mind
mood_info = self.current_state.mood
related_memory_info = 'memory'
sub_flows_info = await self.get_all_subheartflows_minds()
- schedule_info = bot_schedule.get_current_num_task(num = 5,time_info = True)
+ schedule_info = bot_schedule.get_current_num_task(num = 4,time_info = True)
prompt = ""
prompt += f"你刚刚在做的事情是：{schedule_info}\n"
@@ -91,7 +92,7 @@ class Heartflow:
return await self.minds_summary(sub_minds)
async def minds_summary(self,minds_str):
- personality_info = " ".join(BotConfig.PROMPT_PERSONALITY)
+ personality_info = self.personality_info
mood_info = self.current_state.mood
prompt = ""
diff --git a/src/think_flow_demo/sub_heartflow.py b/src/think_flow_demo/sub_heartflow.py
index b2179dc43..805218d5a 100644
--- a/src/think_flow_demo/sub_heartflow.py
+++ b/src/think_flow_demo/sub_heartflow.py
@@ -75,7 +75,7 @@ class SubHeartflow:
related_memory = await HippocampusManager.get_instance().get_memory_from_text(
text=message_stream_info,
- max_memory_num=3,
+ max_memory_num=2,
max_memory_length=2,
max_depth=3,
fast_retrieval=False
@@ -96,10 +96,12 @@ class SubHeartflow:
prompt += f"你刚刚在做的事情是：{schedule_info}\n"
# prompt += f"麦麦的总体想法是：{self.main_heartflow_info}\n\n"
prompt += f"你{self.personality_info}\n"
- prompt += f"现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{message_stream_info}\n"
if related_memory_info:
- prompt += f"你想起来{related_memory_info}。"
- prompt += f"刚刚你的想法是{current_thinking_info}。"
+ prompt += f"你想起来你之前见过的回忆：{related_memory_info}。\n以上是你的回忆，不一定是目前聊天里的人说的，也不一定是现在发生的事情，请记住。\n"
+ prompt += f"刚刚你的想法是{current_thinking_info}。\n"
+ prompt += "-----------------------------------\n"
+ if message_stream_info:
+ prompt += f"现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{message_stream_info}\n"
prompt += f"你现在{mood_info}。\n"
prompt += "现在你接下去继续思考，产生新的想法，不要分点输出，输出连贯的内心独白，不要太长，"
prompt += "但是记得结合上述的消息，要记得维持住你的人设，关注聊天和新内容，不要思考太多:"
@@ -108,7 +110,7 @@ class SubHeartflow:
self.update_current_mind(reponse)
self.current_mind = reponse
- print(prompt)
+ logger.info(f"prompt:\n{prompt}\n")
logger.info(f"麦麦的脑内状态：{self.current_mind}")
async def do_after_reply(self,reply_content,chat_talking_prompt):
@@ -117,7 +119,7 @@ class SubHeartflow:
current_thinking_info = self.current_mind
mood_info = self.current_state.mood
- related_memory_info = 'memory'
+ # related_memory_info = 'memory'
message_stream_info = self.outer_world.talking_summary
message_new_info = chat_talking_prompt
reply_info = reply_content
@@ -129,8 +131,8 @@ class SubHeartflow:
prompt += f"你{self.personality_info}\n"
prompt += f"现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{message_stream_info}\n"
- if related_memory_info:
- prompt += f"你想起来{related_memory_info}。"
+ # if related_memory_info:
+ # prompt += f"你想起来{related_memory_info}。"
prompt += f"刚刚你的想法是{current_thinking_info}。"
prompt += f"你现在看到了网友们发的新消息:{message_new_info}\n"
prompt += f"你刚刚回复了群友们:{reply_info}"