wrapImageModel

Wrap an image model with middleware.

The image-model counterpart to wrapLanguageModel. Commonly used to append house style to every prompt.

let images = wrapImageModel(
  model: OpenAIImageModel("gpt-image-2"),
  middleware: [ImageModelMiddleware(transformRequest: { request in
    var request = request
    request.prompt += ", studio lighting"
    return request
  })]
)

Signature

func wrapImageModel(
    model: any ImageModel,
    middleware: [ImageModelMiddleware]
) -> any ImageModel

Defined in Sources/AI/Core/ModelMiddleware.swift.

Parameters

ParameterTypeDefault
modelany ImageModelrequired
middleware[ImageModelMiddleware]required

Returns

An any ImageModel.

See also