MistralAI

本笔记本解释了如何使用包含在 langchain_mistralai 包中的 MistralAIEmbeddings 来嵌入 langchain 中的文本。

# pip install -U langchain-mistralai

导入库

from langchain_mistralai import MistralAIEmbeddings

embedding = MistralAIEmbeddings(api_key="your-api-key")

使用嵌入模型

使用 MistralAIEmbeddings，您可以直接使用默认模型 'mistral-embed'，或者如果有可用的模型，可以设置为其他模型。

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"])

相关

• 嵌入模型 概念指南
• 嵌入模型 操作指南