UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-12-29 15:25:29 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-12-20 18:37:25 -08:00
parent b1b2a6d78c
commit cf0aca1487
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -528,7 +528,14 @@ async def process_chat_response(request, response, user, events, metadata, tasks
return response
event_emitter = None
if "session_id" in metadata:
if (
"session_id" in metadata
and metadata["session_id"]
and "chat_id" in metadata
and metadata["chat_id"]
and "message_id" in metadata
and metadata["message_id"]
):
event_emitter = get_event_emitter(metadata)
if event_emitter:
@ -701,6 +708,7 @@ async def process_chat_response(request, response, user, events, metadata, tasks
return {"status": True, "task_id": task_id}
else:
# Fallback to the original response
async def stream_wrapper(original_generator, events):
def wrap_item(item):