diff --git a/vllm_omni/entrypoints/omni_stage.py b/vllm_omni/entrypoints/omni_stage.py index 3fa722b03..899d8ddff 100644 --- a/vllm_omni/entrypoints/omni_stage.py +++ b/vllm_omni/entrypoints/omni_stage.py @@ -689,8 +689,10 @@ class OmniStage: Returns: Result dictionary if available, None otherwise. Result contains request_id, engine_outputs (or engine_outputs_shm), and metrics. + Returns None if output queue is already closed (e.g. during shutdown). """ - assert self._out_q is not None + if self._out_q is None: + return None try: return self._out_q.get_nowait() except Exception: