From b9c03f6d4f761b27f3ea27fae76f338b4a30546f Mon Sep 17 00:00:00 2001
From: Christian Byrne <cbyrne@comfy.org>
Date: Fri, 2 May 2025 16:31:04 -0700
Subject: [PATCH] Fix: Dall-e 2 not setting request content-type dynamically
 (#113)

---
 comfy_api_nodes/nodes_openai.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/comfy_api_nodes/nodes_openai.py b/comfy_api_nodes/nodes_openai.py
index 612653a55..e3fbbb868 100644
--- a/comfy_api_nodes/nodes_openai.py
+++ b/comfy_api_nodes/nodes_openai.py
@@ -116,11 +116,13 @@ class OpenAIDalle2(ComfyNodeABC):
     ):
         model = "dall-e-2"
         path = "/proxy/openai/images/generations"
+        content_type = "application/json"
         request_class = OpenAIImageGenerationRequest
         img_binary = None
 
         if image is not None and mask is not None:
             path = "/proxy/openai/images/edits"
+            content_type = "multipart/form-data"
             request_class = OpenAIImageEditRequest
 
             input_tensor = image.squeeze().cpu()
@@ -166,6 +168,7 @@ class OpenAIDalle2(ComfyNodeABC):
                 if img_binary
                 else None
             ),
+            content_type=content_type,
             auth_token=auth_token,
         )
 
@@ -401,7 +404,7 @@ class OpenAIGPTImage1(ComfyNodeABC):
         if image is not None:
             path = "/proxy/openai/images/edits"
             request_class = OpenAIImageEditRequest
-            content_type="multipart/form-data",
+            content_type ="multipart/form-data"
 
             batch_size = image.shape[0]