소스 검색

refac: images "url" fallback

Timothy Jaeryang Baek 2 달 전
부모
커밋
9c107f9d2d
1개의 변경된 파일7개의 추가작업 그리고 1개의 파일을 삭제
  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