Browse Source

fix: comfyui output image's format (#12121)

tags/0.15.0
非法操作 10 months ago
parent
commit
309fd76ddf
No account linked to committer's email address

+ 1
- 1
api/core/tools/provider/builtin/comfyui/tools/comfyui_client.py View File

for output in history["outputs"].values(): for output in history["outputs"].values():
for img in output.get("images", []): for img in output.get("images", []):
image_data = self.get_image(img["filename"], img["subfolder"], img["type"]) image_data = self.get_image(img["filename"], img["subfolder"], img["type"])
images.append(image_data)
images.append((image_data, img["filename"]))
return images return images
finally: finally:
ws.close() ws.close()

+ 5
- 2
api/core/tools/provider/builtin/comfyui/tools/comfyui_workflow.py View File

import json import json
import mimetypes
from typing import Any from typing import Any


from core.file import FileType from core.file import FileType


images = comfyui.generate_image_by_prompt(prompt) images = comfyui.generate_image_by_prompt(prompt)
result = [] result = []
for img in images:
for image_data, filename in images:
result.append( result.append(
self.create_blob_message( self.create_blob_message(
blob=img, meta={"mime_type": "image/png"}, save_as=self.VariableKey.IMAGE.value
blob=image_data,
meta={"mime_type": mimetypes.guess_type(filename)[0]},
save_as=self.VariableKey.IMAGE.value,
) )
) )
return result return result

Loading…
Cancel
Save