wrapLanguageModel()

The wrapLanguageModel function provides a way to enhance the behavior of language models by wrapping them with middleware. See Language Model Middleware for more information on middleware.

import { wrapLanguageModel, gateway } from 'ai';
const wrappedLanguageModel = wrapLanguageModel({
model: gateway('openai/gpt-4.1'),
middleware: yourLanguageModelMiddleware,
});

Import

import { wrapLanguageModel } from "ai"

API Signature

Parameters

model:

LanguageModelV2

middleware:

LanguageModelV2Middleware | LanguageModelV2Middleware[]

modelId:

string

providerId:

string

Returns

A new LanguageModelV2 instance with middleware applied.