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("状态更新任务已在运行")
+        """启动所有后台任务
+        
+        功能说明:
+        - 启动核心后台任务: 状态更新、清理、日志记录和兴趣评估
+        - 每个任务启动前检查是否已在运行
+        - 将任务引用保存到任务列表
+        """
+        
+        # 任务配置列表: (任务变量名, 任务函数, 任务名称, 日志级别, 额外日志信息, 任务对象引用属性名)
+        task_configs = [
+            (self._state_update_task, 
+            lambda: self._run_state_update_cycle(self.update_interval),
+            "hf_state_update", 
+            "debug", 
+            f"聊天状态更新任务已启动 间隔:{self.update_interval}s",
+            "_state_update_task"),
+            
+            (self._cleanup_task,
+            self._run_cleanup_cycle,
+            "hf_cleanup",
+            "info",
+            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"),
 
-        # 清理任务
-        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("清理任务已在运行")
+            # 新增兴趣评估任务配置
+            (self._interest_eval_task,
+            self._run_interest_eval_cycle,
+            "hf_interest_eval",
+            "debug", # 设为debug，避免过多日志
+            f"兴趣评估任务已启动 间隔:{self.interest_eval_interval}s",
+            "_interest_eval_task"),
+        ]
 
-        # 日志任务
-        if self._logging_task is None or self._logging_task.done():
-            self._logging_task = asyncio.create_task(self._run_logging_cycle(), name="hf_logging")
-            self._tasks.append(self._logging_task)
-            logger.info(f"日志任务已启动 间隔:{self.log_interval}s")
-        else:
-            logger.warning("日志任务已在运行")
-
-        # # 初始状态检查
-        # initial_state = self.mai_state_info.get_current_state()
-        # if initial_state != self.mai_state_info.mai_status.OFFLINE:
-        #     logger.info(f"初始状态:{initial_state.value} 触发初始激活检查")
-        #     asyncio.create_task(self.subheartflow_manager.activate_random_subflows_to_chat(initial_state))
+        # 统一启动所有任务
+        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:  # 任务超时处理
-                logger.warning(f"任务 {task_name} 超时执行 ({elapsed:.2f}s > {interval}s)")
+            # if sleep_time < 0.1:  # 任务超时处理, DEBUG 时可能干扰断点
+            #     logger.warning(f"任务 {task_name} 超时执行 ({elapsed:.2f}s > {interval}s)")
             await asyncio.sleep(sleep_time)
 
-        # 非离线状态时评估兴趣
-        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}")
+        logger.debug(f"任务循环结束: {task_name}") # 调整日志信息
 
     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)
-            logger.info(f"已确保日志目录 '{LOG_DIRECTORY}' 存在")
-        except OSError as e:
-            logger.error(f"创建日志目录 '{LOG_DIRECTORY}' 出错: {e}")
+        os.makedirs(LOG_DIRECTORY, exist_ok=True)
+        logger.info(f"已确保日志目录 '{LOG_DIRECTORY}' 存在")
 
-    async def get_all_interest_states(self) -> Dict[str, Dict]:
-        """并发获取所有活跃子心流的当前兴趣状态。"""
-        _states = {}
-        # Get snapshot from the manager
+    async def get_all_subflow_states(self) -> Dict[str, Dict]:
+        """并发获取所有活跃子心流的当前完整状态。"""
         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 = stream_id_str
-                except IndexError:
-                    logger.error(f"无法从任务名 {task.get_name()} 中提取 stream_id")
-                    continue
+                stream_id_str = task.get_name().split("get_state_")[-1]
+                stream_id = stream_id_str
 
-                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):
-        """获取所有子心流的兴趣状态并写入日志文件。"""
-        # logger.debug("开始定期记录兴趣状态...")
+    async def log_all_states(self):
+        """获取主心流状态和所有子心流的完整状态并写入日志文件。"""
         try:
             current_timestamp = time.time()
-            all_interest_states = await self.get_all_interest_states()
 
-            if not all_interest_states:
-                logger.debug("没有获取到任何兴趣状态")
+            main_mind = self.heartflow.current_mind
+            # 获取 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
-            try:
-                with open(self._history_log_file_path, "a", encoding="utf-8") as f:
-                    items_snapshot = list(all_interest_states.items())
-                    for stream_id, state in items_snapshot:
-                        group_name = stream_id
-                        try:
-                            chat_stream = chat_manager.get_stream(stream_id)
-                            if chat_stream and chat_stream.group_info:
-                                group_name = chat_stream.group_info.group_name
-                            elif chat_stream and not chat_stream.group_info:
-                                group_name = (
-                                    f"私聊_{chat_stream.user_info.user_nickname}"
-                                    if chat_stream.user_info
-                                    else stream_id
-                                )
-                        except Exception as e:
-                            logger.trace(f"无法获取 stream_id {stream_id} 的群组名: {e}")
-                            pass
+            subflow_details = []
+            items_snapshot = list(all_subflow_states.items())
+            for stream_id, state in items_snapshot:
+                group_name = stream_id
+                try:
+                    chat_stream = chat_manager.get_stream(stream_id)
+                    if chat_stream:
+                        if chat_stream.group_info:
+                            group_name = chat_stream.group_info.group_name
+                        elif chat_stream.user_info:
+                            group_name = f"私聊_{chat_stream.user_info.user_nickname}"
+                except Exception as e:
+                    logger.trace(f"无法获取 stream_id {stream_id} 的群组名: {e}")
 
-                        log_entry = {
-                            "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}")
+                interest_state = state.get("interest_state", {})
 
+                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 = "不参与"
-    CHAT = "闲聊"
-    FOCUSED = "专注"
+    ABSENT = "没在看群"
+    CHAT = "随便水群"
+    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):
-        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
-
-    async def _update_reply_probability(self, current_time: float):
-        time_delta = current_time - self.last_update_time
-        if time_delta <= 0:
+    async def _calculate_decay(self):
+        """计算兴趣值的衰减
+        
+        参数:
+            current_time: 当前时间戳
+            
+        处理逻辑:
+        1. 计算时间差
+        2. 处理各种异常情况(负值/零值)
+        3. 正常计算衰减
+        4. 更新最后更新时间
+        """
+        
+        # 处理极小兴趣值情况
+        if self.interest_level < 1e-9:
+            self.interest_level = 0.0
             return
+            
+        # 异常情况处理
+        if self.decay_rate_per_second <= 0:
+            interest_logger.warning(f"衰减率({self.decay_rate_per_second})无效，重置兴趣值为0")
+            self.interest_level = 0.0
+            return
+            
+        # 正常衰减计算
+        try:
+            decay_factor = math.pow(self.decay_rate_per_second, self.update_interval)
+            self.interest_level *= decay_factor
+        except ValueError as e:
+            interest_logger.error(f"衰减计算错误: {e} 参数: 衰减率={self.decay_rate_per_second} 时间差={self.update_interval} 当前兴趣={self.interest_level}")
+            self.interest_level = 0.0
+            
 
-        currently_above = self.interest_level >= self.trigger_threshold
-        previous_is_above = self.is_above_threshold
-
-        if currently_above:
-            if not self.is_above_threshold:
-                self.current_reply_probability = self.base_reply_probability
-                interest_logger.debug(
-                    f"兴趣跨过阈值 ({self.trigger_threshold}). 概率重置为基础值: {self.base_reply_probability:.4f}"
-                )
-            else:
-                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.state_change_callback:
-                    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
+            if self.start_hfc_probability != 0:
+                self.start_hfc_probability -= 0.1
 
     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:  # 仅在状态转换时检查限制
-                logger.debug(
-                    f"{log_prefix} 麦麦不能从 {current_state.value} 转换到 聊天。原因：聊不过来了 ({current_chat_count}/{normal_limit})"
-                )
-                return  # 阻止状态转换
+            if current_chat_count >= normal_limit and current_state != ChatState.CHAT:
+                logger.debug(f"{log_prefix} 无法从 {current_state.value} 转到 聊天。原因：聊不过来了 ({current_chat_count}/{normal_limit})")
+                return # 阻止状态转换
             else:
-                logger.debug(f"{log_prefix} 麦麦可以进入或保持 聊天 状态 ({current_chat_count}/{normal_limit})")
-                if current_state == ChatState.FOCUSED and self.heart_fc_instance:
-                    logger.info(f"{log_prefix} 麦麦不再专注聊天，转为随便水水...")
-                    await self.heart_fc_instance.shutdown()  # 正确关闭 HeartFChatting
-                    self.heart_fc_instance = None
-
-                chat_stream = chat_manager.get_stream(self.chat_id)
-                self.normal_chat_instance = NormalChat(chat_stream=chat_stream, interest_dict=self.get_interest_dict())
-                await self.normal_chat_instance.start_monitoring_interest()
-                # NormalChat 启动/停止逻辑将在下面处理
+                logger.debug(f"{log_prefix} 准备进入或保持 聊天 状态 ({current_chat_count}/{normal_limit})")
+                if await self._start_normal_chat():
+                    logger.info(f"{log_prefix} 成功进入或保持 NormalChat 状态。")
+                    state_changed = True
+                else:
+                    logger.error(f"{log_prefix} 启动 NormalChat 失败，无法进入 CHAT 状态。")
+                    # 考虑是否需要回滚状态或采取其他措施
+                    return # 启动失败，不改变状态
 
         elif new_state == ChatState.FOCUSED:
             focused_limit = current_mai_state.get_focused_chat_max_num()
-            current_focused_count = current_states_num[2]
-
-            if current_focused_count >= focused_limit and current_state != ChatState.FOCUSED:  # 仅在状态转换时检查限制
-                logger.debug(
-                    f"{log_prefix} 麦麦不能从 {current_state.value} 转换到 专注的聊天，原因：聊不过来了。({current_focused_count}/{focused_limit})"
-                )
-                return  # 阻止状态转换
+            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:
+                logger.debug(f"{log_prefix} 无法从 {current_state.value} 转到 专注。原因：聊不过来了 ({current_focused_count}/{focused_limit})")
+                return # 阻止状态转换
             else:
-                logger.debug(f"{log_prefix} 麦麦可以进入或保持 专注聊天 状态 ({current_focused_count}/{focused_limit})")
-                if not self.heart_fc_instance:
-                    logger.info(f"{log_prefix} 麦麦准备开始专注聊天...")
-                    try:
-                        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 状态
-
+                logger.debug(f"{log_prefix} 准备进入或保持 专注聊天 状态 ({current_focused_count}/{focused_limit})")
+                if await self._start_heart_fc_chat():
+                    logger.info(f"{log_prefix} 成功进入或保持 HeartFChatting 状态。")
+                    state_changed = True
                 else:
-                    # 已经是 FOCUSED 状态，或者 heart_fc_instance 已存在但未运行（不太可能）
-                    if not self.heart_fc_instance._loop_active:
-                        logger.warning(f"{log_prefix} HeartFChatting 实例存在但未激活，尝试重新激活...")
-                        await self.heart_fc_instance.add_time()  # 尝试添加时间以激活循环
-                    else:
-                        logger.debug(f"{log_prefix} 麦麦已经在专注聊天中。")
-                # NormalChat 启动/停止逻辑将在下面处理
+                    logger.error(f"{log_prefix} 启动 HeartFChatting 失败，无法进入 FOCUSED 状态。")
+                    # 启动失败，状态回滚到之前的状态或ABSENT？这里保持不改变
+                    return # 启动失败，不改变状态
+
 
         elif new_state == ChatState.ABSENT:
-            if current_state == ChatState.FOCUSED and self.heart_fc_instance:
-                logger.info(f"{log_prefix} 麦麦离开专注的聊天，撤退了.....")
-                await self.heart_fc_instance.shutdown()  # 正确关闭 HeartFChatting
-                self.heart_fc_instance = None
-            # NormalChat 启动/停止逻辑将在下面处理
+            logger.info(f"{log_prefix} 进入 ABSENT 状态，停止所有聊天活动...")
+            await self._stop_normal_chat()
+            await self._stop_heart_fc_chat()
+            state_changed = True # 总是可以成功转换到 ABSENT
 
-        # --- 更新状态和最后活动时间 (先更新状态，再根据新状态管理任务)---
-        self.chat_state.chat_status = new_state
-        self.last_active_time = time.time()
-        logger.info(f"{log_prefix} 麦麦的聊天状态从 {current_state.value} 变更为 {new_state.value}")
-
-        # --- 根据新的状态管理 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())
+        # --- 更新状态和最后活动时间 ---
+        if state_changed:
+            logger.info(f"{log_prefix} 麦麦的聊天状态从 {current_state.value} 变更为 {new_state.value}")
+            self.chat_state.chat_status = new_state
+            self.last_active_time = time.time()
         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,8 +461,8 @@ class SubHeartflow:
         logger.info(f"{self.log_prefix} 子心流开始工作...")
 
         while not self.should_stop:
-            # 主循环保持简单，只做状态检查
             await asyncio.sleep(30)  # 30秒检查一次停止标志
+            
 
         logger.info(f"{self.log_prefix} 子心流后台任务已停止。")
 
@@ -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:
-            logger.info(f"{self.log_prefix} 停止 NormalChat 监控任务 (Shutdown)...")
-            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())
+        # 使用新的停止方法
+        await self._stop_normal_chat()
+        await self._stop_heart_fc_chat()
 
-        # 停止 HeartFChatting (如果存在且正在运行)
-        if self.heart_fc_instance:
-            logger.info(f"{self.log_prefix} 关闭 HeartFChatting 实例 (Shutdown)...")
-            try:
-                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  # 清理实例引用
+        # 停止兴趣更新任务
+        if self.interest_chatting:
+            logger.info(f"{self.log_prefix} 停止兴趣系统后台任务...")
+            await self.interest_chatting.stop_updates()
 
         # 取消可能存在的旧后台任务 (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):
-        """评估CHAT状态的子心流兴趣度，满足条件则提升到FOCUSED状态"""
-        logger.debug("[子心流管理器] 开始兴趣评估周期...")
-        evaluated_count = 0
-        promoted_count = 0
+    async def evaluate_interest_and_promote(self, current_mai_state: MaiStateInfo):
+        """评估子心流兴趣度，满足条件且未达上限则提升到FOCUSED状态（基于start_hfc_probability）"""
+        log_prefix_manager = "[子心流管理器-兴趣评估]"
+        logger.debug(f"{log_prefix_manager} 开始周期... 当前状态: {current_mai_state.get_current_state().value}")
+
+        # 获取 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())
-
-        for sub_hf in subflows_snapshot:
-            flow_id = sub_hf.subheartflow_id
-            if flow_id in self.subheartflows and self.subheartflows[flow_id].chat_state.chat_status == ChatState.CHAT:
-                evaluated_count += 1
+        promoted_count = 0 # 记录本次提升的数量
+        try:
+            for sub_hf in subflows_snapshot:
+                flow_id = sub_hf.subheartflow_id
                 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()
-                if should_promote:
-                    logger.info(f"{log_prefix} 兴趣评估触发升级: CHAT -> FOCUSED")
-                    states_num = (
-                        self.count_subflows_by_state(ChatState.ABSENT),
-                        self.count_subflows_by_state(ChatState.CHAT),
-                        self.count_subflows_by_state(ChatState.FOCUSED),
-                    )
-                    await sub_hf.set_chat_state(ChatState.FOCUSED, states_num)
-                    if (
-                        self.subheartflows.get(flow_id)
-                        and self.subheartflows[flow_id].chat_state.chat_status == ChatState.FOCUSED
-                    ):
+                # 只处理 CHAT 状态的子心流
+                if sub_hf.chat_state.chat_status != ChatState.CHAT:
+                    continue
+
+                # 检查是否满足提升概率
+                should_hfc = random.random() < sub_hf.interest_chatting.start_hfc_probability
+                if not should_hfc:
+                    continue
+
+                # --- 关键检查：检查 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:
-                        logger.info(f"{log_prefix} 升级FOCUSED可能被限制阻止")
+                        # 提升成功后，更新当前专注计数，以便后续检查能使用最新值
+                        current_focused_count += 1
+                    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:
-            logger.debug(f"[子心流管理器] 评估完成. 评估{evaluated_count}个CHAT流, 升级{promoted_count}个到FOCUSED")
+        except Exception as e:
+            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:
         """统计指定状态的子心流数量