gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🤖 自动格式化代码 [skip ci]

Browse Source
This commit is contained in:
github-actions[bot]
2025-04-26 13:49:24 +00:00
parent d50c2df0f6
commit 3931423e8e
4 changed files with 42 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/heart_flow/background_tasks.py
View File

@@ -255,7 +255,6 @@ class BackgroundTaskManager:
# --- 结束新增 ---
# --- 结束新增 ---
# --- Specific Task Runners --- #
@@ -286,4 +285,3 @@ class BackgroundTaskManager:
interval=self.interest_eval_interval,
task_func=self._perform_interest_eval_work,
)

6
src/heart_flow/sub_heartflow.py
View File

@@ -243,7 +243,7 @@ class SubHeartflow:
self,
subheartflow_id,
mai_states: MaiStateInfo,
hfc_no_reply_callback: Callable[[], Coroutine[None, None, None]]
hfc_no_reply_callback: Callable[[], Coroutine[None, None, None]],
):
"""子心流初始化函数
@@ -380,8 +380,8 @@ class SubHeartflow:
self.heart_fc_instance = HeartFChatting(
chat_id=self.subheartflow_id,
sub_mind=self.sub_mind,
observations=self.observations, # 传递所有观察者
on_consecutive_no_reply_callback=self.hfc_no_reply_callback # <-- Use stored callback
observations=self.observations, # 传递所有观察者
on_consecutive_no_reply_callback=self.hfc_no_reply_callback, # <-- Use stored callback
)
# 初始化并启动 HeartFChatting

38
src/heart_flow/subheartflow_manager.py
View File

@@ -2,8 +2,8 @@ import asyncio
import time
import random
from typing import Dict, Any, Optional, List
import json # 导入 json 模块
import functools # <-- 新增导入
import json # 导入 json 模块
import functools # <-- 新增导入
# 导入日志模块
from src.common.logger import get_module_logger, LogConfig, SUBHEARTFLOW_MANAGER_STYLE_CONFIG
@@ -88,7 +88,7 @@ class SubHeartflowManager:
new_subflow = SubHeartflow(
subheartflow_id,
self.mai_state_info,
hfc_callback # <-- 传递 partial 创建的回调
hfc_callback, # <-- 传递 partial 创建的回调
)
# 异步初始化
@@ -134,20 +134,23 @@ class SubHeartflowManager:
if subflow.chat_state.chat_status == ChatState.ABSENT:
return True
else:
logger.warning(f"{log_prefix} 调用 change_chat_state 后，{stream_name} 状态仍为 {subflow.chat_state.chat_status.value}")
logger.warning(
f"{log_prefix} 调用 change_chat_state 后，{stream_name} 状态仍为 {subflow.chat_state.chat_status.value}"
)
return False
except Exception as e:
logger.error(f"{log_prefix} 设置 {stream_name} 状态为 ABSENT 时失败: {e}", exc_info=True)
return False
else:
logger.debug(f"{log_prefix} {stream_name} 已是 ABSENT 状态")
return True # 已经是目标状态，视为成功
return True # 已经是目标状态，视为成功
# --- 结束新增 ---
async def sleep_subheartflow(self, subheartflow_id: Any, reason: str) -> bool:
"""停止指定的子心流并将其状态设置为 ABSENT"""
log_prefix = "[子心流管理]"
async with self._lock: # 加锁以安全访问字典
async with self._lock: # 加锁以安全访问字典
subheartflow = self.subheartflows.get(subheartflow_id)
stream_name = chat_manager.get_stream_name(subheartflow_id) or subheartflow_id
@@ -229,7 +232,7 @@ class SubHeartflowManager:
changed_count = 0
processed_count = 0
async with self._lock: # 获取锁以安全迭代
async with self._lock: # 获取锁以安全迭代
# 使用 list() 创建一个当前值的快照，防止在迭代时修改字典
flows_to_update = list(self.subheartflows.values())
processed_count = len(flows_to_update)
@@ -239,8 +242,7 @@ class SubHeartflowManager:
for subflow in flows_to_update:
# 记录原始状态，以便统计实际改变的数量
original_state_was_absent = (subflow.chat_state.chat_status == ChatState.ABSENT)
original_state_was_absent = subflow.chat_state.chat_status == ChatState.ABSENT
success = await self._try_set_subflow_absent_internal(subflow, log_prefix)
@@ -345,7 +347,6 @@ class SubHeartflowManager:
log_prefix = f"[{stream_name}]"
current_subflow_state = sub_hf.chat_state.chat_status
_observation_summary = "没有可用的观察信息。" # 默认值
first_observation = sub_hf.observations[0]
@@ -357,12 +358,10 @@ class SubHeartflowManager:
else:
logger.warning(f"{log_prefix} [{stream_name}] 第一个观察者不是 ChattingObservation 类型。")
# --- 获取麦麦状态 ---
mai_state_description = f"你当前状态: {current_mai_state.value}。"
# 获取个性化信息
# 获取个性化信息
individuality = Individuality.get_instance()
# 构建个性部分
@@ -379,7 +378,6 @@ class SubHeartflowManager:
random_detail = random.choice(individuality.identity.identity_detail)
prompt_personality += f"，{random_detail}"
# --- 针对 ABSENT 状态 ---
if current_subflow_state == ChatState.ABSENT:
# 构建Prompt
@@ -393,13 +391,13 @@ class SubHeartflowManager:
f"给出你的判断，和理由，然后以 JSON 格式回答"
f"包含键 'decision'，如果要开始聊天，值为 true ，否则为 false.\n"
f"包含键 'reason'，其值为你的理由。\n"
f"例如：{{\"decision\": true, \"reason\": \"因为我想聊天\"}}\n"
f'例如：{{"decision": true, "reason": "因为我想聊天"}}\n'
f"请只输出有效的 JSON 对象。"
)
# 调用LLM评估
should_activate = await self._llm_evaluate_state_transition(prompt)
if should_activate is None: # 处理解析失败或意外情况
if should_activate is None: # 处理解析失败或意外情况
logger.warning(f"{log_prefix}LLM评估返回无效结果，跳过。")
continue
@@ -436,13 +434,13 @@ class SubHeartflowManager:
f"给出你的判断，和理由，然后以 JSON 格式回答"
f"包含键 'decision'，如果要离开聊天，值为 true ，否则为 false.\n"
f"包含键 'reason'，其值为你的理由。\n"
f"例如：{{\"decision\": true, \"reason\": \"因为我想休息\"}}\n"
f'例如：{{"decision": true, "reason": "因为我想休息"}}\n'
f"请只输出有效的 JSON 对象。"
)
# 调用LLM评估
should_deactivate = await self._llm_evaluate_state_transition(prompt)
if should_deactivate is None: # 处理解析失败或意外情况
if should_deactivate is None: # 处理解析失败或意外情况
logger.warning(f"{log_prefix}LLM评估返回无效结果，跳过。")
continue
@@ -454,8 +452,6 @@ class SubHeartflowManager:
else:
logger.info(f"{log_prefix}LLM建议不进入ABSENT状态。")
async def _llm_evaluate_state_transition(self, prompt: str) -> Optional[bool]:
"""
使用 LLM 评估是否应进行状态转换，期望 LLM 返回 JSON 格式。
@@ -573,6 +569,7 @@ class SubHeartflowManager:
# 注意：这里不需要再获取锁，因为 request_absent_transition 内部会处理锁
logger.debug(f"[管理器 HFC 处理器] 接收到来自 {subheartflow_id} 的 HFC 无回复信号")
await self.request_absent_transition(subheartflow_id)
# --- 结束新增 --- #
# --- 新增：处理来自 HeartFChatting 的状态转换请求 --- #
@@ -608,4 +605,5 @@ class SubHeartflowManager:
logger.warning(
f"[状态转换请求] 收到对 {stream_name} 的请求，但其状态为 {current_state.value} (非 FOCUSED)，不执行转换"
)
# --- 结束新增 --- #

31
src/plugins/heartFC_chat/heartFC_chat.py
View File

@@ -154,14 +154,14 @@ class HeartFChatting:
其生命周期现在由其关联的 SubHeartflow 的 FOCUSED 状态控制。
"""
CONSECUTIVE_NO_REPLY_THRESHOLD = 4 # 连续不回复的阈值
CONSECUTIVE_NO_REPLY_THRESHOLD = 4 # 连续不回复的阈值
def __init__(
self,
chat_id: str,
sub_mind: SubMind,
observations: Observation,
on_consecutive_no_reply_callback: Callable[[], Coroutine[None, None, None]]
on_consecutive_no_reply_callback: Callable[[], Coroutine[None, None, None]],
):
"""
HeartFChatting 初始化函数
@@ -209,8 +209,8 @@ class HeartFChatting:
self._cycle_counter = 0
self._cycle_history: Deque[CycleInfo] = deque(maxlen=10) # 保留最近10个循环的信息
self._current_cycle: Optional[CycleInfo] = None
self._lian_xu_bu_hui_fu_ci_shu: int = 0 # <--- 新增：连续不回复计数器
self._shutting_down: bool = False # <--- 新增：关闭标志位
self._lian_xu_bu_hui_fu_ci_shu: int = 0 # <--- 新增：连续不回复计数器
self._shutting_down: bool = False # <--- 新增：关闭标志位
async def _initialize(self) -> bool:
"""
@@ -309,9 +309,9 @@ class HeartFChatting:
# 如果未能获取锁（理论上不太可能，除非 shutdown 过程中释放了但又被抢了？）
# 或者也可以在这里再次检查 self._shutting_down
if self._shutting_down:
break # 再次检查，确保退出
break # 再次检查，确保退出
logger.warning(f"{self.log_prefix} 未能获取循环处理锁，跳过本次循环。")
await asyncio.sleep(0.1) # 短暂等待避免空转
await asyncio.sleep(0.1) # 短暂等待避免空转
continue
# 记录规划开始时间点
@@ -600,15 +600,18 @@ class HeartFChatting:
if not self._shutting_down:
self._lian_xu_bu_hui_fu_ci_shu += 1
logger.debug(f"{self.log_prefix} 连续不回复计数增加: {self._lian_xu_bu_hui_fu_ci_shu}/{self.CONSECUTIVE_NO_REPLY_THRESHOLD}")
logger.debug(
f"{self.log_prefix} 连续不回复计数增加: {self._lian_xu_bu_hui_fu_ci_shu}/{self.CONSECUTIVE_NO_REPLY_THRESHOLD}"
)
# 检查是否达到阈值
if self._lian_xu_bu_hui_fu_ci_shu >= self.CONSECUTIVE_NO_REPLY_THRESHOLD:
logger.info(f"{self.log_prefix} 连续不回复达到阈值 ({self._lian_xu_bu_hui_fu_ci_shu}次)，调用回调请求状态转换")
logger.info(
f"{self.log_prefix} 连续不回复达到阈值 ({self._lian_xu_bu_hui_fu_ci_shu}次)，调用回调请求状态转换"
)
# 调用回调。注意：这里不重置计数器，依赖回调函数成功改变状态来隐式重置上下文。
await self.on_consecutive_no_reply_callback()
return True
except asyncio.CancelledError:
@@ -616,11 +619,11 @@ class HeartFChatting:
logger.info(f"{self.log_prefix} 处理 'no_reply' 时等待被中断 (CancelledError)")
# 让异常向上传播，由 _hfc_loop 的异常处理逻辑接管
raise
except Exception as e: # 捕获调用管理器或其他地方可能发生的错误
except Exception as e: # 捕获调用管理器或其他地方可能发生的错误
logger.error(f"{self.log_prefix} 处理 'no_reply' 时发生错误: {e}")
logger.error(traceback.format_exc())
# 发生意外错误时，可以选择是否重置计数器，这里选择不重置
return False # 表示动作未成功
return False # 表示动作未成功
async def _wait_for_new_message(self, observation, planner_start_db_time: float, log_prefix: str) -> bool:
"""
@@ -639,7 +642,7 @@ class HeartFChatting:
# --- 在每次循环开始时检查关闭标志 ---
if self._shutting_down:
logger.info(f"{log_prefix} 等待新消息时检测到关闭信号，中断等待。")
return False # 表示因为关闭而退出
return False # 表示因为关闭而退出
# -----------------------------------
# 检查新消息
@@ -654,7 +657,7 @@ class HeartFChatting:
try:
# 短暂休眠，让其他任务有机会运行，并能更快响应取消或关闭
await asyncio.sleep(0.5) # 缩短休眠时间
await asyncio.sleep(0.5) # 缩短休眠时间
except asyncio.CancelledError:
# 如果在休眠时被取消，再次检查关闭标志
# 如果是正常关闭，则不需要警告
@@ -910,7 +913,7 @@ class HeartFChatting:
async def shutdown(self):
"""优雅关闭HeartFChatting实例，取消活动循环任务"""
logger.info(f"{self.log_prefix} 正在关闭HeartFChatting...")
self._shutting_down = True # <-- 在开始关闭时设置标志位
self._shutting_down = True # <-- 在开始关闭时设置标志位
# 取消循环任务
if self._loop_task and not self._loop_task.done():