Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-03-07 17:13:19 -06:00
parent 2108f420ea
commit d4faa5a5ea
2 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/tools/builtin.py
+6 -2
View File
@@ -249,11 +249,13 @@ async def generate_image(
# Persist files to DB if chat context is available # Persist files to DB if chat context is available
if __chat_id__ and __message_id__ and images: if __chat_id__ and __message_id__ and images:
image_files = Chats.add_message_files_by_id_and_message_id( db_files = Chats.add_message_files_by_id_and_message_id(
__chat_id__, __chat_id__,
__message_id__, __message_id__,
image_files, image_files,
) )
if db_files is not None:
image_files = db_files
# Emit the images to the UI if event emitter is available # Emit the images to the UI if event emitter is available
if __event_emitter__ and image_files: if __event_emitter__ and image_files:
@@ -314,11 +316,13 @@ async def edit_image(
# Persist files to DB if chat context is available # Persist files to DB if chat context is available
if __chat_id__ and __message_id__ and images: if __chat_id__ and __message_id__ and images:
image_files = Chats.add_message_files_by_id_and_message_id( db_files = Chats.add_message_files_by_id_and_message_id(
__chat_id__, __chat_id__,
__message_id__, __message_id__,
image_files, image_files,
) )
if db_files is not None:
image_files = db_files
# Emit the images to the UI if event emitter is available # Emit the images to the UI if event emitter is available
if __event_emitter__ and image_files: if __event_emitter__ and image_files:
backend/open_webui/utils/middleware.py
+7 -4
View File
@@ -3777,14 +3777,17 @@ async def streaming_chat_response_handler(response, ctx):
delta.get("images", []), request, metadata, user delta.get("images", []), request, metadata, user
) )
if image_urls: if image_urls:
image_file_list = [
{"type": "image", "url": url}
for url in image_urls
]
message_files = Chats.add_message_files_by_id_and_message_id( message_files = Chats.add_message_files_by_id_and_message_id(
metadata["chat_id"], metadata["chat_id"],
metadata["message_id"], metadata["message_id"],
[ image_file_list,
{"type": "image", "url": url}
for url in image_urls
],
) )
if message_files is None:
message_files = image_file_list
await event_emitter( await event_emitter(
{ {