优化消息管理中的异步任务处理

This commit is contained in:
Windpicker-owo
2025-12-13 21:06:57 +08:00
parent 8366d5aaad
commit ee30fa5d1d

View File

@@ -109,15 +109,11 @@ class MessageManager:
context = chat_stream.context
if not (context.stream_loop_task and not context.stream_loop_task.done()):
# 异步启动驱动器任务;避免在高并发下阻塞消息入队
context.stream_loop_task = asyncio.create_task(
stream_loop_manager.start_stream_loop(stream_id)
)
# 并行触发打断检查,不阻塞消息入队
context.interruption_task = asyncio.create_task(
self._check_and_handle_interruption(chat_stream, message)
)
await stream_loop_manager.start_stream_loop(stream_id)
# 检查并处理消息打断
await self._check_and_handle_interruption(chat_stream, message)
# 入队消息
await chat_stream.context.add_message(message)