scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-08-28 05:27:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refactor: move background tasks handler call to ensure consistent execution in chat response handlers (#24717)

Browse source
This commit is contained in:
Jacob Leksan 2026-06-01 16:50:15 -04:00 • committed by GitHub
parent 974b13839e
commit 80da840ae5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
backend/open_webui/utils/middleware.py
View file

@ -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!')