feat: 现在使用工具调用来更新关系和心情而不是固定更新

src/heart_flow/README.md

@@ -0,0 +1,82 @@
+# 心流系统 (Heart Flow System)
+
+心流系统是一个模拟AI机器人内心思考和情感流动的核心系统。它通过多层次的心流结构，使AI能够对外界信息进行观察、思考和情感反应，从而产生更自然的对话和行为。
+
+## 系统架构
+
+### 1. 主心流 (Heartflow)
+- 位于 `heartflow.py`
+- 作为整个系统的主控制器
+- 负责管理和协调多个子心流
+- 维护AI的整体思维状态
+- 定期进行全局思考更新
+
+### 2. 子心流 (SubHeartflow)
+- 位于 `sub_heartflow.py`
+- 处理具体的对话场景（如群聊）
+- 维护特定场景下的思维状态
+- 通过观察者模式接收和处理信息
+- 能够进行独立的思考和回复判断
+
+### 3. 观察系统 (Observation)
+- 位于 `observation.py`
+- 负责收集和处理外部信息
+- 支持多种观察类型（如聊天观察）
+- 对信息进行实时总结和更新
+
+## 主要功能
+
+### 思维系统
+- 定期进行思维更新
+- 维护短期记忆和思维连续性
+- 支持多层次的思维处理
+
+### 情感系统
+- 情绪状态管理
+- 回复意愿判断
+- 情感因素影响决策
+
+### 交互系统
+- 群聊消息处理
+- 多场景并行处理
+- 智能回复生成
+
+## 工作流程
+
+1. 主心流启动并创建必要的子心流
+2. 子心流通过观察者接收外部信息
+3. 系统进行信息处理和思维更新
+4. 根据情感状态和思维结果决定是否回复
+5. 生成合适的回复并更新思维状态
+
+## 使用说明
+
+### 创建新的子心流
+```python
+heartflow = Heartflow()
+subheartflow = heartflow.create_subheartflow(chat_id)
+```
+
+### 添加观察者
+```python
+observation = ChattingObservation(chat_id)
+subheartflow.add_observation(observation)
+```
+
+### 启动心流系统
+```python
+await heartflow.heartflow_start_working()
+```
+
+## 配置说明
+
+系统的主要配置参数：
+- `sub_heart_flow_stop_time`: 子心流停止时间
+- `sub_heart_flow_freeze_time`: 子心流冻结时间
+- `heart_flow_update_interval`: 心流更新间隔
+
+## 注意事项
+
+1. 子心流会在长时间不活跃后自动清理
+2. 需要合理配置更新间隔以平衡性能和响应速度
+3. 观察系统会限制消息处理数量以避免过载 

src/heart_flow/sub_heartflow.py

@@ -18,7 +18,6 @@ import random
 from src.plugins.chat.chat_stream import ChatStream
 from src.plugins.person_info.relationship_manager import relationship_manager
 from src.plugins.chat.utils import get_recent_group_speaker
-from src.do_tool.tool_use import ToolUser
 from ..plugins.utils.prompt_builder import Prompt, global_prompt_manager
 
 subheartflow_config = LogConfig(
@@ -32,7 +31,7 @@ logger = get_module_logger("subheartflow", config=subheartflow_config)
 def init_prompt():
     prompt = ""
     # prompt += f"麦麦的总体想法是：{self.main_heartflow_info}\n\n"
-    prompt += "{collected_info}\n"
+    prompt += "{extra_info}\n"
     prompt += "{relation_prompt_all}\n"
     prompt += "{prompt_personality}\n"
     prompt += "刚刚你的想法是{current_thinking_info}。如果有新的内容，记得转换话题\n"
@@ -47,6 +46,7 @@ def init_prompt():
     Prompt(prompt, "sub_heartflow_prompt_before")
     prompt = ""
     # prompt += f"你现在正在做的事情是：{schedule_info}\n"
+    prompt += "{extra_info}\n"
     prompt += "{prompt_personality}\n"
     prompt += "现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{chat_observe_info}\n"
     prompt += "刚刚你的想法是{current_thinking_info}。"
@@ -97,7 +97,7 @@ class SubHeartflow:
 
         self.bot_name = global_config.BOT_NICKNAME
 
-        self.tool_user = ToolUser()
+        
 
     def add_observation(self, observation: Observation):
         """添加一个新的observation对象到列表中，如果已存在相同id的observation则不添加"""
@@ -151,25 +151,14 @@ class SubHeartflow:
         observation = self.observations[0]
         await observation.observe()
 
-    async def do_thinking_before_reply(self, message_txt: str, sender_name: str, chat_stream: ChatStream):
+    async def do_thinking_before_reply(self, message_txt: str, sender_name: str, chat_stream: ChatStream, extra_info: str):
         current_thinking_info = self.current_mind
         mood_info = self.current_state.mood
         # mood_info = "你很生气，很愤怒"
         observation = self.observations[0]
         chat_observe_info = observation.observe_info
-        # print(f"chat_observe_info：{chat_observe_info}")
 
-        # 首先尝试使用工具获取更多信息
-        tool_result = await self.tool_user.use_tool(message_txt, sender_name, chat_stream)
 
-        # 如果工具被使用且获得了结果，将收集到的信息合并到思考中
-        collected_info = ""
-        if tool_result.get("used_tools", False):
-            logger.info("使用工具收集了信息")
-
-            # 如果有收集到的信息，将其添加到当前思考中
-            if "collected_info" in tool_result:
-                collected_info = tool_result["collected_info"]
 
         # 开始构建prompt
         prompt_personality = f"你的名字是{self.bot_name},你"
@@ -226,7 +215,7 @@ class SubHeartflow:
         # prompt += f"记得结合上述的消息，生成内心想法，文字不要浮夸，注意你就是{self.bot_name}，{self.bot_name}指的就是你。"
 
         prompt = (await global_prompt_manager.get_prompt_async("sub_heartflow_prompt_before")).format(
-            collected_info,
+            extra_info,
             relation_prompt_all,
             prompt_personality,
             current_thinking_info,
@@ -250,7 +239,7 @@ class SubHeartflow:
         logger.info(f"麦麦的思考前脑内状态：{self.current_mind}")
         return self.current_mind, self.past_mind
 
-    async def do_thinking_after_reply(self, reply_content, chat_talking_prompt):
+    async def do_thinking_after_reply(self, reply_content, chat_talking_prompt, extra_info):
         # print("麦麦回复之后脑袋转起来了")
 
         # 开始构建prompt
@@ -277,20 +266,15 @@ class SubHeartflow:
 
         message_new_info = chat_talking_prompt
         reply_info = reply_content
-        # schedule_info = bot_schedule.get_current_num_task(num=1, time_info=False)
 
-        # prompt = ""
-        # # prompt += f"你现在正在做的事情是：{schedule_info}\n"
-        # prompt += f"{prompt_personality}\n"
-        # prompt += f"现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{chat_observe_info}\n"
-        # prompt += f"刚刚你的想法是{current_thinking_info}。"
-        # prompt += f"你现在看到了网友们发的新消息:{message_new_info}\n"
-        # prompt += f"你刚刚回复了群友们:{reply_info}"
-        # prompt += f"你现在{mood_info}"
-        # prompt += "现在你接下去继续思考，产生新的想法，记得保留你刚刚的想法，不要分点输出，输出连贯的内心独白"
-        # prompt += "不要太长，但是记得结合上述的消息，要记得你的人设，关注聊天和新内容，关注你回复的内容，不要思考太多:"
         prompt = (await global_prompt_manager.get_prompt_async("sub_heartflow_prompt_after")).format(
-            prompt_personality, chat_observe_info, current_thinking_info, message_new_info, reply_info, mood_info
+            extra_info,
+            prompt_personality,
+            chat_observe_info,
+            current_thinking_info,
+            message_new_info,
+            reply_info,
+            mood_info,
         )
 
         try: