diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index 9261c26b3..a33b34814 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -561,6 +561,15 @@ async def process_chat_response(request, response, user, events, metadata, tasks } ) + # Save message in the database + Chats.upsert_message_to_chat_by_id_and_message_id( + metadata["chat_id"], + metadata["message_id"], + { + **event, + }, + ) + content = "" async for line in response.body_iterator: line = line.decode("utf-8") if isinstance(line, bytes) else line