From 07b5e842211fdaf3ff37142a2ba4a59ca3087f51 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Fri, 16 May 2025 21:29:50 +0400 Subject: [PATCH] refac --- backend/open_webui/utils/middleware.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index 323fd200a..1b8b5c9bd 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -947,12 +947,14 @@ async def process_chat_response( message = message_map.get(metadata["message_id"]) if message_map else None if message: - messages = get_message_list(message_map, message.get("id")) + message_list = get_message_list(message_map, message.get("id")) # Remove details tags and files from the messages. # as get_message_list creates a new list, it does not affect # the original messages outside of this handler - for message in messages: + + messages = [] + for message in message_list: content = message.get("content", "") if isinstance(content, list): for item in content: @@ -968,10 +970,12 @@ async def process_chat_response( flags=re.S | re.I, ).strip() - message = { - "role": message["role"], - "content": content, - } + messages.append( + { + "role": message["role"], + "content": content, + } + ) if tasks and messages: if TASKS.TITLE_GENERATION in tasks: