WaveSpeedAI の代替：そのまま使えるモデルと乗り換え手順

import os
import time
import requests

API = "https://api.omniakey.com"
HEADERS = {
    "Authorization": f"Bearer {os.environ['OMNIAKEY_API_KEY']}",
    "Content-Type": "application/json",
}

# 1. ジョブを作成します。Idempotency-Key があるので再試行しても安全です。
task = requests.post(
    f"{API}/v1/media/tasks",
    headers={**HEADERS, "Idempotency-Key": "switch-from-wavespeed-ai-001"},
    json={
        "model": "gemini-nano-banana-2.1",
        "type": "image",
        "operation": "generate",
        "input": {"prompt": "a red bicycle leaning on a white wall in morning light"},
        "parameters": {"aspect_ratio": "1:1", "image_size": "1K"},
    },
).json()

# 2. ジョブが終わるまでポーリングします。
while task["status"] in ("queued", "processing"):
    time.sleep(3)
    task = requests.get(f"{API}/v1/media/tasks/{task['id']}", headers=HEADERS).json()

if task["status"] != "succeeded":
    raise SystemExit(task.get("error"))

# 3. 署名付きリンクが切れる前に画像をダウンロードします。
open("out.jpg", "wb").write(requests.get(task["assets"][0]["url"]).content)
print(task["id"], task["assets"][0]["mime_type"], task["assets"][0]["byte_size"])