From 846509bf9fc1a925c27e800fb975e78e4b920e4c Mon Sep 17 00:00:00 2001 From: "Davi S. Zucon" Date: Mon, 10 Mar 2025 15:59:08 -0300 Subject: [PATCH] fix url value when response contains url tag When the response contains b64 json, the "url" tag still exists but comes as null, the logic is to resolve this point where it contains the url tag, but empty, thus directing to b64_json --- backend/open_webui/routers/images.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/routers/images.py b/backend/open_webui/routers/images.py index c51d2f996..e4ed9b62d 100644 --- a/backend/open_webui/routers/images.py +++ b/backend/open_webui/routers/images.py @@ -517,9 +517,9 @@ async def image_generations( images = [] for image in res["data"]: - if "url" in image: + if image_url := image.get("url",None): image_data, content_type = load_url_image_data( - image["url"], headers + image_url, headers ) else: image_data, content_type = load_b64_image_data(image["b64_json"])