fix：修复hfc根本不会被触发的bug

src/heart_flow/background_tasks.py

@@ -6,11 +6,14 @@ from src.common.logger import get_module_logger
 
 # Need manager types for dependency injection
 from src.heart_flow.mai_state_manager import MaiStateManager, MaiStateInfo
-from src.heart_flow.subheartflow_manager import SubHeartflowManager
+from src.heart_flow.subheartflow_manager import SubHeartflowManager, ChatState
 from src.heart_flow.interest_logger import InterestLogger
 
 logger = get_module_logger("background_tasks")
 
+# 新增兴趣评估间隔
+INTEREST_EVAL_INTERVAL_SECONDS = 5
+
 
 class BackgroundTaskManager:
     """管理 Heartflow 的后台周期性任务。"""
@@ -25,6 +28,8 @@ class BackgroundTaskManager:
         cleanup_interval: int,
         log_interval: int,
         inactive_threshold: int,
+        # 新增兴趣评估间隔参数
+        interest_eval_interval: int = INTEREST_EVAL_INTERVAL_SECONDS,
     ):
         self.mai_state_info = mai_state_info
         self.mai_state_manager = mai_state_manager
@@ -36,46 +41,70 @@ class BackgroundTaskManager:
         self.cleanup_interval = cleanup_interval
         self.log_interval = log_interval
         self.inactive_threshold = inactive_threshold  # For cleanup task
+        self.interest_eval_interval = interest_eval_interval # 存储兴趣评估间隔
 
         # Task references
         self._state_update_task: Optional[asyncio.Task] = None
         self._cleanup_task: Optional[asyncio.Task] = None
         self._logging_task: Optional[asyncio.Task] = None
+        self._interest_eval_task: Optional[asyncio.Task] = None # 新增兴趣评估任务引用
         self._tasks: List[Optional[asyncio.Task]] = []  # Keep track of all tasks
 
     async def start_tasks(self):
-        """启动所有后台任务"""
+        """启动所有后台任务
-        # 状态更新任务
-        if self._state_update_task is None or self._state_update_task.done():
-            self._state_update_task = asyncio.create_task(
-                self._run_state_update_cycle(self.update_interval), name="hf_state_update"
-            )
-            self._tasks.append(self._state_update_task)
-            logger.debug(f"聊天状态更新任务已启动 间隔:{self.update_interval}s")
-        else:
-            logger.warning("状态更新任务已在运行")
         
-        # 清理任务
+        功能说明:
-        if self._cleanup_task is None or self._cleanup_task.done():
+        - 启动核心后台任务: 状态更新、清理、日志记录和兴趣评估
-            self._cleanup_task = asyncio.create_task(self._run_cleanup_cycle(), name="hf_cleanup")
+        - 每个任务启动前检查是否已在运行
-            self._tasks.append(self._cleanup_task)
+        - 将任务引用保存到任务列表
-            logger.info(f"清理任务已启动 间隔:{self.cleanup_interval}s 阈值:{self.inactive_threshold}s")
+        """
-        else:
-            logger.warning("清理任务已在运行")
         
-        # 日志任务
+        # 任务配置列表: (任务变量名, 任务函数, 任务名称, 日志级别, 额外日志信息, 任务对象引用属性名)
-        if self._logging_task is None or self._logging_task.done():
+        task_configs = [
-            self._logging_task = asyncio.create_task(self._run_logging_cycle(), name="hf_logging")
+            (self._state_update_task, 
-            self._tasks.append(self._logging_task)
+            lambda: self._run_state_update_cycle(self.update_interval),
-            logger.info(f"日志任务已启动 间隔:{self.log_interval}s")
+            "hf_state_update", 
-        else:
+            "debug", 
-            logger.warning("日志任务已在运行")
+            f"聊天状态更新任务已启动 间隔:{self.update_interval}s",
+            "_state_update_task"),
             
-        # # 初始状态检查
+            (self._cleanup_task,
-        # initial_state = self.mai_state_info.get_current_state()
+            self._run_cleanup_cycle,
-        # if initial_state != self.mai_state_info.mai_status.OFFLINE:
+            "hf_cleanup",
-        #     logger.info(f"初始状态:{initial_state.value} 触发初始激活检查")
+            "info",
-        #     asyncio.create_task(self.subheartflow_manager.activate_random_subflows_to_chat(initial_state))
+            f"清理任务已启动 间隔:{self.cleanup_interval}s 阈值:{self.inactive_threshold}s",
+            "_cleanup_task"),
+            
+            (self._logging_task,
+            self._run_logging_cycle,
+            "hf_logging",
+            "info",
+            f"日志任务已启动 间隔:{self.log_interval}s",
+            "_logging_task"),
+
+            # 新增兴趣评估任务配置
+            (self._interest_eval_task,
+            self._run_interest_eval_cycle,
+            "hf_interest_eval",
+            "debug", # 设为debug，避免过多日志
+            f"兴趣评估任务已启动 间隔:{self.interest_eval_interval}s",
+            "_interest_eval_task"),
+        ]
+
+        # 统一启动所有任务
+        for task_var, task_func, task_name, log_level, log_msg, task_attr_name in task_configs:
+            # 检查任务变量是否存在且未完成
+            current_task_var = getattr(self, task_attr_name)
+            if current_task_var is None or current_task_var.done():
+                new_task = asyncio.create_task(task_func(), name=task_name)
+                setattr(self, task_attr_name, new_task)  # 更新任务变量
+                if new_task not in self._tasks: # 避免重复添加
+                    self._tasks.append(new_task)
+                
+                # 根据配置记录不同级别的日志
+                getattr(logger, log_level)(log_msg)
+            else:
+                logger.warning(f"{task_name}任务已在运行")
 
     async def stop_tasks(self):
         """停止所有后台任务。
@@ -126,15 +155,11 @@ class BackgroundTaskManager:
             # 计算并执行间隔等待
             elapsed = asyncio.get_event_loop().time() - start_time
             sleep_time = max(0, interval - elapsed)
-            if sleep_time < 0.1:  # 任务超时处理
+            # if sleep_time < 0.1:  # 任务超时处理, DEBUG 时可能干扰断点
-                logger.warning(f"任务 {task_name} 超时执行 ({elapsed:.2f}s > {interval}s)")
+            #     logger.warning(f"任务 {task_name} 超时执行 ({elapsed:.2f}s > {interval}s)")
             await asyncio.sleep(sleep_time)
 
-        # 非离线状态时评估兴趣
+        logger.debug(f"任务循环结束: {task_name}") # 调整日志信息
-        if self.mai_state_info.get_current_state() != self.mai_state_info.mai_status.OFFLINE:
-            await self.subheartflow_manager.evaluate_interest_and_promote()
-
-        logger.debug(f"任务循环结束, 当前状态: {self.mai_state_info.get_current_state().value}")
 
     async def _perform_state_update_work(self):
         """执行状态更新工作"""
@@ -184,8 +209,15 @@ class BackgroundTaskManager:
             logger.debug("[Background Task Cleanup] Cleanup cycle finished. No inactive flows found.")
 
     async def _perform_logging_work(self):
-        """执行一轮兴趣日志记录。"""
+        """执行一轮状态日志记录。"""
-        await self.interest_logger.log_interest_states()
+        await self.interest_logger.log_all_states()
+
+    # --- 新增兴趣评估工作函数 ---
+    async def _perform_interest_eval_work(self):
+        """执行一轮子心流兴趣评估与提升检查。"""
+        # 直接调用 subheartflow_manager 的方法，并传递当前状态信息
+        await self.subheartflow_manager.evaluate_interest_and_promote(self.mai_state_info)
+    # --- 结束新增 ---
 
     # --- Specific Task Runners --- #
     async def _run_state_update_cycle(self, interval: int):
@@ -200,5 +232,14 @@ class BackgroundTaskManager:
 
     async def _run_logging_cycle(self):
         await self._run_periodic_loop(
-            task_name="Interest Logging", interval=self.log_interval, task_func=self._perform_logging_work
+            task_name="State Logging",
+            interval=self.log_interval,
+            task_func=self._perform_logging_work
         )
+
+    # --- 新增兴趣评估任务运行器 ---
+    async def _run_interest_eval_cycle(self):
+        await self._run_periodic_loop(
+            task_name="Interest Evaluation", interval=self.interest_eval_interval, task_func=self._perform_interest_eval_work
+        )
+    # --- 结束新增 ---

src/heart_flow/heartflow.py

@@ -23,7 +23,7 @@ logger = get_module_logger("heartflow", config=heartflow_config)
 
 # Task Intervals (should be in BackgroundTaskManager or config)
 CLEANUP_INTERVAL_SECONDS = 1200
-STATE_UPDATE_INTERVAL_SECONDS = 30
+STATE_UPDATE_INTERVAL_SECONDS = 60
 
 # Thresholds (should be in SubHeartflowManager or config)
 INACTIVE_THRESHOLD_SECONDS = 1200
@@ -63,7 +63,7 @@ class Heartflow:
 
         # 子系统初始化
         self.mind: Mind = Mind(self.subheartflow_manager, self.llm_model)  # 思考管理器
-        self.interest_logger: InterestLogger = InterestLogger(self.subheartflow_manager)  # 兴趣日志记录器
+        self.interest_logger: InterestLogger = InterestLogger(self.subheartflow_manager, self)  # 兴趣日志记录器
 
         # 后台任务管理器 (整合所有定时任务)
         self.background_task_manager: BackgroundTaskManager = BackgroundTaskManager(

src/heart_flow/interest_logger.py

@@ -12,7 +12,8 @@ from src.plugins.chat.chat_stream import chat_manager
 
 if TYPE_CHECKING:
     from src.heart_flow.subheartflow_manager import SubHeartflowManager
-    from src.heart_flow.sub_heartflow import SubHeartflow  # For type hint in get_interest_states
+    from src.heart_flow.sub_heartflow import SubHeartflow
+    from src.heart_flow.heartflow import Heartflow  # 导入 Heartflow 类型
 
 logger = get_module_logger("interest_logger")
 
@@ -22,25 +23,28 @@ HISTORY_LOG_FILENAME = "interest_history.log"
 
 
 class InterestLogger:
-    """负责定期记录所有子心流的兴趣状态到日志文件。"""
+    """负责定期记录主心流和所有子心流的状态到日志文件。"""
 
-    def __init__(self, subheartflow_manager: "SubHeartflowManager"):
+    def __init__(self, subheartflow_manager: "SubHeartflowManager", heartflow: "Heartflow"):
+        """
+        初始化 InterestLogger。
+
+        Args:
+            subheartflow_manager: 子心流管理器实例。
+            heartflow: 主心流实例，用于获取主心流状态。
+        """
         self.subheartflow_manager = subheartflow_manager
+        self.heartflow = heartflow  # 存储 Heartflow 实例
         self._history_log_file_path = os.path.join(LOG_DIRECTORY, HISTORY_LOG_FILENAME)
         self._ensure_log_directory()
 
     def _ensure_log_directory(self):
         """确保日志目录存在。"""
-        try:
+        os.makedirs(LOG_DIRECTORY, exist_ok=True)
-            os.makedirs(LOG_DIRECTORY, exist_ok=True)
+        logger.info(f"已确保日志目录 '{LOG_DIRECTORY}' 存在")
-            logger.info(f"已确保日志目录 '{LOG_DIRECTORY}' 存在")
-        except OSError as e:
-            logger.error(f"创建日志目录 '{LOG_DIRECTORY}' 出错: {e}")
 
-    async def get_all_interest_states(self) -> Dict[str, Dict]:
+    async def get_all_subflow_states(self) -> Dict[str, Dict]:
-        """并发获取所有活跃子心流的当前兴趣状态。"""
+        """并发获取所有活跃子心流的当前完整状态。"""
-        _states = {}
-        # Get snapshot from the manager
         all_flows: List["SubHeartflow"] = self.subheartflow_manager.get_all_subheartflows()
         tasks = []
         results = {}
@@ -49,12 +53,11 @@ class InterestLogger:
             logger.debug("未找到任何子心流状态")
             return results
 
-        # logger.debug(f"正在获取 {len(all_flows)} 个子心流的兴趣状态...")
         for subheartflow in all_flows:
             if self.subheartflow_manager.get_subheartflow(subheartflow.subheartflow_id):
                 tasks.append(
                     asyncio.create_task(
-                        subheartflow.get_interest_state(), name=f"get_state_{subheartflow.subheartflow_id}"
+                        subheartflow.get_full_state(), name=f"get_state_{subheartflow.subheartflow_id}"
                     )
                 )
             else:
@@ -64,74 +67,87 @@ class InterestLogger:
             done, pending = await asyncio.wait(tasks, timeout=5.0)
 
             if pending:
-                logger.warning(f"获取兴趣状态超时，有 {len(pending)} 个任务未完成")
+                logger.warning(f"获取子心流状态超时，有 {len(pending)} 个任务未完成")
                 for task in pending:
                     task.cancel()
 
             for task in done:
-                try:
+                stream_id_str = task.get_name().split("get_state_")[-1]
-                    stream_id_str = task.get_name().split("get_state_")[-1]
+                stream_id = stream_id_str
-                    stream_id = stream_id_str
-                except IndexError:
-                    logger.error(f"无法从任务名 {task.get_name()} 中提取 stream_id")
-                    continue
 
-                try:
+                if task.cancelled():
+                    logger.warning(f"获取子心流 {stream_id} 状态的任务已取消(超时)", exc_info=False)
+                elif task.exception():
+                    exc = task.exception()
+                    logger.warning(f"获取子心流 {stream_id} 状态出错: {exc}")
+                else:
                     result = task.result()
                     results[stream_id] = result
-                except asyncio.CancelledError:
-                    logger.warning(f"获取子心流 {stream_id} 兴趣状态的任务已取消(超时)", exc_info=False)
-                except Exception as e:
-                    logger.warning(f"获取子心流 {stream_id} 兴趣状态出错: {e}")
 
-        logger.trace(f"成功获取 {len(results)} 个兴趣状态")
+        logger.trace(f"成功获取 {len(results)} 个子心流的完整状态")
         return results
 
-    async def log_interest_states(self):
+    async def log_all_states(self):
-        """获取所有子心流的兴趣状态并写入日志文件。"""
+        """获取主心流状态和所有子心流的完整状态并写入日志文件。"""
-        # logger.debug("开始定期记录兴趣状态...")
         try:
             current_timestamp = time.time()
-            all_interest_states = await self.get_all_interest_states()
 
-            if not all_interest_states:
+            main_mind = self.heartflow.current_mind
-                logger.debug("没有获取到任何兴趣状态")
+            # 获取 Mai 状态名称
+            mai_state_name = self.heartflow.current_state.get_current_state().name
+
+            all_subflow_states = await self.get_all_subflow_states()
+
+            log_entry_base = {
+                "timestamp": round(current_timestamp, 2),
+                "main_mind": main_mind,
+                "mai_state": mai_state_name,
+                "subflow_count": len(all_subflow_states),
+                "subflows": []
+            }
+
+            if not all_subflow_states:
+                logger.debug("没有获取到任何子心流状态，仅记录主心流状态")
+                with open(self._history_log_file_path, "a", encoding="utf-8") as f:
+                    f.write(json.dumps(log_entry_base, ensure_ascii=False) + "\n")
                 return
 
-            count = 0
+            subflow_details = []
-            try:
+            items_snapshot = list(all_subflow_states.items())
-                with open(self._history_log_file_path, "a", encoding="utf-8") as f:
+            for stream_id, state in items_snapshot:
-                    items_snapshot = list(all_interest_states.items())
+                group_name = stream_id
-                    for stream_id, state in items_snapshot:
+                try:
-                        group_name = stream_id
+                    chat_stream = chat_manager.get_stream(stream_id)
-                        try:
+                    if chat_stream:
-                            chat_stream = chat_manager.get_stream(stream_id)
+                        if chat_stream.group_info:
-                            if chat_stream and chat_stream.group_info:
+                            group_name = chat_stream.group_info.group_name
-                                group_name = chat_stream.group_info.group_name
+                        elif chat_stream.user_info:
-                            elif chat_stream and not chat_stream.group_info:
+                            group_name = f"私聊_{chat_stream.user_info.user_nickname}"
-                                group_name = (
+                except Exception as e:
-                                    f"私聊_{chat_stream.user_info.user_nickname}"
+                    logger.trace(f"无法获取 stream_id {stream_id} 的群组名: {e}")
-                                    if chat_stream.user_info
-                                    else stream_id
-                                )
-                        except Exception as e:
-                            logger.trace(f"无法获取 stream_id {stream_id} 的群组名: {e}")
-                            pass
 
-                        log_entry = {
+                interest_state = state.get("interest_state", {})
-                            "timestamp": round(current_timestamp, 2),
-                            "stream_id": stream_id,
-                            "interest_level": state.get("interest_level", 0.0),
-                            "group_name": group_name,
-                            "reply_probability": state.get("current_reply_probability", 0.0),
-                            "is_above_threshold": state.get("is_above_threshold", False),
-                        }
-                        f.write(json.dumps(log_entry, ensure_ascii=False) + "\n")
-                        count += 1
-                # logger.debug(f"成功记录 {count} 条兴趣历史到 {self._history_log_file_path}")
-            except IOError as e:
-                logger.error(f"写入兴趣历史日志到 {self._history_log_file_path} 出错: {e}")
 
+                subflow_entry = {
+                    "stream_id": stream_id,
+                    "group_name": group_name,
+                    "sub_mind": state.get("current_mind", "未知"),
+                    "sub_chat_state": state.get("chat_state", "未知"),
+                    "interest_level": interest_state.get("interest_level", 0.0),
+                    "reply_probability": interest_state.get("current_reply_probability", 0.0),
+                    "is_above_threshold": interest_state.get("is_above_threshold", False),
+                    "last_active_time": state.get("last_active_time", 0.0),
+                    "last_interaction_time": interest_state.get("last_interaction_time", 0.0),
+                }
+                subflow_details.append(subflow_entry)
+
+            log_entry_base["subflows"] = subflow_details
+
+            with open(self._history_log_file_path, "a", encoding="utf-8") as f:
+                f.write(json.dumps(log_entry_base, ensure_ascii=False) + "\n")
+
+        except IOError as e:
+            logger.error(f"写入状态日志到 {self._history_log_file_path} 出错: {e}")
         except Exception as e:
-            logger.error(f"定期记录兴趣历史时发生意外错误: {e}")
+            logger.error(f"记录状态时发生意外错误: {e}")
             logger.error(traceback.format_exc())

src/heart_flow/mai_state_manager.py

@@ -23,9 +23,9 @@ class MaiState(enum.Enum):
     """
 
     OFFLINE = "不在线"
-    PEEKING = "看一眼手机"
+    PEEKING = "看一眼"
     NORMAL_CHAT = "正常聊天"
-    FOCUSED_CHAT = "专注聊天"
+    FOCUSED_CHAT = "专心聊天"
 
     def get_normal_chat_max_num(self):
         if self == MaiState.OFFLINE:
@@ -127,9 +127,9 @@ class MaiStateManager:
         elif current_status == MaiState.PEEKING:
             logger.info("当前[在窥屏]，思考要不要继续聊下去......")
         elif current_status == MaiState.NORMAL_CHAT:
-            logger.info("当前在[闲聊]思考要不要继续聊下去......")
+            logger.info("当前在[随便看]思考要不要继续聊下去......")
         elif current_status == MaiState.FOCUSED_CHAT:
-            logger.info("当前在[激情聊天]思考要不要继续聊下去......")
+            logger.info("当前在[专心看]思考要不要继续聊下去......")
 
         # 1. 麦麦每分钟都有概率离线
         if time_since_last_min_check >= 60:
@@ -149,7 +149,7 @@ class MaiStateManager:
                     next_state_candidate = random.choices(choices_list, weights=weights, k=1)[0]
                     if next_state_candidate != MaiState.OFFLINE:
                         next_state = next_state_candidate
-                        logger.debug(f"上线！开始 {next_state.name}")
+                        logger.debug(f"上线！开始 {next_state.value}")
                     else:
                         # 继续离线状态
                         next_state = MaiState.OFFLINE
@@ -159,7 +159,7 @@ class MaiStateManager:
                     weights = [70, 20, 10]
                     choices_list = [MaiState.OFFLINE, MaiState.NORMAL_CHAT, MaiState.FOCUSED_CHAT]
                     next_state = random.choices(choices_list, weights=weights, k=1)[0]
-                    logger.debug(f"手机看完了，接下来 {next_state.name}")
+                    logger.debug(f"手机看完了，接下来 {next_state.value}")
 
             elif current_status == MaiState.NORMAL_CHAT:
                 if time_in_current_status >= 300:  # NORMAL_CHAT 最多持续 300 秒
@@ -167,16 +167,16 @@ class MaiStateManager:
                     choices_list = [MaiState.OFFLINE, MaiState.FOCUSED_CHAT]
                     next_state = random.choices(choices_list, weights=weights, k=1)[0]
                     if next_state == MaiState.FOCUSED_CHAT:
-                        logger.debug(f"继续深入聊天， {next_state.name}")
+                        logger.debug(f"继续深入聊天， {next_state.value}")
                     else:
-                        logger.debug(f"聊完了，接下来 {next_state.name}")
+                        logger.debug(f"聊完了，接下来 {next_state.value}")
 
             elif current_status == MaiState.FOCUSED_CHAT:
                 if time_in_current_status >= 600:  # FOCUSED_CHAT 最多持续 600 秒
                     weights = [80, 20]
                     choices_list = [MaiState.OFFLINE, MaiState.NORMAL_CHAT]
                     next_state = random.choices(choices_list, weights=weights, k=1)[0]
-                    logger.debug(f"深入聊天结束，接下来 {next_state.name}")
+                    logger.debug(f"深入聊天结束，接下来 {next_state.value}")
 
         # 如果决定了下一个状态，且这个状态与当前状态不同，则返回下一个状态
         if next_state is not None and next_state != current_status:

src/heart_flow/sub_heartflow.py

@@ -18,7 +18,6 @@ from src.plugins.chat.chat_stream import chat_manager
 import math
 from src.plugins.heartFC_chat.heartFC_chat import HeartFChatting
 from src.plugins.heartFC_chat.normal_chat import NormalChat
-from src.plugins.heartFC_chat.normal_chat_generator import ResponseGenerator
 from src.do_tool.tool_use import ToolUser
 from src.heart_flow.mai_state_manager import MaiStateInfo
 
@@ -61,16 +60,15 @@ def init_prompt():
 
 
 class ChatState(enum.Enum):
-    ABSENT = "不参与"
+    ABSENT = "没在看群"
-    CHAT = "闲聊"
+    CHAT = "随便水群"
-    FOCUSED = "专注"
+    FOCUSED = "激情水群"
 
 
 class ChatStateInfo:
     def __init__(self):
-        self.willing = 0
-
         self.chat_status: ChatState = ChatState.ABSENT
+        self.current_state_time = 120
 
         self.mood_manager = MoodManager()
         self.mood = self.mood_manager.get_prompt()
@@ -106,110 +104,74 @@ class InterestChatting:
         self.max_reply_probability: float = max_probability
         self.current_reply_probability: float = 0.0
         self.is_above_threshold: bool = False
-        self.state_change_callback = state_change_callback
+        self.update_task: Optional[asyncio.Task] = None
+        self._stop_event = asyncio.Event()
 
         self.interest_dict: Dict[str, tuple[MessageRecv, float, bool]] = {}
+        self.update_interval = 1.0
+        self.start_updates(self.update_interval) # 初始化时启动后台更新任务
+        
+        self.above_threshold = False
+        self.start_hfc_probability = 0.0
 
     def add_interest_dict(self, message: MessageRecv, interest_value: float, is_mentioned: bool):
         self.interest_dict[message.message_info.message_id] = (message, interest_value, is_mentioned)
         self.last_interaction_time = time.time()
 
-    async def _calculate_decay(self, current_time: float):
+    async def _calculate_decay(self):
-        time_delta = current_time - self.last_update_time
+        """计算兴趣值的衰减
-        if time_delta > 0:
-            old_interest = self.interest_level
-            if self.interest_level < 1e-9:
-                self.interest_level = 0.0
-            else:
-                if self.decay_rate_per_second <= 0:
-                    interest_logger.warning(
-                        f"InterestChatting encountered non-positive decay rate: {self.decay_rate_per_second}. Setting interest to 0."
-                    )
-                    self.interest_level = 0.0
-                elif self.interest_level < 0:
-                    interest_logger.warning(
-                        f"InterestChatting encountered negative interest level: {self.interest_level}. Setting interest to 0."
-                    )
-                    self.interest_level = 0.0
-                else:
-                    try:
-                        decay_factor = math.pow(self.decay_rate_per_second, time_delta)
-                        self.interest_level *= decay_factor
-                    except ValueError as e:
-                        interest_logger.error(
-                            f"Math error during decay calculation: {e}. Rate: {self.decay_rate_per_second}, Delta: {time_delta}, Level: {self.interest_level}. Setting interest to 0."
-                        )
-                        self.interest_level = 0.0
         
-            if old_interest != self.interest_level:
+        参数:
-                self.last_update_time = current_time
+            current_time: 当前时间戳
             
-    async def _update_reply_probability(self, current_time: float):
+        处理逻辑:
-        time_delta = current_time - self.last_update_time
+        1. 计算时间差
-        if time_delta <= 0:
+        2. 处理各种异常情况(负值/零值)
+        3. 正常计算衰减
+        4. 更新最后更新时间
+        """
+        
+        # 处理极小兴趣值情况
+        if self.interest_level < 1e-9:
+            self.interest_level = 0.0
             return
             
-        currently_above = self.interest_level >= self.trigger_threshold
+        # 异常情况处理
-        previous_is_above = self.is_above_threshold
+        if self.decay_rate_per_second <= 0:
+            interest_logger.warning(f"衰减率({self.decay_rate_per_second})无效，重置兴趣值为0")
+            self.interest_level = 0.0
+            return
             
-        if currently_above:
+        # 正常衰减计算
-            if not self.is_above_threshold:
+        try:
-                self.current_reply_probability = self.base_reply_probability
+            decay_factor = math.pow(self.decay_rate_per_second, self.update_interval)
-                interest_logger.debug(
+            self.interest_level *= decay_factor
-                    f"兴趣跨过阈值 ({self.trigger_threshold}). 概率重置为基础值: {self.base_reply_probability:.4f}"
+        except ValueError as e:
-                )
+            interest_logger.error(f"衰减计算错误: {e} 参数: 衰减率={self.decay_rate_per_second} 时间差={self.update_interval} 当前兴趣={self.interest_level}")
-            else:
+            self.interest_level = 0.0
-                increase_amount = self.probability_increase_rate * time_delta
-                self.current_reply_probability += increase_amount
             
-            self.current_reply_probability = min(self.current_reply_probability, self.max_reply_probability)
 
+    async def _update_reply_probability(self):
+        self.above_threshold = self.interest_level >= self.trigger_threshold
+        if self.above_threshold:
+            self.start_hfc_probability += 0.1
         else:
-            if previous_is_above:
+            if self.start_hfc_probability != 0:
-                if self.state_change_callback:
+                self.start_hfc_probability -= 0.1
-                    try:
-                        await self.state_change_callback(ChatState.ABSENT)
-                    except Exception as e:
-                        interest_logger.error(f"Error calling state_change_callback for ABSENT: {e}")
-
-            if 0 < self.probability_decay_factor < 1:
-                decay_multiplier = math.pow(self.probability_decay_factor, time_delta)
-                self.current_reply_probability *= decay_multiplier
-                if self.current_reply_probability < 1e-6:
-                    self.current_reply_probability = 0.0
-            elif self.probability_decay_factor <= 0:
-                if self.current_reply_probability > 0:
-                    interest_logger.warning(f"无效的衰减因子 ({self.probability_decay_factor}). 设置概率为0.")
-                    self.current_reply_probability = 0.0
-
-            self.current_reply_probability = max(self.current_reply_probability, 0.0)
-
-        self.is_above_threshold = currently_above
 
     async def increase_interest(self, current_time: float, value: float):
-        await self._update_reply_probability(current_time)
-        await self._calculate_decay(current_time)
         self.interest_level += value
         self.interest_level = min(self.interest_level, self.max_interest)
-        self.last_update_time = current_time
-        self.last_interaction_time = current_time
 
     async def decrease_interest(self, current_time: float, value: float):
-        await self._update_reply_probability(current_time)
         self.interest_level -= value
         self.interest_level = max(self.interest_level, 0.0)
-        self.last_update_time = current_time
-        self.last_interaction_time = current_time
 
     async def get_interest(self) -> float:
-        current_time = time.time()
-        await self._update_reply_probability(current_time)
-        await self._calculate_decay(current_time)
-        self.last_update_time = current_time
         return self.interest_level
 
     async def get_state(self) -> dict:
-        interest = await self.get_interest()
+        interest = self.interest_level # 直接使用属性值
         return {
             "interest_level": round(interest, 2),
             "last_update_time": self.last_update_time,
@@ -219,8 +181,6 @@ class InterestChatting:
         }
 
     async def should_evaluate_reply(self) -> bool:
-        current_time = time.time()
-        await self._update_reply_probability(current_time)
 
         if self.current_reply_probability > 0:
             trigger = random.random() < self.current_reply_probability
@@ -228,6 +188,67 @@ class InterestChatting:
         else:
             return False
 
+    # --- 新增后台更新任务相关方法 ---
+    async def _run_update_loop(self, update_interval: float = 1.0):
+        """后台循环，定期更新兴趣和回复概率。"""
+        while not self._stop_event.is_set():
+            try:
+                if self.interest_level != 0:
+                    await self._calculate_decay()
+                    
+                await self._update_reply_probability()
+                
+
+                # 等待下一个周期或停止事件
+                await asyncio.wait_for(self._stop_event.wait(), timeout=update_interval)
+            except asyncio.TimeoutError:
+                # 正常超时，继续循环
+                continue
+            except asyncio.CancelledError:
+                interest_logger.info("InterestChatting 更新循环被取消。")
+                break
+            except Exception as e:
+                interest_logger.error(f"InterestChatting 更新循环出错: {e}")
+                interest_logger.error(traceback.format_exc())
+                # 防止错误导致CPU飙升，稍作等待
+                await asyncio.sleep(5)
+        interest_logger.info("InterestChatting 更新循环已停止。")
+
+
+    def start_updates(self, update_interval: float = 1.0):
+        """启动后台更新任务"""
+        if self.update_task is None or self.update_task.done():
+            self._stop_event.clear()
+            self.update_task = asyncio.create_task(self._run_update_loop(update_interval))
+            interest_logger.debug("后台兴趣更新任务已创建并启动。")
+        else:
+            interest_logger.debug("后台兴趣更新任务已在运行中。")
+
+
+    async def stop_updates(self):
+        """停止后台更新任务"""
+        if self.update_task and not self.update_task.done():
+            interest_logger.info("正在停止 InterestChatting 后台更新任务...")
+            self._stop_event.set() # 发送停止信号
+            try:
+                # 等待任务结束，设置超时
+                await asyncio.wait_for(self.update_task, timeout=5.0)
+                interest_logger.info("InterestChatting 后台更新任务已成功停止。")
+            except asyncio.TimeoutError:
+                interest_logger.warning("停止 InterestChatting 后台任务超时，尝试取消...")
+                self.update_task.cancel()
+                try:
+                    await self.update_task # 等待取消完成
+                except asyncio.CancelledError:
+                    interest_logger.info("InterestChatting 后台更新任务已被取消。")
+            except Exception as e:
+                interest_logger.error(f"停止 InterestChatting 后台任务时发生异常: {e}")
+            finally:
+                self.update_task = None
+        else:
+            interest_logger.debug("InterestChatting 后台更新任务未运行或已完成。")
+    # --- 结束 新增方法 ---
+
 
 class SubHeartflow:
     def __init__(self, subheartflow_id, mai_states: MaiStateInfo):
@@ -272,122 +293,163 @@ class SubHeartflow:
             request_type="sub_heart_flow",
         )
 
-        self.gpt_instance = ResponseGenerator()  # 响应生成器
-        self.tool_user_instance = ToolUser()  # 工具使用模块
-
         self.log_prefix = chat_manager.get_stream_name(self.subheartflow_id) or self.subheartflow_id
 
+    async def add_time_current_state(self, add_time: float):
+        self.current_state_time += add_time
+        
+    async def change_to_state_chat(self):
+        self.current_state_time = 120
+        self._start_normal_chat()
+    
+    async def change_to_state_focused(self):
+        self.current_state_time = 60
+        self._start_heart_fc_chat()
+
+    async def _stop_normal_chat(self):
+        """停止 NormalChat 的兴趣监控"""
+        if self.normal_chat_instance:
+            logger.info(f"{self.log_prefix} 停止 NormalChat 兴趣监控...")
+            try:
+                await self.normal_chat_instance.stop_chat() # 调用 stop_chat
+            except Exception as e:
+                logger.error(f"{self.log_prefix} 停止 NormalChat 监控任务时出错: {e}")
+                logger.error(traceback.format_exc())
+
+    async def _start_normal_chat(self) -> bool:
+        """启动 NormalChat 实例及其兴趣监控，确保 HeartFChatting 已停止"""
+        await self._stop_heart_fc_chat() # 确保专注聊天已停止
+
+        log_prefix = self.log_prefix
+        try:
+            # 总是尝试创建或获取最新的 stream 和 interest_dict
+            chat_stream = chat_manager.get_stream(self.chat_id)
+            if not chat_stream:
+                logger.error(f"{log_prefix} 无法获取 chat_stream，无法启动 NormalChat。")
+                return False
+
+            # 如果实例不存在或需要更新，则创建新实例
+            # if not self.normal_chat_instance: # 或者总是重新创建以获取最新的 interest_dict?
+            self.normal_chat_instance = NormalChat(chat_stream=chat_stream, interest_dict=self.get_interest_dict())
+            logger.info(f"{log_prefix} 创建或更新 NormalChat 实例。")
+
+            logger.info(f"{log_prefix} 启动 NormalChat 兴趣监控...")
+            await self.normal_chat_instance.start_chat() # <--- 修正：调用 start_chat
+            return True
+        except Exception as e:
+            logger.error(f"{log_prefix} 启动 NormalChat 时出错: {e}")
+            logger.error(traceback.format_exc())
+            self.normal_chat_instance = None # 启动失败，清理实例
+            return False
+
+    async def _stop_heart_fc_chat(self):
+        """停止并清理 HeartFChatting 实例"""
+        if self.heart_fc_instance:
+            logger.info(f"{self.log_prefix} 关闭 HeartFChatting 实例...")
+            try:
+                await self.heart_fc_instance.shutdown()
+            except Exception as e:
+                logger.error(f"{self.log_prefix} 关闭 HeartFChatting 实例时出错: {e}")
+                logger.error(traceback.format_exc())
+            finally:
+                # 无论是否成功关闭，都清理引用
+                self.heart_fc_instance = None
+
+    async def _start_heart_fc_chat(self) -> bool:
+        """启动 HeartFChatting 实例，确保 NormalChat 已停止"""
+        await self._stop_normal_chat() # 确保普通聊天监控已停止
+        self.clear_interest_dict() # 清理兴趣字典，准备专注聊天
+
+        log_prefix = self.log_prefix
+        if self.heart_fc_instance:
+            if not self.heart_fc_instance._loop_active:
+                logger.warning(f"{log_prefix} HeartFChatting 实例存在但未激活，尝试重新激活...")
+                await self.heart_fc_instance.add_time() # 尝试添加时间以激活循环
+                return True # 假设 add_time 会处理激活逻辑
+            else:
+                logger.debug(f"{log_prefix} HeartFChatting 已在运行中。")
+                return True # 已经在运行
+
+        logger.info(f"{log_prefix} 麦麦准备开始专注聊天...")
+        try:
+            self.heart_fc_instance = HeartFChatting(
+                chat_id=self.chat_id,
+            )
+            if await self.heart_fc_instance._initialize():
+                await self.heart_fc_instance.add_time()  # 初始化成功后添加初始时间
+                logger.info(f"{log_prefix} 麦麦已成功进入专注聊天模式。")
+                return True
+            else:
+                logger.error(f"{log_prefix} HeartFChatting 初始化失败，无法进入专注模式。")
+                self.heart_fc_instance = None # 初始化失败，清理实例
+                return False
+        except Exception as e:
+            logger.error(f"{log_prefix} 创建或初始化 HeartFChatting 实例时出错: {e}")
+            logger.error(traceback.format_exc())
+            self.heart_fc_instance = None # 创建或初始化异常，清理实例
+            return False
+
     async def set_chat_state(self, new_state: "ChatState", current_states_num: tuple = ()):
         """更新sub_heartflow的聊天状态，并管理 HeartFChatting 和 NormalChat 实例及任务"""
-
         current_state = self.chat_state.chat_status
         if current_state == new_state:
-            logger.trace(f"{self.log_prefix} 状态已为 {current_state.value}, 无需更改。")
+            # logger.trace(f"{self.log_prefix} 状态已为 {current_state.value}, 无需更改。") # 减少日志噪音
             return
 
-        log_prefix = self.log_prefix  # 使用实例属性
+        log_prefix = self.log_prefix
         current_mai_state = self.mai_states.get_current_state()
+        state_changed = False # 标记状态是否实际发生改变
 
         # --- 状态转换逻辑 ---
         if new_state == ChatState.CHAT:
             normal_limit = current_mai_state.get_normal_chat_max_num()
-            current_chat_count = current_states_num[1]
+            current_chat_count = current_states_num[1] if len(current_states_num) > 1 else 0
 
-            if current_chat_count >= normal_limit and current_state != ChatState.CHAT:  # 仅在状态转换时检查限制
+            if current_chat_count >= normal_limit and current_state != ChatState.CHAT:
-                logger.debug(
+                logger.debug(f"{log_prefix} 无法从 {current_state.value} 转到 聊天。原因：聊不过来了 ({current_chat_count}/{normal_limit})")
-                    f"{log_prefix} 麦麦不能从 {current_state.value} 转换到 聊天。原因：聊不过来了 ({current_chat_count}/{normal_limit})"
+                return # 阻止状态转换
-                )
-                return  # 阻止状态转换
             else:
-                logger.debug(f"{log_prefix} 麦麦可以进入或保持 聊天 状态 ({current_chat_count}/{normal_limit})")
+                logger.debug(f"{log_prefix} 准备进入或保持 聊天 状态 ({current_chat_count}/{normal_limit})")
-                if current_state == ChatState.FOCUSED and self.heart_fc_instance:
+                if await self._start_normal_chat():
-                    logger.info(f"{log_prefix} 麦麦不再专注聊天，转为随便水水...")
+                    logger.info(f"{log_prefix} 成功进入或保持 NormalChat 状态。")
-                    await self.heart_fc_instance.shutdown()  # 正确关闭 HeartFChatting
+                    state_changed = True
-                    self.heart_fc_instance = None
+                else:
-
+                    logger.error(f"{log_prefix} 启动 NormalChat 失败，无法进入 CHAT 状态。")
-                chat_stream = chat_manager.get_stream(self.chat_id)
+                    # 考虑是否需要回滚状态或采取其他措施
-                self.normal_chat_instance = NormalChat(chat_stream=chat_stream, interest_dict=self.get_interest_dict())
+                    return # 启动失败，不改变状态
-                await self.normal_chat_instance.start_monitoring_interest()
-                # NormalChat 启动/停止逻辑将在下面处理
 
         elif new_state == ChatState.FOCUSED:
             focused_limit = current_mai_state.get_focused_chat_max_num()
-            current_focused_count = current_states_num[2]
+            current_focused_count = current_states_num[2] if len(current_states_num) > 2 else 0
             
-            if current_focused_count >= focused_limit and current_state != ChatState.FOCUSED:  # 仅在状态转换时检查限制
+            if current_focused_count >= focused_limit and current_state != ChatState.FOCUSED:
-                logger.debug(
+                logger.debug(f"{log_prefix} 无法从 {current_state.value} 转到 专注。原因：聊不过来了 ({current_focused_count}/{focused_limit})")
-                    f"{log_prefix} 麦麦不能从 {current_state.value} 转换到 专注的聊天，原因：聊不过来了。({current_focused_count}/{focused_limit})"
+                return # 阻止状态转换
-                )
-                return  # 阻止状态转换
             else:
-                logger.debug(f"{log_prefix} 麦麦可以进入或保持 专注聊天 状态 ({current_focused_count}/{focused_limit})")
+                logger.debug(f"{log_prefix} 准备进入或保持 专注聊天 状态 ({current_focused_count}/{focused_limit})")
-                if not self.heart_fc_instance:
+                if await self._start_heart_fc_chat():
-                    logger.info(f"{log_prefix} 麦麦准备开始专注聊天...")
+                    logger.info(f"{log_prefix} 成功进入或保持 HeartFChatting 状态。")
-                    try:
+                    state_changed = True
-                        await self.normal_chat_instance.stop_monitoring_interest()
-                        self.clear_interest_dict()
-
-                        logger.info(f"{log_prefix} 停止 NormalChat 兴趣监控成功。")
-                    except Exception as e:
-                        logger.error(f"{log_prefix} 停止 NormalChat 兴趣监控时出错: {e}")
-                        logger.error(traceback.format_exc())
-                    try:
-                        self.heart_fc_instance = HeartFChatting(
-                            chat_id=self.chat_id,
-                            gpt_instance=self.gpt_instance,
-                            tool_user_instance=self.tool_user_instance,
-                        )
-                        if await self.heart_fc_instance._initialize():
-                            await self.heart_fc_instance.add_time()  # 初始化成功后添加初始时间
-                            logger.info(f"{log_prefix} 麦麦已成功进入专注聊天模式。")
-                        else:
-                            logger.error(
-                                f"{log_prefix} 麦麦不能专注聊天，因为 HeartFChatting 初始化失败了，状态回滚到 {current_state.value}"
-                            )
-                            self.heart_fc_instance = None
-                            return  # 阻止进入 FOCUSED 状态
-
-                    except Exception as e:
-                        logger.error(f"{log_prefix} 创建麦麦专注聊天实例时出错: {e}")
-                        logger.error(traceback.format_exc())
-                        self.heart_fc_instance = None
-                        return  # 创建实例异常，阻止进入 FOCUSED 状态
-
                 else:
-                    # 已经是 FOCUSED 状态，或者 heart_fc_instance 已存在但未运行（不太可能）
+                    logger.error(f"{log_prefix} 启动 HeartFChatting 失败，无法进入 FOCUSED 状态。")
-                    if not self.heart_fc_instance._loop_active:
+                    # 启动失败，状态回滚到之前的状态或ABSENT？这里保持不改变
-                        logger.warning(f"{log_prefix} HeartFChatting 实例存在但未激活，尝试重新激活...")
+                    return # 启动失败，不改变状态
-                        await self.heart_fc_instance.add_time()  # 尝试添加时间以激活循环
+
-                    else:
-                        logger.debug(f"{log_prefix} 麦麦已经在专注聊天中。")
-                # NormalChat 启动/停止逻辑将在下面处理
 
         elif new_state == ChatState.ABSENT:
-            if current_state == ChatState.FOCUSED and self.heart_fc_instance:
+            logger.info(f"{log_prefix} 进入 ABSENT 状态，停止所有聊天活动...")
-                logger.info(f"{log_prefix} 麦麦离开专注的聊天，撤退了.....")
+            await self._stop_normal_chat()
-                await self.heart_fc_instance.shutdown()  # 正确关闭 HeartFChatting
+            await self._stop_heart_fc_chat()
-                self.heart_fc_instance = None
+            state_changed = True # 总是可以成功转换到 ABSENT
-            # NormalChat 启动/停止逻辑将在下面处理
 
-        # --- 更新状态和最后活动时间 (先更新状态，再根据新状态管理任务)---
+        # --- 更新状态和最后活动时间 ---
-        self.chat_state.chat_status = new_state
+        if state_changed:
-        self.last_active_time = time.time()
+            logger.info(f"{log_prefix} 麦麦的聊天状态从 {current_state.value} 变更为 {new_state.value}")
-        logger.info(f"{log_prefix} 麦麦的聊天状态从 {current_state.value} 变更为 {new_state.value}")
+            self.chat_state.chat_status = new_state
-
+            self.last_active_time = time.time()
-        # --- 根据新的状态管理 NormalChat 的监控任务 ---
-        if self.normal_chat_instance:
-            try:
-                if new_state == ChatState.ABSENT:
-                    logger.info(f"{log_prefix} 状态变为 ABSENT，停止 NormalChat 兴趣监控...")
-                    await self.normal_chat_instance.stop_monitoring_interest()
-                else:  # CHAT or FOCUSED
-                    logger.info(f"{log_prefix} 状态变为 {new_state.value}，启动或确认 NormalChat 兴趣监控...")
-                    await self.normal_chat_instance.start_monitoring_interest()
-            except Exception as e:
-                logger.error(f"{log_prefix} 管理 NormalChat 监控任务时出错 (新状态: {new_state.value}): {e}")
-                logger.error(traceback.format_exc())
         else:
-            logger.warning(f"{log_prefix} NormalChat 实例不可用，无法管理其监控任务。")
+            # 如果因为某些原因（如启动失败）没有成功改变状态，记录一下
+            logger.debug(f"{log_prefix} 尝试将状态从 {current_state.value} 变为 {new_state.value}，但未成功或未执行更改。")
 
     async def subheartflow_start_working(self):
         """启动子心流的后台任务
@@ -399,9 +461,9 @@ class SubHeartflow:
         logger.info(f"{self.log_prefix} 子心流开始工作...")
 
         while not self.should_stop:
-            # 主循环保持简单，只做状态检查
             await asyncio.sleep(30)  # 30秒检查一次停止标志
             
+
         logger.info(f"{self.log_prefix} 子心流后台任务已停止。")
 
     async def do_thinking_before_reply(
@@ -543,6 +605,16 @@ class SubHeartflow:
     def clear_interest_dict(self):
         self.interest_chatting.interest_dict.clear()
 
+    async def get_full_state(self) -> dict:
+        """获取子心流的完整状态，包括兴趣、思维和聊天状态。"""
+        interest_state = await self.get_interest_state()
+        return {
+            "interest_state": interest_state,
+            "current_mind": self.current_mind,
+            "chat_state": self.chat_state.chat_status.value,
+            "last_active_time": self.last_active_time,
+        }
+
     async def shutdown(self):
         """安全地关闭子心流及其管理的任务"""
         if self.should_stop:
@@ -552,24 +624,14 @@ class SubHeartflow:
         logger.info(f"{self.log_prefix} 开始关闭子心流...")
         self.should_stop = True  # 标记为停止，让后台任务退出
 
-        # 停止 NormalChat 监控 (保持调用，确保清理)
+        # 使用新的停止方法
-        if self.normal_chat_instance:
+        await self._stop_normal_chat()
-            logger.info(f"{self.log_prefix} 停止 NormalChat 监控任务 (Shutdown)...")
+        await self._stop_heart_fc_chat()
-            try:
-                await self.normal_chat_instance.stop_monitoring_interest()
-            except Exception as e:
-                logger.error(f"{self.log_prefix} 停止 NormalChat 监控任务时出错 (Shutdown): {e}")
-                logger.error(traceback.format_exc())
 
-        # 停止 HeartFChatting (如果存在且正在运行)
+        # 停止兴趣更新任务
-        if self.heart_fc_instance:
+        if self.interest_chatting:
-            logger.info(f"{self.log_prefix} 关闭 HeartFChatting 实例 (Shutdown)...")
+            logger.info(f"{self.log_prefix} 停止兴趣系统后台任务...")
-            try:
+            await self.interest_chatting.stop_updates()
-                await self.heart_fc_instance.shutdown()
-            except Exception as e:
-                logger.error(f"{self.log_prefix} 关闭 HeartFChatting 实例时出错 (Shutdown): {e}")
-                logger.error(traceback.format_exc())
-            self.heart_fc_instance = None  # 清理实例引用
 
         # 取消可能存在的旧后台任务 (self.task)
         if self.task and not self.task.done():

src/heart_flow/subheartflow_manager.py

@@ -257,42 +257,89 @@ class SubHeartflowManager:
 
         logger.info(f"[停用] 完成, 尝试停止{len(flow_ids)}个, 成功{stopped_count}个")
 
-    async def evaluate_interest_and_promote(self):
+    async def evaluate_interest_and_promote(self, current_mai_state: MaiStateInfo):
-        """评估CHAT状态的子心流兴趣度，满足条件则提升到FOCUSED状态"""
+        """评估子心流兴趣度，满足条件且未达上限则提升到FOCUSED状态（基于start_hfc_probability）"""
-        logger.debug("[子心流管理器] 开始兴趣评估周期...")
+        log_prefix_manager = "[子心流管理器-兴趣评估]"
-        evaluated_count = 0
+        logger.debug(f"{log_prefix_manager} 开始周期... 当前状态: {current_mai_state.get_current_state().value}")
-        promoted_count = 0
+
+        # 获取 FOCUSED 状态的数量上限
+        current_state_enum = current_mai_state.get_current_state()
+        focused_limit = current_state_enum.get_focused_chat_max_num()
+        if focused_limit <= 0:
+            logger.debug(f"{log_prefix_manager} 当前状态 ({current_state_enum.value}) 不允许 FOCUSED 子心流, 跳过提升检查。")
+            return
+
+        # 获取当前 FOCUSED 状态的数量 (初始值)
+        current_focused_count = self.count_subflows_by_state(ChatState.FOCUSED)
+        logger.debug(f"{log_prefix_manager} 专注上限: {focused_limit}, 当前专注数: {current_focused_count}")
 
         # 使用快照安全遍历
         subflows_snapshot = list(self.subheartflows.values())
-
+        promoted_count = 0 # 记录本次提升的数量
-        for sub_hf in subflows_snapshot:
+        try:
-            flow_id = sub_hf.subheartflow_id
+            for sub_hf in subflows_snapshot:
-            if flow_id in self.subheartflows and self.subheartflows[flow_id].chat_state.chat_status == ChatState.CHAT:
+                flow_id = sub_hf.subheartflow_id
-                evaluated_count += 1
                 stream_name = chat_manager.get_stream_name(flow_id) or flow_id
-                log_prefix = f"[{stream_name}]"
+                log_prefix_flow = f"[{stream_name}]"
 
-                should_promote = await sub_hf.should_evaluate_reply()
+                # 只处理 CHAT 状态的子心流
-                if should_promote:
+                if sub_hf.chat_state.chat_status != ChatState.CHAT:
-                    logger.info(f"{log_prefix} 兴趣评估触发升级: CHAT -> FOCUSED")
+                    continue
-                    states_num = (
+
-                        self.count_subflows_by_state(ChatState.ABSENT),
+                # 检查是否满足提升概率
-                        self.count_subflows_by_state(ChatState.CHAT),
+                should_hfc = random.random() < sub_hf.interest_chatting.start_hfc_probability
-                        self.count_subflows_by_state(ChatState.FOCUSED),
+                if not should_hfc:
-                    )
+                    continue
-                    await sub_hf.set_chat_state(ChatState.FOCUSED, states_num)
+
-                    if (
+                # --- 关键检查：检查 FOCUSED 数量是否已达上限 ---
-                        self.subheartflows.get(flow_id)
+                # 注意：在循环内部再次获取当前数量，因为之前的提升可能已经改变了计数
-                        and self.subheartflows[flow_id].chat_state.chat_status == ChatState.FOCUSED
+                # 使用已经记录并在循环中更新的 current_focused_count
-                    ):
+                if current_focused_count >= focused_limit:
+                    logger.debug(f"{log_prefix_manager} {log_prefix_flow} 达到专注上限 ({current_focused_count}/{focused_limit}), 无法提升。概率={sub_hf.interest_chatting.start_hfc_probability:.2f}")
+                    continue # 跳过这个子心流，继续检查下一个
+
+                # --- 执行提升 ---
+                # 获取当前实例以检查最新状态 (防御性编程)
+                current_subflow = self.subheartflows.get(flow_id)
+                if not current_subflow or current_subflow.chat_state.chat_status != ChatState.CHAT:
+                    logger.warning(f"{log_prefix_manager} {log_prefix_flow} 尝试提升时状态已改变或实例消失，跳过。")
+                    continue
+
+                logger.info(f"{log_prefix_manager} {log_prefix_flow} 兴趣评估触发升级 (prob={sub_hf.interest_chatting.start_hfc_probability:.2f}, 上限:{focused_limit}, 当前:{current_focused_count}) -> FOCUSED")
+
+                states_num = (
+                    self.count_subflows_by_state(ChatState.ABSENT),
+                    self.count_subflows_by_state(ChatState.CHAT), # 这个值在提升前计算
+                    current_focused_count, # 这个值在提升前计算
+                )
+
+                # --- 状态设置 ---
+                original_state = current_subflow.chat_state.chat_status # 记录原始状态
+                await current_subflow.set_chat_state(ChatState.FOCUSED, states_num)
+
+                # --- 状态验证 ---
+                final_subflow = self.subheartflows.get(flow_id)
+                if final_subflow:
+                    final_state = final_subflow.chat_state.chat_status
+                    if final_state == ChatState.FOCUSED:
+                        logger.debug(f"{log_prefix_manager} {log_prefix_flow} 成功从 {original_state.value} 升级到 FOCUSED 状态")
                         promoted_count += 1
-                        logger.debug(f"{log_prefix} 成功升级到FOCUSED状态")
+                        # 提升成功后，更新当前专注计数，以便后续检查能使用最新值
-                    else:
+                        current_focused_count += 1
-                        logger.info(f"{log_prefix} 升级FOCUSED可能被限制阻止")
+                    elif final_state == original_state: # 状态未变
+                        logger.warning(f"{log_prefix_manager} {log_prefix_flow} 尝试从 {original_state.value} 升级 FOCUSED 失败，状态仍为: {final_state.value} (可能被内部逻辑阻止)")
+                    else: # 状态变成其他了?
+                        logger.warning(f"{log_prefix_manager} {log_prefix_flow} 尝试从 {original_state.value} 升级 FOCUSED 后状态变为 {final_state.value}")
+                else: # 子心流消失了?
+                    logger.warning(f"{log_prefix_manager} {log_prefix_flow} 升级后验证时子心流 {flow_id} 消失")
 
-        if evaluated_count > 0:
+        except Exception as e:
-            logger.debug(f"[子心流管理器] 评估完成. 评估{evaluated_count}个CHAT流, 升级{promoted_count}个到FOCUSED")
+            logger.error(f"{log_prefix_manager} 兴趣评估周期出错: {e}", exc_info=True)
+
+        if promoted_count > 0:
+            logger.info(f"{log_prefix_manager} 评估周期结束, 成功提升 {promoted_count} 个子心流到 FOCUSED。")
+        else:
+            logger.debug(f"{log_prefix_manager} 评估周期结束, 未提升任何子心流。")
 
     def count_subflows_by_state(self, state: ChatState) -> int:
         """统计指定状态的子心流数量

src/plugins/chat/bot.py

@@ -3,7 +3,7 @@ from ...config.config import global_config
 from .message import MessageRecv
 from ..PFC.pfc_manager import PFCManager
 from .chat_stream import chat_manager
-from ..chat_module.only_process.only_message_process import MessageProcessor
+from .only_message_process import MessageProcessor
 
 from src.common.logger import get_module_logger, CHAT_STYLE_CONFIG, LogConfig
 from ..heartFC_chat.heartflow_processor import HeartFCProcessor

src/plugins/heartFC_chat/heartFC_chat.py

@@ -13,9 +13,7 @@ from src.plugins.models.utils_model import LLMRequest
 from src.config.config import global_config
 from src.plugins.chat.utils_image import image_path_to_base64  # Local import needed after move
 from src.plugins.utils.timer_calculater import Timer  # <--- Import Timer
-
+from src.plugins.heartFC_chat.heartFC_generator import HeartFCGenerator
-# --- Import necessary dependencies directly ---
-from .heartFC_generator import ResponseGenerator  # Assuming this is the type for gpt
 from src.do_tool.tool_use import ToolUser
 from ..chat.message_sender import message_manager  # <-- Import the global manager
 from src.plugins.chat.emoji_manager import emoji_manager
@@ -77,9 +75,7 @@ class HeartFChatting:
 
     def __init__(
         self,
-        chat_id: str,
+        chat_id: str
-        gpt_instance: ResponseGenerator,  # 文本回复生成器
-        tool_user_instance: ToolUser,  # 工具使用实例
     ):
         """
         HeartFChatting 初始化函数
@@ -97,13 +93,12 @@ class HeartFChatting:
 
         # 初始化状态控制
         self._initialized = False  # 是否已初始化标志
-        self._init_lock = asyncio.Lock()  # 初始化锁(确保只初始化一次)
         self._processing_lock = asyncio.Lock()  # 处理锁(确保单次Plan-Replier-Sender周期)
         self._timer_lock = asyncio.Lock()  # 计时器锁(安全更新计时器)
 
         # 依赖注入存储
-        self.gpt_instance = gpt_instance  # 文本回复生成器
+        self.gpt_instance = HeartFCGenerator()  # 文本回复生成器
-        self.tool_user = tool_user_instance  # 工具使用实例
+        self.tool_user = ToolUser()  # 工具使用实例
 
         # LLM规划器配置
         self.planner_llm = LLMRequest(
@@ -117,7 +112,6 @@ class HeartFChatting:
         self._loop_timer: float = 0.0  # 循环剩余时间(秒)
         self._loop_active: bool = False  # 循环是否正在运行
         self._loop_task: Optional[asyncio.Task] = None  # 主循环任务
-        self._trigger_count_this_activation: int = 0  # 当前激活周期内的触发计数
         self._initial_duration: float = INITIAL_DURATION  # 首次触发增加的时间
         self._last_added_duration: float = self._initial_duration  # 上次增加的时间
 
@@ -131,35 +125,34 @@ class HeartFChatting:
         懒初始化以使用提供的标识符解析chat_stream和sub_hf。
         确保实例已准备好处理触发器。
         """
-        async with self._init_lock:
+        if self._initialized:
-            if self._initialized:
+            return True
-                return True
+        log_prefix = self._get_log_prefix()  # 获取前缀
-            log_prefix = self._get_log_prefix()  # 获取前缀
+        try:
-            try:
+            self.chat_stream = chat_manager.get_stream(self.stream_id)
-                self.chat_stream = chat_manager.get_stream(self.stream_id)
 
-                if not self.chat_stream:
+            if not self.chat_stream:
-                    logger.error(f"{log_prefix} 获取ChatStream失败。")
+                logger.error(f"{log_prefix} 获取ChatStream失败。")
-                    return False
-
-                # <-- 在这里导入 heartflow 实例
-                from src.heart_flow.heartflow import heartflow
-
-                self.sub_hf = heartflow.get_subheartflow(self.stream_id)
-                if not self.sub_hf:
-                    logger.warning(f"{log_prefix} 获取SubHeartflow失败。一些功能可能受限。")
-
-                self._initialized = True
-                logger.info(f"麦麦感觉到了，激发了HeartFChatting{log_prefix} 初始化成功。")
-                return True
-            except Exception as e:
-                logger.error(f"{log_prefix} 初始化失败: {e}")
-                logger.error(traceback.format_exc())
                 return False
 
+            # <-- 在这里导入 heartflow 实例
+            from src.heart_flow.heartflow import heartflow
+
+            self.sub_hf = heartflow.get_subheartflow(self.stream_id)
+            if not self.sub_hf:
+                logger.warning(f"{log_prefix} 获取SubHeartflow失败。一些功能可能受限。")
+
+            self._initialized = True
+            logger.info(f"麦麦感觉到了，激发了HeartFChatting{log_prefix} 初始化成功。")
+            return True
+        except Exception as e:
+            logger.error(f"{log_prefix} 初始化失败: {e}")
+            logger.error(traceback.format_exc())
+            return False
+
     async def add_time(self):
         """
-        为麦麦添加时间，麦麦有兴趣时，时间增加。
+        为麦麦添加时间，麦麦有兴趣时，固定增加15秒
         """
         log_prefix = self._get_log_prefix()
         if not self._initialized:
@@ -168,45 +161,61 @@ class HeartFChatting:
                 return
 
         async with self._timer_lock:
-            duration_to_add: float = 0.0
+            duration_to_add: float = 15.0  # 固定增加15秒
+            if not self._loop_active:  # 首次触发
+                logger.info(f"{log_prefix} 麦麦有兴趣！ 打算聊：15s.")
+            else:  # 循环已激活
+                logger.info(f"{log_prefix} 麦麦想继续聊：15s, 还能聊： {self._loop_timer:.1f}s.")
 
-            if not self._loop_active:  # First trigger for this activation cycle
+            # 添加固定时间
-                duration_to_add = self._initial_duration  # 使用初始值
-                self._last_added_duration = duration_to_add  # 更新上次增加的值
-                self._trigger_count_this_activation = 1  # Start counting
-                logger.info(
-                    f"{log_prefix} 麦麦有兴趣！ #{self._trigger_count_this_activation}. 麦麦打算聊： {duration_to_add:.2f}s."
-                )
-            else:  # Loop is already active, apply 50% reduction
-                self._trigger_count_this_activation += 1
-                duration_to_add = self._last_added_duration * 0.5
-                if duration_to_add < 1.5:
-                    duration_to_add = 1.5
-                # Update _last_added_duration only if it's >= 0.5 to prevent it from becoming too small
-                self._last_added_duration = duration_to_add
-                logger.info(
-                    f"{log_prefix} 麦麦兴趣增加！ #{self._trigger_count_this_activation}. 想继续聊： {duration_to_add:.2f}s, 麦麦还能聊： {self._loop_timer:.1f}s."
-                )
-
-            # 添加计算出的时间
             new_timer_value = self._loop_timer + duration_to_add
-            # Add max timer duration limit? e.g., max(0, min(new_timer_value, 300))
             self._loop_timer = max(0, new_timer_value)
-            # Log less frequently, e.g., every 10 seconds or significant change?
-            # if self._trigger_count_this_activation % 5 == 0:
-            # logger.info(f"{log_prefix} 麦麦现在想聊{self._loop_timer:.1f}秒")
 
-            # Start the loop if it wasn't active and timer is positive
+        # 添加时间后，检查是否需要启动循环
+        await self._start_loop_if_needed()
+
+    async def start(self):
+        """
+        显式尝试启动 HeartFChatting 的主循环。
+        如果循环未激活且计时器 > 0，则启动循环。
+        """
+        log_prefix = self._get_log_prefix()
+        if not self._initialized:
+            if not await self._initialize():
+                logger.error(f"{log_prefix} 无法启动循环: 初始化失败。")
+                return
+        logger.info(f"{log_prefix} 尝试显式启动循环...")
+        await self._start_loop_if_needed()
+
+    async def _start_loop_if_needed(self):
+        """检查是否需要启动主循环，如果未激活且计时器大于0，则启动。"""
+        log_prefix = self._get_log_prefix()
+        should_start_loop = False
+        async with self._timer_lock:
+            # 检查是否满足启动条件：未激活且计时器有时间
             if not self._loop_active and self._loop_timer > 0:
-                self._loop_active = True
+                should_start_loop = True
-                if self._loop_task and not self._loop_task.done():
+                self._loop_active = True # 在锁内标记为活动，防止重复启动
-                    logger.warning(f"{log_prefix} 发现意外的循环任务正在进行。取消它。")
-                    self._loop_task.cancel()
 
-                self._loop_task = asyncio.create_task(self._run_pf_loop())
+        if should_start_loop:
-                self._loop_task.add_done_callback(self._handle_loop_completion)
+            # 检查是否已有任务在运行（理论上不应该，因为 _loop_active=False）
-            elif self._loop_active:
+            if self._loop_task and not self._loop_task.done():
-                logger.trace(f"{log_prefix} 循环已经激活。计时器延长。")
+                logger.warning(f"{log_prefix} 发现之前的循环任务仍在运行（不符合预期）。取消旧任务。")
+                self._loop_task.cancel()
+                try:
+                    # 等待旧任务确实被取消
+                    await asyncio.wait_for(self._loop_task, timeout=0.5)
+                except (asyncio.CancelledError, asyncio.TimeoutError):
+                    pass # 忽略取消或超时错误
+                self._loop_task = None # 清理旧任务引用
+
+            logger.info(f"{log_prefix} 计时器 > 0 且循环未激活，启动主循环...")
+            # 创建新的循环任务
+            self._loop_task = asyncio.create_task(self._run_pf_loop())
+            # 添加完成回调
+            self._loop_task.add_done_callback(self._handle_loop_completion)
+        # else:
+            # logger.trace(f"{log_prefix} 不需要启动循环（已激活或计时器为0）") # 可以取消注释以进行调试
 
     def _handle_loop_completion(self, task: asyncio.Task):
         """当 _run_pf_loop 任务完成时执行的回调。"""
@@ -228,8 +237,6 @@ class HeartFChatting:
             if self._processing_lock.locked():
                 logger.warning(f"{log_prefix} HeartFChatting: 处理锁在循环结束时仍被锁定，强制释放。")
                 self._processing_lock.release()
-            # Instance removal is now handled by SubHeartflow
-            # asyncio.create_task(self.heartfc_controller._remove_heartFC_chat_instance(self.stream_id)) # Removed
 
     async def _run_pf_loop(self):
         """

src/plugins/heartFC_chat/heartFC_generator.py

@@ -22,7 +22,7 @@ llm_config = LogConfig(
 logger = get_module_logger("llm_generator", config=llm_config)
 
 
-class ResponseGenerator:
+class HeartFCGenerator:
     def __init__(self):
         self.model_normal = LLMRequest(
             model=global_config.llm_normal,

src/plugins/heartFC_chat/heartflow_processor.py

@@ -67,11 +67,8 @@ class HeartFCProcessor:
                 group_info=groupinfo,
             )
 
-            # --- 确保 SubHeartflow 存在 ---
             subheartflow = await heartflow.create_subheartflow(chat.stream_id)
-            if not subheartflow:
+
-                logger.error(f"无法为 stream_id {chat.stream_id} 创建或获取 SubHeartflow，中止处理")
-                return
 
             message.update_chat_stream(chat)
 
@@ -137,33 +134,16 @@ class HeartFCProcessor:
 
             # --- 修改：兴趣度更新逻辑 --- #
             if is_mentioned:
-                interest_increase_on_mention = 2
+                interest_increase_on_mention = 1
                 mentioned_boost = interest_increase_on_mention  # 从配置获取提及增加值
                 interested_rate += mentioned_boost
-                logger.trace(f"消息提及机器人，额外增加兴趣 {mentioned_boost:.2f}")
 
             # 更新兴趣度 (调用 SubHeartflow 的方法)
-            current_interest = 0.0  # 初始化
+            current_time = time.time()
-            try:
+            await subheartflow.interest_chatting.increase_interest(current_time, value=interested_rate)
-                # 获取当前时间，传递给 increase_interest
-                current_time = time.time()
-                await subheartflow.interest_chatting.increase_interest(current_time, value=interested_rate)
-                current_interest = await subheartflow.get_interest_level()  # 获取更新后的值
 
-                logger.trace(
+            # 添加到 SubHeartflow 的 interest_dict，给normal_chat处理
-                    f"使用激活率 {interested_rate:.2f} 更新后 (通过缓冲后)，当前兴趣度: {current_interest:.2f} (Stream: {chat.stream_id})"
+            await subheartflow.add_interest_dict_entry(message, interested_rate, is_mentioned)
-                )
-
-                # 添加到 SubHeartflow 的 interest_dict
-                await subheartflow.add_interest_dict_entry(message, interested_rate, is_mentioned)
-                logger.trace(
-                    f"Message {message.message_info.message_id} added to interest dict for stream {chat.stream_id}"
-                )
-
-            except Exception as e:
-                logger.error(f"更新兴趣度失败 (Stream: {chat.stream_id}): {e}")
-                logger.error(traceback.format_exc())
-            # --- 结束修改 --- #
 
             # 打印消息接收和处理信息
             mes_name = chat.group_info.group_name if chat.group_info else "私聊"
@@ -172,7 +152,7 @@ class HeartFCProcessor:
                 f"[{current_time}][{mes_name}]"
                 f"{chat.user_info.user_nickname}:"
                 f"{message.processed_plain_text}"
-                f"兴趣度: {current_interest:.2f}"
+                f"[兴趣度: {interested_rate:.2f}]"
             )
 
             try:

src/plugins/heartFC_chat/normal_chat.py

@@ -7,7 +7,7 @@ from typing import List, Optional  # 导入 Optional
 from ..moods.moods import MoodManager
 from ...config.config import global_config
 from ..chat.emoji_manager import emoji_manager
-from .normal_chat_generator import ResponseGenerator
+from .normal_chat_generator import NormalChatGenerator
 from ..chat.message import MessageSending, MessageRecv, MessageThinking, MessageSet
 from ..chat.message_sender import message_manager
 from ..chat.utils_image import image_path_to_base64
@@ -43,12 +43,10 @@ class NormalChat:
 
         self.interest_dict = interest_dict
 
-        logger.info(f"[{self.stream_name}] 正在初始化 NormalChat 实例...")
+        self.gpt = NormalChatGenerator()
-
-        self.gpt = ResponseGenerator()
         self.mood_manager = MoodManager.get_instance()  # MoodManager 保持单例
         # 存储此实例的兴趣监控任务
-        self._interest_monitoring_task: Optional[asyncio.Task] = None
+        self._chat_task: Optional[asyncio.Task] = None
         logger.info(f"[{self.stream_name}] NormalChat 实例初始化完成。")
 
     # 改为实例方法
@@ -73,7 +71,6 @@ class NormalChat:
         )
 
         await message_manager.add_message(thinking_message)
-
         return thinking_id
 
     # 改为实例方法
@@ -176,7 +173,7 @@ class NormalChat:
             await asyncio.sleep(1)  # 每秒检查一次
 
             # 检查任务是否已被取消
-            if self._interest_monitoring_task is None or self._interest_monitoring_task.cancelled():
+            if self._chat_task is None or self._chat_task.cancelled():
                 logger.info(f"[{self.stream_name}] 兴趣监控任务被取消或置空，退出")
                 break
 
@@ -352,19 +349,21 @@ class NormalChat:
         return False
 
     # 改为实例方法, 移除 chat 参数
-    async def start_monitoring_interest(self):
+    
-        """为此 NormalChat 实例关联的 ChatStream 启动兴趣消息监控任务（如果尚未运行）。"""
+    async def start_chat(self):
-        if self._interest_monitoring_task is None or self._interest_monitoring_task.done():
+        """为此 NormalChat 实例关联的 ChatStream 启动聊天任务（如果尚未运行）。"""
-            logger.info(f"[{self.stream_name}] 启动兴趣消息监控任务...")
+        if self._chat_task is None or self._chat_task.done():
+            logger.info(f"[{self.stream_name}] 启动聊天任务...")
             task = asyncio.create_task(self._find_interested_message())
             task.add_done_callback(lambda t: self._handle_task_completion(t))  # 回调现在是实例方法
-            self._interest_monitoring_task = task
+            self._chat_task = task
+
 
     # 改为实例方法, 移除 stream_id 参数
     def _handle_task_completion(self, task: asyncio.Task):
         """兴趣监控任务完成时的回调函数。"""
         # 检查完成的任务是否是当前实例的任务
-        if task is not self._interest_monitoring_task:
+        if task is not self._chat_task:
             logger.warning(f"[{self.stream_name}] 收到一个未知或过时任务的完成回调。")
             return
 
@@ -382,27 +381,26 @@ class NormalChat:
             logger.error(f"[{self.stream_name}] 处理任务完成回调时出错: {e}")
         finally:
             # 标记任务已完成/移除
-            if self._interest_monitoring_task is task:  # 再次确认是当前任务
+            if self._chat_task is task:  # 再次确认是当前任务
-                self._interest_monitoring_task = None
+                self._chat_task = None
-                logger.debug(f"[{self.stream_name}] 兴趣监控任务已被标记为完成/移除。")
+                logger.debug(f"[{self.stream_name}] 聊天任务已被标记为完成/移除。")
 
     # 改为实例方法, 移除 stream_id 参数
-    async def stop_monitoring_interest(self):
+    async def stop_chat(self):
         """停止当前实例的兴趣监控任务。"""
-        if self._interest_monitoring_task and not self._interest_monitoring_task.done():
+        if self._chat_task and not self._chat_task.done():
-            task = self._interest_monitoring_task
+            task = self._chat_task
-            logger.info(f"[{self.stream_name}] 尝试取消兴趣监控任务。")
+            logger.info(f"[{self.stream_name}] 尝试取消聊天任务。")
             task.cancel()
             try:
                 await task  # 等待任务响应取消
             except asyncio.CancelledError:
-                logger.info(f"[{self.stream_name}] 兴趣监控任务已成功取消。")
+                logger.info(f"[{self.stream_name}] 聊天任务已成功取消。")
             except Exception as e:
                 # 回调函数 _handle_task_completion 会处理异常日志
                 logger.warning(f"[{self.stream_name}] 等待监控任务取消时捕获到异常 (可能已在回调中记录): {e}")
             finally:
                 # 确保任务状态更新，即使等待出错 (回调函数也会尝试更新)
-                if self._interest_monitoring_task is task:
+                if self._chat_task is task:
-                    self._interest_monitoring_task = None
+                    self._chat_task = None
-        # else:
+
-        # logger.debug(f"[{self.stream_name}] 没有正在运行的兴趣监控任务可停止。")

src/plugins/heartFC_chat/normal_chat_generator.py

@@ -19,7 +19,7 @@ llm_config = LogConfig(
 logger = get_module_logger("llm_generator", config=llm_config)
 
 
-class ResponseGenerator:
+class NormalChatGenerator:
     def __init__(self):
         self.model_reasoning = LLMRequest(
             model=global_config.llm_reasoning,
@@ -77,8 +77,6 @@ class ResponseGenerator:
             sender_name = f"({message.chat_stream.user_info.user_id}){message.chat_stream.user_info.user_nickname}"
         else:
             sender_name = f"用户({message.chat_stream.user_info.user_id})"
-
-        logger.debug("开始使用生成回复-2")
         # 构建prompt
         with Timer() as t_build_prompt:
             prompt = await prompt_builder.build_prompt(