图片 API 调用须知与最佳实践

# 按模型设置客户端超时（秒），而不是全局一个值
IMAGE_TIMEOUTS = {
    "gpt-image-2": 360,                      # high + 2K/4K 实测 3-5 分钟
    "gpt-image-2-all": 300,
    "gpt-image-2-vip": 300,
    "gemini-3-pro-image": 600,               # Nano Banana Pro，4K 按 600s 兜底
    "gemini-3.1-flash-image-preview": 360,   # Nano Banana 2
    "gemini-3.1-flash-lite-image": 300,      # Nano Banana Lite
    "flux": 120,                             # flex 系列建议 180
    "seedream": 60,                          # 4K + hd 约 30-60 秒
}

import requests

def generate_image(model: str, payload: dict, api_key: str) -> dict:
    resp = requests.post(
        "https://api.apiyi.com/v1/images/generations",
        headers={"Authorization": f"Bearer {api_key}"},
        json={"model": model, **payload},
        timeout=IMAGE_TIMEOUTS.get(model, 300),  # 未知模型给 300s 兜底
    )
    resp.raise_for_status()
    return resp.json()

import base64

b64 = response["data"][0]["b64_json"]
if b64.startswith("data:"):          # 防御性剥前缀，兼容曾出现过的 data: 前缀响应
    b64 = b64.split(",", 1)[1]
with open("output.png", "wb") as f:
    f.write(base64.b64decode(b64))

let b64 = response.data[0].b64_json;
if (b64.startsWith("data:")) {
  b64 = b64.slice(b64.indexOf(",") + 1);
}
require("fs").writeFileSync("output.png", Buffer.from(b64, "base64"));

{
  "error": {
    "message": "Invalid image file or mode for image 1, please check your image file. ...",
    "code": "invalid_image_file"
  }
}

from PIL import Image
Image.open("photo.jpg").format   # 返回 "MPO" 即中招；标准图返回 "JPEG"/"PNG"

from PIL import Image
import io

def normalize_image(raw: bytes) -> bytes:
    """任意来源图片 → 标准 JPEG，通过各图片编辑接口的格式校验"""
    im = Image.open(io.BytesIO(raw))
    im.load()                      # 多帧格式（MPO 等）只取第一帧
    if im.mode not in ("RGB", "RGBA"):
        im = im.convert("RGB")     # CMYK / P 等模式统一转 RGB
    out = io.BytesIO()
    im.save(out, format="JPEG", quality=92)
    return out.getvalue()