Skip to main content
The MixedbreadAIEmbeddings class uses the Mixedbread AI API to generate text embeddings. This guide will walk you through setting up and using the MixedbreadAIEmbeddings class, helping you integrate it into your project effectively.

Installation

To install the @langchain/mixedbread-ai package, use the following command:
npm

Initialization

First, sign up on the Mixedbread AI website and get your API key from Mixedbread AI. You can then use this key to initialize the MixedbreadAIEmbeddings class. You can pass the API key directly to the constructor or set it as an environment variable (MXBAI_API_KEY).

Basic usage

Here’s how to create an instance of MixedbreadAIEmbeddings:
If the apiKey is not provided, it will be read from the MXBAI_API_KEY environment variable.

Generating embeddings

Embedding a single query

To generate embeddings for a single text query, use the embedQuery method:

Embedding multiple documents

To generate embeddings for multiple documents, use the embedDocuments method. This method handles batching automatically based on the batchSize parameter:

Customizing requests

You can customize the SDK by passing additional parameters.

Error handling

If the API key is not provided and cannot be found in the environment variables, an error will be thrown: