From aedb281628370463a950816041c91ef55dc35f09 Mon Sep 17 00:00:00 2001 From: Marigold1122 Date: Wed, 20 May 2026 04:02:47 +0800 Subject: [PATCH] chore: rename drawing node to GPT Image --- README.md | 18 +++++++++--------- chat_image_bridge.py | 42 +++++++++++++++++++++--------------------- 2 files changed, 30 insertions(+), 30 deletions(-) diff --git a/README.md b/README.md index 7593132..57ef6b2 100644 --- a/README.md +++ b/README.md @@ -1,8 +1,8 @@ # ComfyUI Chat Image Bridge -一个用于 ComfyUI 的第三方 API 生图节点集合，主要面向 OpenAI-compatible 聊天接口，也包含面向 Right Codes 绘图接口的专用节点。 +一个用于 ComfyUI 的第三方 API 生图节点集合，主要面向 OpenAI-compatible 聊天接口，也包含 GPT Image 专用请求节点。 -仓库不会内置任何 API Key。通用节点不内置服务商地址；Right Codes 专用节点会预填官方 `base_url`，方便直接使用，也可以自行修改。 +仓库不会内置任何 API Key 或服务商地址。用户需要在节点里填写自己的 `base_url`、`api_key` 和模型名。 ## 功能 @@ -13,7 +13,7 @@ - 支持 1K、2K、4K 分辨率和常见长宽比 - 支持 1-2 张参考图输入 - 支持解析 Markdown 图片、图片 URL、`b64_json`、Gemini `inlineData` -- 提供 Right Codes GPT Image 专用节点，使用流式 `/v1/chat/completions` +- 提供 GPT Image 专用节点，使用流式 `/v1/chat/completions` - 提供高级节点，可输出原始响应和图片引用，方便调试 ## 安装 @@ -74,15 +74,15 @@ api -> Chat Image Bridge -> Chat Image Bridge 可以直接接到 `Save Image` 或其他 ComfyUI 图像节点。 -## Right Codes GPT Image 节点 +## GPT Image 节点 添加节点： ```text -api -> Chat Image Bridge -> Right Codes GPT Image +api -> Chat Image Bridge -> GPT Image ``` -这个节点面向 Right Codes 绘图接口，默认使用： +这个节点面向 GPT Image / Nano Banana 这类绘图接口，使用流式 `/v1/chat/completions`。如果你的服务商使用 Right Codes draw 格式，`base_url` 可以填写： ```text https://www.right.codes/draw/v1/chat/completions @@ -120,7 +120,7 @@ https://example.com/v1/chat/completions 节点会自动整理为请求需要的接口地址。 -Right Codes 专用节点默认填的是： +如果使用 Right Codes draw 格式，可以填： ```text https://www.right.codes/draw @@ -174,7 +174,7 @@ gemini-3-pro-image-preview 这样 `4K` 和长宽比会作为真正的图片参数发送，而不是只写进 prompt。 -Right Codes GPT Image 节点则会按接口文档传入 `size`，例如 `3840x2160`。 +GPT Image 节点则会按接口文档传入 `size`，例如 `3840x2160`。 ## 高级节点 @@ -197,5 +197,5 @@ api -> Chat Image Bridge -> Chat Image Bridge Advanced - API Key 只在节点 UI 里填写，本仓库不会硬编码任何 Key。 - 不同中转站对 OpenAI-compatible 的兼容程度不同，字段支持可能不完全一致。 -- `gpt-image-2` 和 `nano-banana` 在 Right Codes 节点里只开放 1K，选择更高分辨率会直接报错，避免请求失败或产生无效消耗。 +- `gpt-image-2` 和 `nano-banana` 在 GPT Image 节点里只开放 1K，选择更高分辨率会直接报错，避免请求失败或产生无效消耗。 - 旧版工作流如果使用过 `endpoint_url`，建议改成新版的 `base_url`。 diff --git a/chat_image_bridge.py b/chat_image_bridge.py index 2d722d2..bf22829 100644 --- a/chat_image_bridge.py +++ b/chat_image_bridge.py @@ -23,18 +23,18 @@ GRSAI_PROXY_RE = re.compile( RESOLUTION_OPTIONS = ["auto", "1K", "2K", "4K"] ASPECT_RATIO_OPTIONS = ["auto", "1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3", "4:5", "5:4", "21:9"] DEFAULT_MODEL = "gemini-3-pro-image-preview" -RIGHT_CODES_DEFAULT_BASE_URL = "https://www.right.codes/draw" -RIGHT_CODES_MODELS = ["gpt-image-2-vip", "gpt-image-2", "nano-banana", "nano-banana-2", "nano-banana-pro"] -RIGHT_CODES_RESOLUTION_OPTIONS = ["1K", "2K", "4K"] -RIGHT_CODES_ASPECT_RATIO_OPTIONS = ["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3", "4:5", "5:4", "21:9"] -RIGHT_CODES_MODEL_RESOLUTIONS = { +GPT_IMAGE_DEFAULT_BASE_URL = "" +GPT_IMAGE_MODELS = ["gpt-image-2-vip", "gpt-image-2", "nano-banana", "nano-banana-2", "nano-banana-pro"] +GPT_IMAGE_RESOLUTION_OPTIONS = ["1K", "2K", "4K"] +GPT_IMAGE_ASPECT_RATIO_OPTIONS = ["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3", "4:5", "5:4", "21:9"] +GPT_IMAGE_MODEL_RESOLUTIONS = { "gpt-image-2-vip": {"1K", "2K", "4K"}, "gpt-image-2": {"1K"}, "nano-banana": {"1K"}, "nano-banana-2": {"1K", "2K", "4K"}, "nano-banana-pro": {"1K", "2K", "4K"}, } -RIGHT_CODES_SIZE_TABLE = { +GPT_IMAGE_SIZE_TABLE = { "1K": { "1:1": "1280x1280", "2:3": "848x1280", @@ -163,22 +163,22 @@ def normalize_generation_option(value): return value -def right_codes_size_for(model, resolution, aspect_ratio): +def gpt_image_size_for(model, resolution, aspect_ratio): model = (model or "").strip() resolution = (resolution or "").strip() aspect_ratio = (aspect_ratio or "").strip() - supported = RIGHT_CODES_MODEL_RESOLUTIONS.get(model) + supported = GPT_IMAGE_MODEL_RESOLUTIONS.get(model) if not supported: - raise ValueError(f"Unsupported Right Codes image model: {model}") + raise ValueError(f"Unsupported GPT Image model: {model}") if resolution not in supported: allowed = ", ".join(sorted(supported)) raise ValueError(f"{model} supports resolution {allowed}, but got {resolution}") try: - return RIGHT_CODES_SIZE_TABLE[resolution][aspect_ratio] + return GPT_IMAGE_SIZE_TABLE[resolution][aspect_ratio] except KeyError as exc: - raise ValueError(f"Unsupported Right Codes size option: {resolution} {aspect_ratio}") from exc + raise ValueError(f"Unsupported GPT Image size option: {resolution} {aspect_ratio}") from exc def tensor_to_png_data_url(tensor): @@ -682,19 +682,19 @@ class ChatImageBridgeAdvanced(ChatImageBridgeBase): ) -class RightCodesGPTImage(ChatImageBridgeBase): - """Right Codes draw node using streaming chat/completions requests.""" +class GPTImage(ChatImageBridgeBase): + """GPT image node using streaming chat/completions requests.""" @classmethod def INPUT_TYPES(cls): return { "required": { "api_key": ("STRING", {"default": "", "multiline": False}), - "base_url": ("STRING", {"default": RIGHT_CODES_DEFAULT_BASE_URL, "multiline": False}), - "model": (RIGHT_CODES_MODELS, {"default": "gpt-image-2-vip"}), + "base_url": ("STRING", {"default": GPT_IMAGE_DEFAULT_BASE_URL, "multiline": False}), + "model": (GPT_IMAGE_MODELS, {"default": "gpt-image-2-vip"}), "prompt": ("STRING", {"default": "", "multiline": True}), - "resolution": (RIGHT_CODES_RESOLUTION_OPTIONS, {"default": "1K"}), - "aspect_ratio": (RIGHT_CODES_ASPECT_RATIO_OPTIONS, {"default": "1:1"}), + "resolution": (GPT_IMAGE_RESOLUTION_OPTIONS, {"default": "1K"}), + "aspect_ratio": (GPT_IMAGE_ASPECT_RATIO_OPTIONS, {"default": "1:1"}), "timeout_seconds": ("INT", {"default": 600, "min": 60, "max": 3600}), }, "optional": { @@ -805,7 +805,7 @@ class RightCodesGPTImage(ChatImageBridgeBase): preview = json.dumps(redact_object(data), ensure_ascii=False)[:2000] if raw_preview and preview == '{"stream": true, "content": "", "chunks": []}': preview = "\n".join(raw_preview)[:2000] - raise RuntimeError(f"Right Codes stream did not contain image data or image URLs: {preview}") + raise RuntimeError(f"GPT Image stream did not contain image data or image URLs: {preview}") return data, refs @@ -818,7 +818,7 @@ class RightCodesGPTImage(ChatImageBridgeBase): raise ValueError("prompt is required") model = (model or "").strip() - size = right_codes_size_for(model, resolution, aspect_ratio) + size = gpt_image_size_for(model, resolution, aspect_ratio) endpoint = normalize_endpoint(base_url) payload = self._build_payload(model, prompt, size, [image_1, image_2]) _, refs = self._post_streaming_chat(endpoint, api_key, payload, int(timeout_seconds)) @@ -829,11 +829,11 @@ class RightCodesGPTImage(ChatImageBridgeBase): NODE_CLASS_MAPPINGS = { "ChatImageBridge": ChatImageBridge, "ChatImageBridgeAdvanced": ChatImageBridgeAdvanced, - "RightCodesGPTImage": RightCodesGPTImage, + "GPTImage": GPTImage, } NODE_DISPLAY_NAME_MAPPINGS = { "ChatImageBridge": "Chat Image Bridge", "ChatImageBridgeAdvanced": "Chat Image Bridge Advanced", - "RightCodesGPTImage": "Right Codes GPT Image", + "GPTImage": "GPT Image", }