ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 12:25:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-12-08 11:30:38 -05:00
parent 8cea0cf746
commit bcd50ed8f1
1 changed files with 17 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
backend/open_webui/utils/middleware.py
View file

@ -716,17 +716,18 @@ async def chat_web_search_handler(
return form_data return form_data
def get_last_images(message_list): def get_images_from_messages(message_list):
images = [] images = []
for message in reversed(message_list): for message in reversed(message_list):
images_flag = False
message_images = []
for file in message.get("files", []): for file in message.get("files", []):
if file.get("type") == "image": if file.get("type") == "image":
images.append(file.get("url")) message_images.append(file.get("url"))
images_flag = True
if images_flag: if message_images:
break images.append(message_images)
return images return images
@ -780,7 +781,16 @@ async def chat_image_generation_handler(
user_message = get_last_user_message(message_list) user_message = get_last_user_message(message_list)
prompt = user_message prompt = user_message
input_images = get_last_images(message_list) message_images = get_images_from_messages(message_list)
# Limit to first 2 sets of images
# We may want to change this in the future to allow more images
input_images = []
for idx, images in enumerate(message_images):
if idx >= 2:
break
for image in images:
input_images.append(image)
system_message_content = "" system_message_content = ""