Package dev.langchain4j.model.mistralai

Class MistralAiEmbeddingModel

java.lang.Object

dev.langchain4j.model.embedding.DimensionAwareEmbeddingModel

dev.langchain4j.model.mistralai.MistralAiEmbeddingModel

All Implemented Interfaces:

EmbeddingModel

public class MistralAiEmbeddingModel
extends DimensionAwareEmbeddingModel

Represents a Mistral AI embedding model, such as mistral-embed. You can find description of parameters here.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	MistralAiEmbeddingModel.MistralAiEmbeddingModelBuilder

Field Summary

Fields inherited from class dev.langchain4j.model.embedding.DimensionAwareEmbeddingModel

dimension

Constructor Summary

Constructors

Constructor	Description
MistralAiEmbeddingModel(String baseUrl, String apiKey, String modelName, Duration timeout, Boolean logRequests, Boolean logResponses, Integer maxRetries)	Constructs a new MistralAiEmbeddingModel instance.

Method Summary

Modifier and Type	Method	Description
static MistralAiEmbeddingModel.MistralAiEmbeddingModelBuilder	builder()
Response<List<Embedding>>	embedAll(List<TextSegment> textSegments)	Embeds a list of text segments using the Mistral AI embedding model.

Methods inherited from class dev.langchain4j.model.embedding.DimensionAwareEmbeddingModel

dimension, knownDimension

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface dev.langchain4j.model.embedding.EmbeddingModel

embed, embed

Constructor Details

MistralAiEmbeddingModel

public MistralAiEmbeddingModel(String baseUrl,
 String apiKey,
 String modelName,
 Duration timeout,
 Boolean logRequests,
 Boolean logResponses,
 Integer maxRetries)

Constructs a new MistralAiEmbeddingModel instance.

Parameters:

baseUrl - the base URL of the Mistral AI API. It use a default value if not specified

apiKey - the API key for authentication

modelName - the name of the embedding model. It uses a default value if not specified

timeout - the timeout duration for API requests. It uses a default value of 60 seconds if not specified

The default value is 60 seconds

logRequests - a flag indicating whether to log API requests

logResponses - a flag indicating whether to log API responses

maxRetries - the maximum number of retries for API requests. It uses a default value of 3 if not specified

Method Details

embedAll

public Response<List<Embedding>> embedAll(List<TextSegment> textSegments)

Embeds a list of text segments using the Mistral AI embedding model.

Parameters:

textSegments - the list of text segments to embed

Returns:

a Response object containing the embeddings and token usage information

builder

public static MistralAiEmbeddingModel.MistralAiEmbeddingModelBuilder builder()