From 5d3e0557531816713b8a3cdf4846d5cb596ffcdd Mon Sep 17 00:00:00 2001 From: Robin Huang Date: Mon, 5 May 2025 22:36:55 -0700 Subject: [PATCH] Kling Image Gen v2 + improve node descriptions for Flux/OpenAI (#160) --- comfy_api_nodes/apis/__init__.py | 1 + comfy_api_nodes/nodes_bfl.py | 2 +- comfy_api_nodes/nodes_openai.py | 9 --------- 3 files changed, 2 insertions(+), 10 deletions(-) diff --git a/comfy_api_nodes/apis/__init__.py b/comfy_api_nodes/apis/__init__.py index cb448f4fc..aa1c4ce0b 100644 --- a/comfy_api_nodes/apis/__init__.py +++ b/comfy_api_nodes/apis/__init__.py @@ -568,6 +568,7 @@ class KlingImageGenImageReferenceType(str, Enum): class KlingImageGenModelName(str, Enum): kling_v1 = 'kling-v1' kling_v1_5 = 'kling-v1-5' + kling_v2 = 'kling-v2' class KlingImageResult(BaseModel): diff --git a/comfy_api_nodes/nodes_bfl.py b/comfy_api_nodes/nodes_bfl.py index 7f02df88e..122a6ddf8 100644 --- a/comfy_api_nodes/nodes_bfl.py +++ b/comfy_api_nodes/nodes_bfl.py @@ -112,7 +112,7 @@ def convert_image_to_base64(image: torch.Tensor): class FluxProUltraImageNode(ComfyNodeABC): """ - Generates images synchronously based on prompt and resolution. + Generates images using Flux Pro 1.1 Ultra via api based on prompt and resolution. """ MINIMUM_RATIO = 1 / 4 diff --git a/comfy_api_nodes/nodes_openai.py b/comfy_api_nodes/nodes_openai.py index 88db82f09..c18c65d7a 100644 --- a/comfy_api_nodes/nodes_openai.py +++ b/comfy_api_nodes/nodes_openai.py @@ -28,9 +28,6 @@ from comfy_api_nodes.apinode_utils import ( class OpenAIDalle2(ComfyNodeABC): """ Generates images synchronously via OpenAI's DALL·E 2 endpoint. - - Uses the proxy at /proxy/openai/images/generations. Returned URLs are short‑lived, - so download or cache results if you need to keep them. """ def __init__(self): @@ -183,9 +180,6 @@ class OpenAIDalle2(ComfyNodeABC): class OpenAIDalle3(ComfyNodeABC): """ Generates images synchronously via OpenAI's DALL·E 3 endpoint. - - Uses the proxy at /proxy/openai/images/generations. Returned URLs are short‑lived, - so download or cache results if you need to keep them. """ def __init__(self): @@ -291,9 +285,6 @@ class OpenAIDalle3(ComfyNodeABC): class OpenAIGPTImage1(ComfyNodeABC): """ Generates images synchronously via OpenAI's GPT Image 1 endpoint. - - Uses the proxy at /proxy/openai/images/generations. Returned URLs are short‑lived, - so download or cache results if you need to keep them. """ def __init__(self):