MistralAIEmbeddingFunction

MistralAIEmbeddingFunction 是 pymilvus 中的一个类，用于使用 Mistral AI 嵌入模型将文本编码为嵌入，以支持在 Milvus 中进行嵌入检索。

python

pymilvus.model.dense.MistralAIEmbeddingFunction

构造函数

构造一个适用于常见用例的 MistralAIEmbeddingFunction。

python

MistralAIEmbeddingFunction(
    api_key: str,
    model_name: str = "mistral-embed",
    **kwargs
)

参数：

• api_key (string)

  用于访问 Mistral AI API 的 API 密钥。

• model_name (string)

  用于编码的 Mistral AI 嵌入模型名称。该值默认为 mistral-embed。更多信息，请参见 Embeddings。

• kwargs

  允许将其他关键字参数传递给模型初始化。更多信息，请参见 Embedding API。

示例

python

from pymilvus.model.dense import MistralAIEmbeddingFunction

ef = MistralAIEmbeddingFunction(
    model_name="mistral-embed", # Defaults to `mistral-embed`
    api_key="MISTRAL_API_KEY" # Provide your Mistral AI API key
)