diff --git a/backend/open_webui/routers/images.py b/backend/open_webui/routers/images.py index 131fa2df4..f8be93719 100644 --- a/backend/open_webui/routers/images.py +++ b/backend/open_webui/routers/images.py @@ -517,7 +517,13 @@ async def image_generations( images = [] 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) images.append({"url": url}) return images