GPT Image 2.5 流式返回

curl --fail-with-body --silent --show-error --max-time 600 --no-buffer \
  --request POST \
  --url https://code.heihuzi.ai/v1/images/generations \
  --header "Authorization: Bearer $HEIHUZI_API_KEY" \
  --header 'Content-Type: application/json' \
  --data '{
  "model": "gpt-image-2.5-sunburst",
  "prompt": "A simple blue ceramic cup on a white background.",
  "n": 1,
  "size": "1024x1024",
  "quality": "low",
  "output_format": "png",
  "stream": true,
  "partial_images": 1
}' \
  --output stream.sse

import base64
import os
from pathlib import Path
from openai import OpenAI

client = OpenAI(
    api_key=os.environ["HEIHUZI_API_KEY"],
    base_url="https://code.heihuzi.ai/v1",
    timeout=600.0,
    max_retries=0,
)
events = client.images.generate(
    model="gpt-image-2.5-sunburst",
    prompt="A simple blue ceramic cup on a white background.",
    n=1,
    size="1024x1024",
    quality="low",
    output_format="png",
    stream=True,
    partial_images=1,
)
completed = 0
for event in events:
    print(event.type, flush=True)
    if event.type == "image_generation.partial_image":
        Path(f"preview-{event.partial_image_index}.png").write_bytes(
            base64.b64decode(event.b64_json)
        )
    elif event.type == "image_generation.completed":
        completed += 1
        Path(f"final-{completed}.png").write_bytes(base64.b64decode(event.b64_json))
if completed != 1:
    raise RuntimeError(f"Expected 1 final image, got {completed}")