mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-12 12:37:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(OpenAIGPTImage1): set correct MIME type for multipart uploads to OpenAI edits (#9348)

Browse Source
This commit is contained in:
Alexander Piskun
2025-08-15 21:57:47 +03:00
committed by GitHub
parent e08ecfbd8a
commit 027c63f63a
1 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
comfy_api_nodes/nodes_openai.py
View File

@@ -464,8 +464,6 @@ class OpenAIGPTImage1(ComfyNodeABC):
path = "/proxy/openai/images/generations" path = "/proxy/openai/images/generations"
content_type = "application/json" content_type = "application/json"
request_class = OpenAIImageGenerationRequest request_class = OpenAIImageGenerationRequest
img_binaries = []
mask_binary = None
files = [] files = []
if image is not None: if image is not None:
@@ -484,14 +482,11 @@ class OpenAIGPTImage1(ComfyNodeABC):
img_byte_arr = io.BytesIO() img_byte_arr = io.BytesIO()
img.save(img_byte_arr, format="PNG") img.save(img_byte_arr, format="PNG")
img_byte_arr.seek(0) img_byte_arr.seek(0)
img_binary = img_byte_arr
img_binary.name = f"image_{i}.png"
img_binaries.append(img_binary)
if batch_size == 1: if batch_size == 1:
files.append(("image", img_binary)) files.append(("image", (f"image_{i}.png", img_byte_arr, "image/png")))
else: else:
files.append(("image[]", img_binary)) files.append(("image[]", (f"image_{i}.png", img_byte_arr, "image/png")))
if mask is not None: if mask is not None:
if image is None: if image is None:
@@ -511,9 +506,7 @@ class OpenAIGPTImage1(ComfyNodeABC):
mask_img_byte_arr = io.BytesIO() mask_img_byte_arr = io.BytesIO()
mask_img.save(mask_img_byte_arr, format="PNG") mask_img.save(mask_img_byte_arr, format="PNG")
mask_img_byte_arr.seek(0) mask_img_byte_arr.seek(0)
mask_binary = mask_img_byte_arr files.append(("mask", ("mask.png", mask_img_byte_arr, "image/png")))
mask_binary.name = "mask.png"
files.append(("mask", mask_binary))
# Build the operation # Build the operation
operation = SynchronousOperation( operation = SynchronousOperation(