Image generation

import os
import base64
from openai import OpenAI

client = OpenAI(
    api_key=os.environ["ARBYTRA_API_KEY"],
    base_url="https://api.arbytra.com/v1",
)

response = client.chat.completions.create(
    model="gemini-2.5-flash-image",
    messages=[{
        "role": "user",
        "content": "Draw a cartoon cat wearing a top hat",
    }],
)

image = response.choices[0].message.images[0]
image_bytes = base64.b64decode(image["data"])

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

print(f"Saved {image['mime_type']} image ({len(image_bytes)} bytes)")

import OpenAI from "openai";
import { writeFileSync } from "fs";

const client = new OpenAI({
    apiKey: process.env.ARBYTRA_API_KEY,
    baseURL: "https://api.arbytra.com/v1",
});

const response = await client.chat.completions.create({
    model: "gemini-2.5-flash-image",
    messages: [{
        role: "user",
        content: "Draw a cartoon cat wearing a top hat",
    }],
});

const image = response.choices[0].message.images[0];
const imageBytes = Buffer.from(image.data, "base64");

writeFileSync("output.png", imageBytes);

console.log(`Saved ${image.mime_type} image (${imageBytes.length} bytes)`);

import os
import base64
from arbytra import Client

client = Client(
    api_key=os.environ["ARBYTRA_API_KEY"],
    base_url="https://api.arbytra.com/v1",
)

response = client.chat.completions.create(
    model="gemini-2.5-flash-image",
    messages=[{
        "role": "user",
        "content": "Draw a cartoon cat wearing a top hat",
    }],
)

image = response.choices[0].message.images[0]
image_bytes = base64.b64decode(image["data"])

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

print(f"Saved {image['mime_type']} image ({len(image_bytes)} bytes)")

import { Client } from "@arbytra/sdk";
import { writeFileSync } from "fs";

const client = new Client({
    apiKey: process.env.ARBYTRA_API_KEY,
    baseUrl: "https://api.arbytra.com/v1",
});

const response = await client.chat.completions.create({
    model: "gemini-2.5-flash-image",
    messages: [{
        role: "user",
        content: "Draw a cartoon cat wearing a top hat",
    }],
});

const image = response.choices[0].message.images[0];
const imageBytes = Buffer.from(image.data, "base64");

writeFileSync("output.png", imageBytes);

console.log(`Saved ${image.mime_type} image (${imageBytes.length} bytes)`);

curl https://api.arbytra.com/v1/chat/completions \
  -H "Authorization: Bearer $ARBYTRA_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gemini-2.5-flash-image",
    "messages": [{
      "role": "user",
      "content": "Draw a cartoon cat wearing a top hat"
    }]
  }' | jq -r '.choices[0].message.images[0].data' | base64 --decode > output.png