OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-20 05:07:59 +00:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

refac: images "url" fallback

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-03-03 20:07:59 -08:00
parent 46cea1f904
commit 9c107f9d2d
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -517,7 +517,13 @@ async def image_generations(
images = [] images = []
for image in res["data"]: for image in res["data"]:
image_data, content_type = load_b64_image_data(image["b64_json"]) if "url" in image:
image_data, content_type = load_url_image_data(
image["url"], headers
)
else:
image_data, content_type = load_b64_image_data(image["b64_json"])
url = upload_image(request, data, image_data, content_type, user) url = upload_image(request, data, image_data, content_type, user)
images.append({"url": url}) images.append({"url": url})
return images return images