ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-01-02 22:55:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-12-26 18:30:50 +04:00
parent d29271184e
commit 9c2f5148d9
1 changed files with 8 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
backend/open_webui/utils/images/comfyui.py
View file

@ -79,15 +79,14 @@ def get_images(ws, prompt, client_id, base_url, api_key):
continue # previews are binary data continue # previews are binary data
history = get_history(prompt_id, base_url, api_key)[prompt_id] history = get_history(prompt_id, base_url, api_key)[prompt_id]
for o in history["outputs"]: for node_id in history["outputs"]:
for node_id in history["outputs"]: node_output = history["outputs"][node_id]
node_output = history["outputs"][node_id] if "images" in node_output:
if "images" in node_output: for image in node_output["images"]:
for image in node_output["images"]: url = get_image_url(
url = get_image_url( image["filename"], image["subfolder"], image["type"], base_url
image["filename"], image["subfolder"], image["type"], base_url )
) output_images.append({"url": url})
output_images.append({"url": url})
return {"data": output_images} return {"data": output_images}