[Bugfix] Assign original prompt back to RequestOutput (#2498)
Signed-off-by: Yuanheng Zhao <jonathan.zhaoyh@gmail.com>
This commit is contained in:
1 parent
dd9ca6feef
commit
88f7ed9fb6
1 file changed
+2
@@ -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,
|
||||
|
||||
Reference in new issue
Block a user