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/fix: openai edit multiple images

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-11-21 04:13:59 -05:00
parent 0dd9ad7ffc
commit 0c18cd67d5
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
backend/open_webui/routers/images.py
View file

@ -838,13 +838,13 @@ async def image_edits(
except Exception as e: except Exception as e:
raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e)) raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e))
def get_image_file_item(base64_string): def get_image_file_item(base64_string, param_name="image"):
data = base64_string data = base64_string
header, encoded = data.split(",", 1) header, encoded = data.split(",", 1)
mime_type = header.split(";")[0].lstrip("data:") mime_type = header.split(";")[0].lstrip("data:")
image_data = base64.b64decode(encoded) image_data = base64.b64decode(encoded)
return ( return (
"image", param_name,
( (
f"{uuid.uuid4()}.png", f"{uuid.uuid4()}.png",
io.BytesIO(image_data), io.BytesIO(image_data),
@ -879,7 +879,7 @@ async def image_edits(
files = [get_image_file_item(form_data.image)] files = [get_image_file_item(form_data.image)]
elif isinstance(form_data.image, list): elif isinstance(form_data.image, list):
for img in form_data.image: for img in form_data.image:
files.append(get_image_file_item(img)) files.append(get_image_file_item(img, "image[]"))
url_search_params = "" url_search_params = ""
if request.app.state.config.IMAGES_EDIT_OPENAI_API_VERSION: if request.app.state.config.IMAGES_EDIT_OPENAI_API_VERSION: