mirror of https://github.com/open-webui/open-webui
Merge pull request #11515 from davizucon/main
fix: url value when response contains url tag
This commit is contained in:
commit
6399aaf4b2
|
@ -517,10 +517,8 @@ async def image_generations(
|
|||
images = []
|
||||
|
||||
for image in res["data"]:
|
||||
if "url" in image:
|
||||
image_data, content_type = load_url_image_data(
|
||||
image["url"], headers
|
||||
)
|
||||
if image_url := image.get("url", None):
|
||||
image_data, content_type = load_url_image_data(image_url, headers)
|
||||
else:
|
||||
image_data, content_type = load_b64_image_data(image["b64_json"])
|
||||
|
||||
|
|
Loading…
Reference in New Issue