From 80da840ae5ef33a14f3157c77c176fd7d14089fb Mon Sep 17 00:00:00 2001 From: Jacob Leksan <63938553+jmleksan@users.noreply.github.com> Date: Mon, 1 Jun 2026 16:50:15 -0400 Subject: [PATCH] refactor: move background tasks handler call to ensure consistent execution in chat response handlers (#24717) --- backend/open_webui/utils/middleware.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index c5f9f01803..b5f043356a 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -3545,13 +3545,13 @@ async def non_streaming_chat_response_handler(response, ctx): }, ) - await background_tasks_handler(ctx) ctx['assistant_message'] = { 'content': content, 'output': response_output, **({'usage': usage} if usage else {}), } await outlet_filter_handler(ctx) + await background_tasks_handler(ctx) response = build_response_object(response, merge_events_into_response(response_data, events)) except Exception as e: @@ -5164,13 +5164,13 @@ async def streaming_chat_response_handler(response, ctx): } ) - await background_tasks_handler(ctx) ctx['assistant_message'] = { 'content': serialize_output(output), 'output': output, **({'usage': usage} if usage else {}), } await outlet_filter_handler(ctx) + await background_tasks_handler(ctx) except asyncio.CancelledError: log.warning('Task was cancelled!')