Image Generation

import os
from openai import OpenAI

# set env variables
os.environ["LLMGRID_API_KEY"] = "your-llmgrid-virtual-key"

client = OpenAI(
  api_key=os.environ["LLMGRID_API_KEY"],
  base_url="https://api.llmgrid.ai/v1"
)

response = client.images.generate(
  model="gpt-image-1",
  prompt="A modern retail storefront with colorful product displays, high detail, soft lighting",
  size="1024x1024",
  n=1,
  response_format="url"
)

print(response.data[0].url)

curl -X POST "https://api.llmgrid.ai/v1/images/generations" \
  -H "Authorization: Bearer $LLMGRID_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gpt-image-1",
    "prompt": "A modern retail storefront with colorful product displays, high detail, soft lighting",
    "n": 1,
    "size": "1024x1024",
    "response_format": "url"
  }'

  def image_generation(
    prompt: str,
    # Optional OpenAI-compatible params
    model: str | None = None,
    n: int | None = None,
    size: str | None = None,
    quality: str | None = None,
    style: str | None = None,
    response_format: str | None = None,  # "url" or "b64_json"
    user: str | None = None,
    timeout: int | float | None = None,
    # Optional gateway/provider passthrough
    **kwargs,
):
    ...

{
  "created": 1710000000,
  "data": [
    {
      "url": "https://..."
    }
  ]
}

{
  "created": 1710000000,
  "data": [
    {
      "b64_json": "iVBORw0KGgoAAA..."
    }
  ]
}

import base64

b64 = response.data[0].b64_json
img_bytes = base64.b64decode(b64)

with open("output.png", "wb") as f:
    f.write(img_bytes)