github-starred/open-webui
2
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-07-18 10:48:00 -05:00
Code Issues 2.4k Packages Projects Releases 129 Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-06-29 10:54:53 -05:00
parent 39837e0a3a
commit 2d3035a112
1 changed files with 9 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
backend/open_webui/routers/images.py
View File

@@ -153,9 +153,8 @@ def normalize_openai_edit_image_data_url(data_url: str) -> str:
return data_url
def get_image_file_item(base64_string, param_name='image', normalize=False):
data = normalize_openai_edit_image_data_url(base64_string) if normalize else base64_string
header, encoded = data.split(',', 1)
def get_image_file_item(base64_string, param_name='image'):
header, encoded = base64_string.split(',', 1)
mime_type = header.split(';')[0].lstrip('data:') or 'image/png'
image_data = base64.b64decode(encoded)
extension = IMAGE_FILE_EXTENSIONS.get(mime_type.lower()) or mimetypes.guess_extension(mime_type) or '.png'
@@ -1007,21 +1006,15 @@ async def image_edits(
files = []
if isinstance(form_data.image, str):
files = [
get_image_file_item(
form_data.image,
normalize=ENABLE_OPENAI_IMAGE_EDIT_NORMALIZATION,
)
]
image = form_data.image
if ENABLE_OPENAI_IMAGE_EDIT_NORMALIZATION:
image = normalize_openai_edit_image_data_url(image)
files = [get_image_file_item(image)]
elif isinstance(form_data.image, list):
for img in form_data.image:
files.append(
get_image_file_item(
img,
'image[]',
normalize=ENABLE_OPENAI_IMAGE_EDIT_NORMALIZATION,
)
)
if ENABLE_OPENAI_IMAGE_EDIT_NORMALIZATION:
img = normalize_openai_edit_image_data_url(img)
files.append(get_image_file_item(img, 'image[]'))
url_search_params = ''
if image_config.IMAGES_EDIT_OPENAI_API_VERSION: