OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-12-27 22:46:50 -08:00
parent d55884b50e
commit 5a720a4a31
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -929,6 +929,10 @@ async def process_chat_response(
# Handle as a background task # Handle as a background task
async def post_response_handler(response, events): async def post_response_handler(response, events):
assistant_message = get_last_assistant_message(form_data["messages"])
content = assistant_message if assistant_message else ""
try: try:
for event in events: for event in events:
await event_emitter( await event_emitter(
@ -947,9 +951,6 @@ async def process_chat_response(
}, },
) )
assistant_message = get_last_assistant_message(form_data["messages"])
content = assistant_message if assistant_message else ""
async for line in response.body_iterator: async for line in response.body_iterator:
line = line.decode("utf-8") if isinstance(line, bytes) else line line = line.decode("utf-8") if isinstance(line, bytes) else line
data = line data = line
@ -1051,6 +1052,16 @@ async def process_chat_response(
print("Task was cancelled!") print("Task was cancelled!")
await event_emitter({"type": "task-cancelled"}) await event_emitter({"type": "task-cancelled"})
if not ENABLE_REALTIME_CHAT_SAVE:
# Save message in the database
Chats.upsert_message_to_chat_by_id_and_message_id(
metadata["chat_id"],
metadata["message_id"],
{
"content": content,
},
)
if response.background is not None: if response.background is not None:
await response.background() await response.background()