Skip to main content

MistralAI

This notebook explains how to use MistralAIEmbeddings, which is included in the langchain_mistralai package, to embed texts in langchain.

# pip install -U langchain-mistralai

import the library​

from langchain_mistralai import MistralAIEmbeddings
API Reference:MistralAIEmbeddings
embedding = MistralAIEmbeddings(api_key="your-api-key")

Using the Embedding Model

With MistralAIEmbeddings, you can directly use the default model 'mistral-embed', or set a different one if available.

embedding.model = "mistral-embed"  # or your preferred model if available
res_query = embedding.embed_query("The test information")
res_document = embedding.embed_documents(["test1", "another test"])

Was this page helpful?


You can leave detailed feedback on GitHub.