fix(chat): 修复消息打断会取消正在进行的回复任务的问题

之前的消息打断逻辑会无差别地取消处理流中的所有任务。这会导致一个问题:当用户在机器人生成回复期间快速发送新消息时,回复任务会被意外中断,导致机器人无法正常完成回复。

本次修改通过引入 `is_replying` 状态来解决此问题:
1.  在 `StreamContext` 中新增 `is_replying` 状态标志,用于追踪回复生成过程。
2.  当开始生成回复时,设置该标志为 `True`,并在回复完成或取消后通过 `finally` 块确保其恢复为 `False`。
3.  `MessageManager` 的打断检查逻辑现在会首先检查此标志,如果为 `True` 则跳过打断,从而保护正在进行的回复。
4.  `cancel_all_stream_tasks` 也增加了 `exclude_reply` 选项,确保即使触发打断,也不会取消回复任务。
This commit is contained in:
tt-P607
2025-10-11 02:11:32 +08:00
parent a3d62a265d
commit 94e34c9370
4 changed files with 31 additions and 23 deletions

View File

@@ -42,6 +42,7 @@ class StreamContext(BaseDataModel):
processing_task: asyncio.Task | None = None
interruption_count: int = 0 # 打断计数器
last_interruption_time: float = 0.0 # 上次打断时间
is_replying: bool = False # 是否正在回复
# 独立分发周期字段
next_check_time: float = field(default_factory=time.time) # 下次检查时间