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
This commit is contained in:
Davi S. Zucon 2025-03-10 15:59:08 -03:00 committed by GitHub
parent b03fc97e28
commit 846509bf9f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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"])