feat: 发布 gpt-image-1.5 生图模型

- 新增 gpt-image-1.5 / gpt-image-1.5-特价模型选项 - 图生图统一走 edits 接口，支持多图批次输入（image[]） - 节点参数中文化：模型、分辨率、生图数量、图片、遮罩 - 移除 quality / background 前端参数（服务端暂不支持） - 请求超时上限调整为 900s
2026-04-16 01:27:39 +08:00
parent a2665b4010
commit c4bb8d9724
2 changed files with 94 additions and 86 deletions
@@ -31,8 +31,14 @@ from ..utils.image_utils import tensor_to_pil, encode_image_to_base64
 _ENDPOINT_GENERATIONS = "/v1/images/generations/"
 _ENDPOINT_EDITS       = "/v1/images/edits/"

+# ── 模型名映射（UI 显示名 → API 实际参数名）─────────────────────────────────
+_MODEL_NAME_MAP = {
+    "gpt-image-1-特价":   "gpt-image-1-special",
+    "gpt-image-1.5-特价": "gpt-image-1.5-special",
+}
+
 # ── 超时 ──────────────────────────────────────────────────────────────────────
-_REQUEST_TIMEOUT = 300   # 秒
+_REQUEST_TIMEOUT = 900   # 秒


 class GptImageClient:
@@ -196,8 +202,11 @@ class GptImageClient:
        调用 /v1/images/generations/ 接口。
        当传入 image_tensor 时，以 data URI 格式内联图像（图生图）。
        """
+        # 模型名映射：UI 显示名 → API 参数名
+        api_model = _MODEL_NAME_MAP.get(model, model)
+
        body: dict = {
-            "model":      model,
+            "model":      api_model,
            "prompt":     prompt,
            "quality":    quality,
            "background": background,
@@ -259,7 +268,9 @@ class GptImageClient:
            print(f"[o1key GPT Image] API 响应耗时 {elapsed:.1f}s")
            return await self._parse_response(resp_json, session)

-    # ── 图像编辑（edits 接口，必须带蒙版）───────────────────────────────────
+    # ── 图像编辑（edits 接口，multipart/form-data）──────────────────────────
+    # 注意：o1key 中转服务的 edits 接口暂不支持 quality / background / moderation 参数，
+    # 这些字段暂时不传递，待服务方更新后可恢复。

    async def _edit_async(
        self,
@@ -275,37 +286,40 @@ class GptImageClient:
    ) -> List[Image.Image]:
        """
        调用 /v1/images/edits/ 接口（multipart/form-data）。
-        image_tensor 取第一帧（edits 接口仅支持单张输入图）。
+        当前仅传递 model / prompt / n / size / image / mask，
+        quality / background / moderation 暂不支持（o1key 服务端限制）。
        """
-        # 取第一张图
-        single = image_tensor[:1] if image_tensor.dim() == 4 else image_tensor.unsqueeze(0)
-        image_png = self._tensor_to_png_bytes(single)
-        ih, iw = single.shape[1], single.shape[2]
+        # 模型名映射：UI 显示名 → API 参数名
+        api_model = _MODEL_NAME_MAP.get(model, model)

-        # 构建 multipart 表单
+        # 将 batch tensor 拆成逐帧列表
+        if image_tensor.dim() == 3:
+            image_tensor = image_tensor.unsqueeze(0)   # [H,W,C] → [1,H,W,C]
+        num_images = image_tensor.shape[0]
+
+        # o1key 中转服务的 edits 接口暂不支持 quality / background / moderation / seed，
+        # 待服务方更新后可重新加入。
        form = aiohttp.FormData()
-        form.add_field("model",      model)
+        form.add_field("model",  api_model)
        form.add_field("prompt", prompt)
-        form.add_field("quality",    quality)
-        form.add_field("background", background)
        form.add_field("n",      str(n))
-        form.add_field("moderation", "low")

        if size and size != "auto":
            form.add_field("size", size)

-        if seed > 0:
-            form.add_field("seed", str(seed))
-
-        # 主图（PNG）
+        # 多图：用 image[] 数组字段逐张附加，支持 gpt-image-1.5 最多 16 张
+        for i in range(num_images):
+            frame = image_tensor[i:i+1]               # [1,H,W,C]
+            img_bytes = self._tensor_to_png_bytes(frame)
            form.add_field(
-            "image",
-            image_png,
-            filename="image.png",
+                "image[]",
+                img_bytes,
+                filename=f"image_{i}.png",
                content_type="image/png",
            )

-        # 蒙版（PNG，RGBA 格式，透明区域为待编辑区）
+        ih, iw = image_tensor.shape[1], image_tensor.shape[2]
+
        if mask_tensor is not None:
            mask_png = self._mask_tensor_to_rgba_png_bytes(mask_tensor, (ih, iw))
            form.add_field(
@@ -319,7 +333,7 @@ class GptImageClient:
            mode = "图像编辑（无蒙版）"

        url = f"{self.base_url}{_ENDPOINT_EDITS}"
-        print(f"[o1key GPT Image] {mode} | 模型={model} | quality={quality} | "
+        print(f"[o1key GPT Image] {mode} | 模型={model} | 参考图={num_images}张 | quality={quality} | "
              f"background={background} | size={size} | n={n}")

        connector = aiohttp.TCPConnector(ssl=False, force_close=True)
@@ -330,7 +344,7 @@ class GptImageClient:
            async with session.post(
                url,
                data=form,
-                headers=self._auth_headers(),   # Content-Type 由 FormData 自动设置
+                headers=self._auth_headers(),
            ) as resp:
                elapsed = time.time() - t0
                text = await resp.text()
@@ -374,11 +388,12 @@ class GptImageClient:
        同步入口，在独立线程中运行事件循环，避免与 ComfyUI 主循环冲突。

        路由逻辑：
-          - 无 image_tensor           → generations 接口（文生图）
-          - 有 image_tensor，无 mask  → generations 接口（图生图，data URI）
-          - 有 image_tensor，有 mask  → edits 接口（图像编辑 + 蒙版）
+          - 无 image_tensor  → generations 接口（文生图，JSON body）
+          - 有 image_tensor  → edits 接口（图生图/编辑，multipart/form-data）
+            所有模型统一走 multipart，quality/background 通过表单字段传递，
+            new-api 开启"透传请求体"后原样转发给上游。
        """
-        use_edits = (image_tensor is not None and mask_tensor is not None)
+        use_edits = (image_tensor is not None)

        if use_edits:
            coro = self._edit_async(
@@ -15,19 +15,17 @@ class O1keyGPTImage:

    功能：
      - 文生图：仅提供 prompt
-      - 图生图：提供 prompt + image（无 mask）
-      - 图像编辑：提供 prompt + image + mask（白色区域将被替换）
+      - 图生图：提供 prompt + 图片（无遮罩）
+      - 图像编辑：提供 prompt + 图片 + 遮罩（白色区域将被替换）

    参数：
      - prompt   : 文本提示词（多行）
+      - 模型     : 模型选择
+      - 分辨率   : 图像尺寸（auto 让 API 自动决定）
+      - 生图数量 : 生成数量 1-8
      - seed     : 随机种子（0 表示不指定）
-      - quality     : 图像质量 low / medium / high
-      - background  : 背景模式 auto / opaque / transparent
-      - size        : 图像尺寸（auto 让 API 自动决定）
-      - n           : 生成数量 1-8
-      - image       : 可选参考图（用于图生图或编辑）
-      - mask        : 可选蒙版（白色区域将被替换）
-      - model       : 模型选择 gpt-image-1 / gpt-image-1.5
+      - 图片     : 可选参考图（用于图生图或编辑）
+      - 遮罩     : 可选蒙版（白色区域将被替换）
    """

    @classmethod
@@ -41,6 +39,26 @@ class O1keyGPTImage:
                }),
            },
            "optional": {
+                "模型": ([
+                    "gpt-image-1",
+                    "gpt-image-1.5",
+                    "gpt-image-1-特价",
+                    "gpt-image-1.5-特价",
+                ], {
+                    "default": "gpt-image-1.5",
+                }),
+                "分辨率": (["auto", "1024x1024", "1024x1536", "1536x1024"], {
+                    "default": "auto",
+                    "tooltip": "Image size (auto = API decides)",
+                }),
+                "生图数量": ("INT", {
+                    "default": 1,
+                    "min": 1,
+                    "max": 8,
+                    "step": 1,
+                    "display": "number",
+                    "tooltip": "How many images to generate",
+                }),
                "seed": ("INT", {
                    "default": 0,
                    "min": 0,
@@ -50,35 +68,12 @@ class O1keyGPTImage:
                    "control_after_generate": True,
                    "tooltip": "Random seed (0 = not specified)",
                }),
-                "quality": (["low", "medium", "high"], {
-                    "default": "low",
-                    "tooltip": "Image quality, affects cost and generation time.",
-                }),
-                "background": (["auto", "opaque", "transparent"], {
-                    "default": "auto",
-                    "tooltip": "Return image with or without background",
-                }),
-                "size": (["auto", "1024x1024", "1024x1536", "1536x1024"], {
-                    "default": "auto",
-                    "tooltip": "Image size (auto = API decides)",
-                }),
-                "n": ("INT", {
-                    "default": 1,
-                    "min": 1,
-                    "max": 8,
-                    "step": 1,
-                    "display": "number",
-                    "tooltip": "How many images to generate",
-                }),
-                "image": ("IMAGE", {
+                "图片": ("IMAGE", {
                    "tooltip": "Optional reference image for image editing.",
                }),
-                "mask": ("MASK", {
+                "遮罩": ("MASK", {
                    "tooltip": "Optional mask for inpainting (white areas will be replaced)",
                }),
-                "model": (["gpt-image-1", "gpt-image-1.5"], {
-                    "default": "gpt-image-1.5",
-                }),
            },
        }

@@ -91,22 +86,20 @@ class O1keyGPTImage:
    def generate(
        self,
        prompt: str,
+        模型: str = "gpt-image-1.5",
+        分辨率: str = "auto",
+        生图数量: int = 1,
        seed: int = 0,
-        quality: str = "low",
-        background: str = "auto",
-        size: str = "auto",
-        n: int = 1,
-        image=None,
-        mask=None,
-        model: str = "gpt-image-1.5",
+        图片=None,
+        遮罩=None,
    ):
        """
        生成图像（文生图 / 图生图 / 图像编辑）

        路由逻辑：
-          - 无 image           → generations 接口（文生图）
-          - 有 image，无 mask  → generations 接口（图生图）
-          - 有 image，有 mask  → edits 接口（图像编辑 + 蒙版）
+          - 无图片           → generations 接口（文生图）
+          - 有图片，无遮罩   → edits 接口（图生图）
+          - 有图片，有遮罩   → edits 接口（图像编辑 + 蒙版）
        """
        start_time = time.time()

@@ -114,8 +107,8 @@ class O1keyGPTImage:
        if not prompt or not prompt.strip():
            raise ValueError("提示词不能为空")

-        if mask is not None and image is None:
-            raise ValueError("提供了蒙版但未提供图像，请同时提供 image 和 mask")
+        if 遮罩 is not None and 图片 is None:
+            raise ValueError("提供了遮罩但未提供图片，请同时提供图片和遮罩")

        # ── 2. 创建客户端 ─────────────────────────────────────────────────────
        try:
@@ -130,14 +123,14 @@ class O1keyGPTImage:
        try:
            pil_images = client.run_sync(
                prompt=prompt,
-                model=model,
-                quality=quality,
-                background=background,
-                size=size,
-                n=n,
+                model=模型,
+                quality="low",
+                background="auto",
+                size=分辨率,
+                n=生图数量,
                seed=seed,
-                image_tensor=image,
-                mask_tensor=mask,
+                image_tensor=图片,
+                mask_tensor=遮罩,
            )
        except Exception as e:
            error_msg = str(e).split('\n')[0]