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

Return tuple consistently and strip potential left string

Browse source
This commit is contained in:
Rodrigo Agundez 2025-06-08 16:26:08 +08:00
parent 5e35aab292
commit 168997a58e
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -420,7 +420,7 @@ def load_b64_image_data(b64_str):
try: try:
if "," in b64_str: if "," in b64_str:
header, encoded = b64_str.split(",", 1) header, encoded = b64_str.split(",", 1)
mime_type = header.split(";")[0] mime_type = header.split(";")[0].lstrip("data:")
img_data = base64.b64decode(encoded) img_data = base64.b64decode(encoded)
else: else:
mime_type = "image/png" mime_type = "image/png"
@ -428,7 +428,7 @@ def load_b64_image_data(b64_str):
return img_data, mime_type return img_data, mime_type
except Exception as e: except Exception as e:
log.exception(f"Error loading image data: {e}") log.exception(f"Error loading image data: {e}")
return None return None, None
def load_url_image_data(url, headers=None): def load_url_image_data(url, headers=None):