Fix image generation output

This commit is contained in:
Weves 2024-06-21 15:14:51 -07:00 committed by Chris Weaver
parent 98b2507045
commit 3e39a921b0

View File

@ -352,7 +352,7 @@ class Answer:
list[ImageGenerationResponse], response.response
)
img_urls = [img.url for img in img_generation_response]
break
yield response
prompt_builder.update_user_prompt(