Image Generation

import OpenAI from "openai"

const client = new OpenAI({
  apiKey: process.env.XAI_API_KEY,
  baseURL: "https://api.x.ai/v1",
})

const response = await client.images.generate({
  model: "grok-2-image",
  prompt: "A futuristic cityscape at sunset with flying cars",
  n: 1,
  size: "1024x1024",
})

console.log(response.data[0].url)

// URL response (default)
const urlResponse = await client.images.generate({
  model: "grok-2-image",
  prompt: "A red apple on a white table",
  response_format: "url",
})

// Base64 response
const b64Response = await client.images.generate({
  model: "grok-2-image",
  prompt: "A red apple on a white table",
  response_format: "b64_json",
})

// Save base64 to file
import { writeFileSync } from "fs"
const imageData = b64Response.data[0].b64_json
writeFileSync("apple.png", Buffer.from(imageData!, "base64"))

const response = await client.images.generate({
  model: "grok-2-image",
  prompt: "A cute robot mascot",
  n: 4, // Generate 4 variations
})

for (const [index, image] of response.data.entries()) {
  console.log(`Image ${index + 1}: ${image.url}`)
}

[Subject] [Action/Pose] [Setting/Background] [Style] [Lighting] [Details]

const prompt = `
A majestic owl perched on an ancient oak branch,
forest background with morning mist,
digital art style, volumetric lighting,
highly detailed feathers, 8K resolution
`.trim()

try {
  const response = await client.images.generate({
    model: "grok-2-image",
    prompt: userPrompt,
  })
} catch (error) {
  if (error instanceof OpenAI.APIError) {
    if (error.status === 400) {
      // Content policy violation or invalid prompt
      console.error("Invalid prompt:", error.message)
    } else if (error.status === 429) {
      // Rate limited
      console.error("Rate limited, retry after delay")
    }
  }
  throw error
}