OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-08 23:47:14 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

enh: include sources field in non-streaming response

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-05-26 22:22:37 +04:00
parent ffa51ece0c
commit 2d5b82df8c
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
backend/open_webui/utils/middleware.py
View File

@ -1217,8 +1217,34 @@ async def process_chat_response(
await background_tasks_handler()
if events and isinstance(events, list) and isinstance(response, dict):
extra_response = {}
for event in events:
if isinstance(event, dict):
extra_response.update(event)
else:
extra_response[event] = True
response = {
**extra_response,
**response,
}
return response
else:
if events and isinstance(events, list) and isinstance(response, dict):
extra_response = {}
for event in events:
if isinstance(event, dict):
extra_response.update(event)
else:
extra_response[event] = True
response = {
**extra_response,
**response,
}
return response
# Non standard response