From 88f7ed9fb6a33c8ffd0211e891f8396543c615e9 Mon Sep 17 00:00:00 2001 From: Yuanheng Zhao <54058983+yuanheng-zhao@users.noreply.github.com> Date: Sun, 5 Apr 2026 18:46:11 +0800 Subject: [PATCH] [Bugfix] Assign original prompt back to RequestOutput (#2498) Signed-off-by: Yuanheng Zhao --- vllm_omni/engine/async_omni_engine.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/vllm_omni/engine/async_omni_engine.py b/vllm_omni/engine/async_omni_engine.py index 092b341e4..28c6d6caa 100644 --- a/vllm_omni/engine/async_omni_engine.py +++ b/vllm_omni/engine/async_omni_engine.py @@ -716,6 +716,8 @@ class AsyncOmniEngine: # Register with stage 0's output processor. output_prompt_text = prompt_text + if output_prompt_text is None and isinstance(original_prompt, dict): + output_prompt_text = original_prompt.get("prompt") self.output_processors[0].add_request( request=request, prompt=output_prompt_text,