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(