pip install openai

from openai import OpenAI

client = OpenAI(
    base_url="https://api.nunchux.ai/v1",
    api_key="YOUR_API_KEY",
)

import base64
from pathlib import Path

response = client.images.generate(
    model="nunchux-flux.2-klein-4b",
    prompt="A beautiful sunset over mountains",
    size="1024x1024",
    response_format="b64_json",
    n=1,
    extra_body={
        "tier": "radical_speed",
        "seed": 42,
    },
)

image_b64 = response.data[0].b64_json
Path("output.png").write_bytes(base64.b64decode(image_b64))

response = client.images.edit(
    model="nunchux-flux.2-klein-4b-edit",
    image=open("photo.jpg", "rb"),
    prompt="Transform into a watercolor painting",
    size="1024x1024",
    response_format="url",
    extra_body={
        "tier": "radical_speed",
    },
)

# URL format requested above
print(response.data[0].url)

// Rejecting n=4
{
  "detail": "only n=1 is currently supported for image requests, got n=4",
  "request_id": "9f3c1e2a4b5d4f6e8a7b9c0d1e2f3a4b"
}

// Rejecting n="not-a-number" on a per-megapixel model (HTTP 400)
{
  "detail": "n must be a positive integer for per-megapixel billing, got 'not-a-number'",
  "request_id": "9f3c1e2a4b5d4f6e8a7b9c0d1e2f3a4b"
}