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 2025-05-16 21:29:50 +04:00
parent 2bd7db12a2
commit 07b5e84221
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -947,12 +947,14 @@ async def process_chat_response(
message = message_map.get(metadata["message_id"]) if message_map else None message = message_map.get(metadata["message_id"]) if message_map else None
if message: 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. # Remove details tags and files from the messages.
# as get_message_list creates a new list, it does not affect # as get_message_list creates a new list, it does not affect
# the original messages outside of this handler # the original messages outside of this handler
for message in messages:
messages = []
for message in message_list:
content = message.get("content", "") content = message.get("content", "")
if isinstance(content, list): if isinstance(content, list):
for item in content: for item in content:
@ -968,10 +970,12 @@ async def process_chat_response(
flags=re.S | re.I, flags=re.S | re.I,
).strip() ).strip()
message = { messages.append(
"role": message["role"], {
"content": content, "role": message["role"],
} "content": content,
}
)
if tasks and messages: if tasks and messages:
if TASKS.TITLE_GENERATION in tasks: if TASKS.TITLE_GENERATION in tasks: