Using Grok Imagine Image 2.0 on Vercel AI Gateway

Implementation guide · Grok Image · xAI

Serverless

Vercel AI Gateway exposes Grok Imagine Image 2.0 through model ID spacexai/grok-imagine-image-2.0. Use the setup steps, capabilities, and official provider links below to validate this route before deployment.

Last refreshed 2026-10-19. Next refresh: weekly.

Quick Start

  1. 1
    Create an account at Vercel AI Gateway and generate an API key.
  2. 2
    Use the Vercel AI Gateway SDK or REST API to call spacexai/grok-imagine-image-2.0 — see the documentation for request format.

Code Examples

Install
pip install openai
API key
AI_GATEWAY_API_KEY
Model ID
spacexai/grok-imagine-image-2.0

creator/model-name e.g. kwaipilot/kat-coder-pro-v2

import os
from openai import OpenAI

client = OpenAI(
    api_key=os.environ["AI_GATEWAY_API_KEY"],
    base_url="https://ai-gateway.vercel.sh/v1"
)
response = client.chat.completions.create(
    model="spacexai/grok-imagine-image-2.0",
    messages=[{"role": "user", "content": "Hello"}]
)
print(response.choices[0].message.content)

Capabilities

VisionMultimodal

About Grok Imagine Image 2.0

Grok Imagine Image 2.0 (API id grok-imagine-image-2.0) is xAI's Quality Mode image generation/editing model on grok.com/imagine and the xAI API. $0.04/image output, $0.01/image input.