Browse Source

refac: images "url" fallback

Timothy Jaeryang Baek 2 months ago
parent
commit
9c107f9d2d
1 changed files with 7 additions and 1 deletions
  1. 7 1
      backend/open_webui/routers/images.py

+ 7 - 1
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