ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +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:
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
header, encoded = data.split(",", 1)
mime_type = header.split(";")[0].lstrip("data:")
image_data = base64.b64decode(encoded)
return (
"image",
param_name,
(
f"{uuid.uuid4()}.png",
io.BytesIO(image_data),
@ -879,7 +879,7 @@ async def image_edits(
files = [get_image_file_item(form_data.image)]
elif isinstance(form_data.image, list):
for img in form_data.image:
files.append(get_image_file_item(img))
files.append(get_image_file_item(img, "image[]"))
url_search_params = ""
if request.app.state.config.IMAGES_EDIT_OPENAI_API_VERSION: