diff --git a/src/chat/message_manager/message_manager.py b/src/chat/message_manager/message_manager.py index e55fe077e..91a9ed5db 100644 --- a/src/chat/message_manager/message_manager.py +++ b/src/chat/message_manager/message_manager.py @@ -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)