From 9b99ea3cb26e480ec1c825b03e48d9c12eef28b2 Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Sat, 26 Apr 2025 23:23:11 +0800 Subject: [PATCH] fixLruff --- src/heart_flow/subheartflow_manager.py | 8 +++++--- src/plugins/heartFC_chat/normal_chat.py | 12 ++++++------ 2 files changed, 11 insertions(+), 9 deletions(-) diff --git a/src/heart_flow/subheartflow_manager.py b/src/heart_flow/subheartflow_manager.py index 35efbc550..6d994d769 100644 --- a/src/heart_flow/subheartflow_manager.py +++ b/src/heart_flow/subheartflow_manager.py @@ -267,7 +267,7 @@ class SubHeartflowManager: # 使用 self.mai_state_info 获取当前状态和限制 current_state = self.mai_state_info.get_current_state() focused_limit = current_state.get_focused_chat_max_num() - + logger.debug(f"{log_prefix} 当前状态 ({current_state.value}) 开始尝试提升到FOCUSED状态") if int(time.time()) % 20 == 0: # 每20秒输出一次 @@ -287,7 +287,7 @@ class SubHeartflowManager: stream_name = chat_manager.get_stream_name(flow_id) or flow_id logger.debug(f"{log_prefix} 检查子心流: {stream_name}，现在状态: {sub_hf.chat_state.chat_status.value}") - + # 跳过非CHAT状态或已经是FOCUSED状态的子心流 if sub_hf.chat_state.chat_status == ChatState.FOCUSED: continue @@ -299,7 +299,9 @@ class SubHeartflowManager: continue # 检查是否满足提升概率 - logger.debug(f"{log_prefix} 检查子心流: {stream_name}，现在概率: {sub_hf.interest_chatting.start_hfc_probability}") + logger.debug( + f"{log_prefix} 检查子心流: {stream_name}，现在概率: {sub_hf.interest_chatting.start_hfc_probability}" + ) if random.random() >= sub_hf.interest_chatting.start_hfc_probability: continue diff --git a/src/plugins/heartFC_chat/normal_chat.py b/src/plugins/heartFC_chat/normal_chat.py index 55f7cb133..890b5a82a 100644 --- a/src/plugins/heartFC_chat/normal_chat.py +++ b/src/plugins/heartFC_chat/normal_chat.py @@ -358,15 +358,15 @@ class NormalChat: # 处理需要回复的消息 processed_count = 0 # --- 修改：迭代前创建要处理的ID列表副本，防止迭代时修改 --- - messages_to_process_initially = list(messages_to_reply) # 创建副本 + messages_to_process_initially = list(messages_to_reply) # 创建副本 # --- 修改结束 --- - for item in messages_to_process_initially: # 使用副本迭代 + for item in messages_to_process_initially: # 使用副本迭代 msg_id, (message, interest_value, is_mentioned) = item # --- 修改：在处理前尝试 pop，防止竞争 --- popped_item = self.interest_dict.pop(msg_id, None) if popped_item is None: logger.warning(f"[{self.stream_name}] 初始兴趣消息 {msg_id} 在处理前已被移除，跳过。") - continue # 如果消息已被其他任务处理（pop），则跳过 + continue # 如果消息已被其他任务处理（pop），则跳过 # --- 修改结束 --- try: @@ -416,7 +416,7 @@ class NormalChat: async def start_chat(self): """为此 NormalChat 实例关联的 ChatStream 启动聊天任务（如果尚未运行）， - 并在后台处理一次初始的高兴趣消息。""" # 文言文注释示例：启聊之始，若有遗珠，当于暗处拂拭，勿碍正途。 + 并在后台处理一次初始的高兴趣消息。""" # 文言文注释示例：启聊之始，若有遗珠，当于暗处拂拭，勿碍正途。 if self._chat_task is None or self._chat_task.done(): # --- 修改：使用 create_task 启动初始消息处理 --- logger.info(f"[{self.stream_name}] 开始后台处理初始兴趣消息...") @@ -428,9 +428,9 @@ class NormalChat: # 启动后台轮询任务 (这部分不变) logger.info(f"[{self.stream_name}] 启动后台兴趣消息轮询任务...") - polling_task = asyncio.create_task(self._reply_interested_message()) # 注意变量名区分 + polling_task = asyncio.create_task(self._reply_interested_message()) # 注意变量名区分 polling_task.add_done_callback(lambda t: self._handle_task_completion(t)) - self._chat_task = polling_task # self._chat_task 仍然指向主要的轮询任务 + self._chat_task = polling_task # self._chat_task 仍然指向主要的轮询任务 else: logger.info(f"[{self.stream_name}] 聊天轮询任务已在运行中。")