From 3e39a921b005ad38b18b7aac9e545859a3fda0b5 Mon Sep 17 00:00:00 2001 From: Weves Date: Fri, 21 Jun 2024 15:14:51 -0700 Subject: [PATCH] Fix image generation output --- backend/danswer/llm/answering/answer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/danswer/llm/answering/answer.py b/backend/danswer/llm/answering/answer.py index 328bb7251..ae8118d21 100644 --- a/backend/danswer/llm/answering/answer.py +++ b/backend/danswer/llm/answering/answer.py @@ -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(