[Bugfix] Assign original prompt back to RequestOutput (#2498)

Signed-off-by: Yuanheng Zhao <jonathan.zhaoyh@gmail.com>
This commit is contained in:
Yuanheng Zhao authored and GitHub committed 2026-04-05 10:46:11 +00:00
1 parent dd9ca6feef
commit 88f7ed9fb6
1 file changed
+2
+2
View File
@@ -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,