wrapEmbeddingModel
Wrap an embedding model with middleware.
The embedding-model counterpart to
wrapLanguageModel, using
EmbeddingModelMiddleware.
let embeddings = wrapEmbeddingModel(
model: OpenAIEmbeddingModel("text-embedding-3-small"),
middleware: [.defaultSettings(maxBatchSize: 96)]
)Signature
func wrapEmbeddingModel(
model: any EmbeddingModel,
middleware: [EmbeddingModelMiddleware]
) -> any EmbeddingModelDefined in Sources/AI/Core/ModelMiddleware.swift.
Parameters
| Parameter | Type | Default |
|---|---|---|
model | any EmbeddingModel | required |
middleware | [EmbeddingModelMiddleware] | required |
Returns
An any EmbeddingModel.