Image Generation

Portkey's AI gateway supports image generation capabilities that many foundational model providers offer. The most common use case is that of text-to-image where the user sends a prompt which the image model processes and returns an image.

The guide for vision models is available here.

Text-to-Image Usage

Portkey supports the OpenAI signature to make text-to-image requests.

import Portkey from 'portkey-ai';

// Initialize the Portkey client
const portkey = new Portkey({
    apiKey: "PORTKEY_API_KEY",  // Replace with your Portkey API key
    virtualKey: "VIRTUAL_KEY"   // Add your provider's virtual key

async function main() {
  const image = await portkey.images.generate({ 
    model: "dall-e-3", 
    prompt: "Lucy in the sky with diamonds" 


API Reference

Create Image

On completion, the request will get logged in the logs UI where the image can be viewed.

(Note that providers may remove the hosted image after a period of time, so some logs might only contain the url)

Supported Providers and Models

The following providers are supported for image generation with more providers getting added soon. Please raise a request or a PR to add model or provider to the AI gateway.


dall-e-2, dall-e-3

Create Image (text to image)

dall-e-2, dall-e-3

Create Image (text to image)

stable-diffusion-v1-6, stable-diffusion-xl-1024-v1-0

Create Image (text to image)

Replicate (Coming Soon)

Monster API (Coming Soon)

Segmind (Coming Soon)

Together AI (Coming Soon)


Here's a detailed cookbook on image generation using Portkey which demonstrates the use of multiple providers and routing between them through Configs.

Last updated