From 3a9d5b0494f06c8a92ce648675b5362cdc6a06ef Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Mon, 7 Jul 2025 23:51:01 +0400 Subject: [PATCH] refac --- backend/open_webui/utils/middleware.py | 31 ++++++++++++++------------ 1 file changed, 17 insertions(+), 14 deletions(-) diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index eb8ea5e47d..0dbe47b583 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -2217,6 +2217,7 @@ async def process_chat_response( content_blocks[-1]["type"] == "code_interpreter" and retries < MAX_RETRIES ): + await event_emitter( { "type": "chat:completion", @@ -2349,26 +2350,28 @@ async def process_chat_response( ) try: + new_form_data = { + "model": model_id, + "stream": True, + "messages": [ + *form_data["messages"], + { + "role": "assistant", + "content": serialize_content_blocks( + content_blocks, raw=True + ), + }, + ], + } + res = await generate_chat_completion( request, - { - "model": model_id, - "stream": True, - "messages": [ - *form_data["messages"], - { - "role": "assistant", - "content": serialize_content_blocks( - content_blocks, raw=True - ), - }, - ], - }, + new_form_data, user, ) if isinstance(res, StreamingResponse): - await stream_body_handler(res) + await stream_body_handler(res, new_form_data) else: break except Exception as e: