wrapImageModel()

The wrapImageModel function provides a way to enhance the behavior of image models by wrapping them with middleware.

import { generateImage, wrapImageModel } from 'ai';
import { openai } from '@ai-sdk/openai';


const model = wrapImageModel({
  model: openai.image('gpt-image-2'),
  middleware: yourImageModelMiddleware,
});


const { image } = await generateImage({
  model,
  prompt: 'Santa Claus driving a Cadillac',
});

Import 

import { wrapImageModel } from "ai"

API Signature

Parameters

model:

ImageModelV4

middleware:

ImageModelV4Middleware | ImageModelV4Middleware[]

modelId:

string

providerId:

string

Returns

A new ImageModelV4 instance with middleware applied.