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

fix(chat): 增强回复生成的健壮性，增加超时和类型检查

Browse Source 
在 `DefaultReplyer` 的提示词构建流程中，为并行的子任务（如记忆、工具、关系等）增加了15秒的超时机制。这可以防止因某个子任务耗时过长或卡死而导致整个回复生成过程停滞。

同时，在能量系统中增加了对计算器分数和兴趣值的类型检查，确保它们是数值类型，避免了潜在的 `TypeError` 异常，提高了系统的稳定性。
This commit is contained in:
tt-P607
2025-10-02 23:22:00 +08:00
committed by Windpicker-owo
parent f448802cfd
commit 40b6348d3e
2 changed files with 40 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/chat/replyer/default_generator.py
View File

@@ -1287,21 +1287,36 @@ class DefaultReplyer:
from src.chat.utils.prompt import Prompt
# 并行执行六个构建任务
task_results = await asyncio.gather(
self._time_and_run_task(
self.build_expression_habits(chat_talking_prompt_short, target), "expression_habits"
),
self._time_and_run_task(self.build_relation_info(sender, target), "relation_info"),
self._time_and_run_task(self.build_memory_block(chat_talking_prompt_short, target), "memory_block"),
self._time_and_run_task(
self.build_tool_info(chat_talking_prompt_short, sender, target, enable_tool=enable_tool), "tool_info"
),
self._time_and_run_task(self.get_prompt_info(chat_talking_prompt_short, sender, target), "prompt_info"),
self._time_and_run_task(
Prompt.build_cross_context(chat_id, global_config.personality.prompt_mode, target_user_info),
"cross_context",
),
)
tasks = {
"expression_habits": asyncio.create_task(self._time_and_run_task(self.build_expression_habits(chat_talking_prompt_short, target), "expression_habits")),
"relation_info": asyncio.create_task(self._time_and_run_task(self.build_relation_info(sender, target), "relation_info")),
"memory_block": asyncio.create_task(self._time_and_run_task(self.build_memory_block(chat_talking_prompt_short, target), "memory_block")),
"tool_info": asyncio.create_task(self._time_and_run_task(self.build_tool_info(chat_talking_prompt_short, sender, target, enable_tool=enable_tool), "tool_info")),
"prompt_info": asyncio.create_task(self._time_and_run_task(self.get_prompt_info(chat_talking_prompt_short, sender, target), "prompt_info")),
"cross_context": asyncio.create_task(self._time_and_run_task(Prompt.build_cross_context(chat_id, global_config.personality.prompt_mode, target_user_info), "cross_context")),
}
# 设置超时
timeout = 15.0 # 秒
async def get_task_result(task_name, task):
try:
return await asyncio.wait_for(task, timeout=timeout)
except asyncio.TimeoutError:
logger.warning(f"构建任务{task_name}超时 ({timeout}s)，使用默认值")
# 为超时任务提供默认值
default_values = {
"expression_habits": "",
"relation_info": "",
"memory_block": "",
"tool_info": "",
"prompt_info": "",
"cross_context": "",
}
logger.info(f"为超时任务 {task_name} 提供默认值")
return task_name, default_values[task_name], timeout
task_results = await asyncio.gather(*(get_task_result(name, task) for name, task in tasks.items()))
# 任务名称中英文映射
task_name_mapping = {