Package org.springframework.ai.mistralai.api

Class MistralAiApi

java.lang.Object

org.springframework.ai.mistralai.api.MistralAiApi

public class MistralAiApi
extends Object

Single-class, Java Client library for Mistral AI platform. Provides implementation for the Embeddings and the Chat Completion APIs.

Implements Synchronous and Streaming chat completion and supports latest Function Calling features.

Since:

1.0.0

Author:

Ricken Bazolo, Christian Tzolov, Thomas Vitale, Jason Smith

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	MistralAiApi.Builder
static final record	MistralAiApi.ChatCompletion	Represents a chat completion response returned by model, based on the provided input.
static final record	MistralAiApi.ChatCompletionChunk	Represents a streamed chunk of a chat completion response returned by model, based on the provided input.
static enum	MistralAiApi.ChatCompletionFinishReason	The reason the model stopped generating tokens.
static final record	MistralAiApi.ChatCompletionMessage	Message comprising the conversation.
static final record	MistralAiApi.ChatCompletionRequest	Creates a model request for chat conversation.
static enum	MistralAiApi.ChatModel	List of well-known Mistral chat models.
static final record	MistralAiApi.Embedding	Represents an embedding vector returned by embedding endpoint.
static final record	MistralAiApi.EmbeddingList<T>	List of multiple embedding responses.
static enum	MistralAiApi.EmbeddingModel	List of well-known Mistral embedding models.
static final record	MistralAiApi.EmbeddingRequest<T>	Creates an embedding vector representing the input text.
static class	MistralAiApi.FunctionTool	Represents a tool the model may call.
static final record	MistralAiApi.LogProbs	Log probability information for the choice.
static final record	MistralAiApi.Usage	Usage statistics.

Field Summary

Fields

Modifier and Type	Field	Description
static final String	PROVIDER_NAME

Constructor Summary

Constructors

Constructor	Description
MistralAiApi(String apiKey)	Deprecated.
MistralAiApi(String baseUrl, String apiKey)	Deprecated.
MistralAiApi(String baseUrl, String apiKey, org.springframework.web.client.RestClient.Builder restClientBuilder, org.springframework.web.client.ResponseErrorHandler responseErrorHandler)	Deprecated.
MistralAiApi(String baseUrl, String apiKey, org.springframework.web.client.RestClient.Builder restClientBuilder, org.springframework.web.reactive.function.client.WebClient.Builder webClientBuilder, org.springframework.web.client.ResponseErrorHandler responseErrorHandler)	Create a new client api.

Method Summary

Modifier and Type	Method	Description
static MistralAiApi.Builder	builder()
org.springframework.http.ResponseEntity<MistralAiApi.ChatCompletion>	chatCompletionEntity(MistralAiApi.ChatCompletionRequest chatRequest)	Creates a model response for the given chat conversation.
reactor.core.publisher.Flux<MistralAiApi.ChatCompletionChunk>	chatCompletionStream(MistralAiApi.ChatCompletionRequest chatRequest)	Creates a streaming chat response for the given chat conversation.
<T> org.springframework.http.ResponseEntity<MistralAiApi.EmbeddingList<MistralAiApi.Embedding>>	embeddings(MistralAiApi.EmbeddingRequest<T> embeddingRequest)	Creates an embedding vector representing the input text or token array.

Methods inherited from class java.lang.Object

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

Field Details

PROVIDER_NAME

public static final String PROVIDER_NAME

Constructor Details

MistralAiApi

@Deprecated
public MistralAiApi(String apiKey)

Deprecated.

Create a new client api with DEFAULT_BASE_URL

Parameters:

apiKey - Mistral api Key.

MistralAiApi

@Deprecated
public MistralAiApi(String baseUrl,
 String apiKey)

Deprecated.

Create a new client api.

Parameters:

baseUrl - api base URL.

apiKey - Mistral api Key.

MistralAiApi

@Deprecated
public MistralAiApi(String baseUrl,
 String apiKey,
 org.springframework.web.client.RestClient.Builder restClientBuilder,
 org.springframework.web.client.ResponseErrorHandler responseErrorHandler)

Deprecated.

Create a new client api.

Parameters:

baseUrl - api base URL.

apiKey - Mistral api Key.

restClientBuilder - RestClient builder.

responseErrorHandler - Response error handler.

MistralAiApi

public MistralAiApi(String baseUrl,
 String apiKey,
 org.springframework.web.client.RestClient.Builder restClientBuilder,
 org.springframework.web.reactive.function.client.WebClient.Builder webClientBuilder,
 org.springframework.web.client.ResponseErrorHandler responseErrorHandler)

Create a new client api.

Parameters:

baseUrl - api base URL.

apiKey - Mistral api Key.

restClientBuilder - RestClient builder.

responseErrorHandler - Response error handler.

Method Details

builder

public static MistralAiApi.Builder builder()

embeddings

public <T>
org.springframework.http.ResponseEntity<MistralAiApi.EmbeddingList<MistralAiApi.Embedding>> embeddings(MistralAiApi.EmbeddingRequest<T> embeddingRequest)

Creates an embedding vector representing the input text or token array.

Type Parameters:

T - Type of the entity in the data list. Can be a String or List of tokens (e.g. Integers). For embedding multiple inputs in a single request, You can pass a List of String or List of List of tokens. For example:

List.of("text1", "text2", "text3") or List.of(List.of(1, 2, 3), List.of(3, 4, 5)) 

Parameters:

embeddingRequest - The embedding request.

Returns:

Returns list of MistralAiApi.Embedding wrapped in MistralAiApi.EmbeddingList.

chatCompletionEntity

public org.springframework.http.ResponseEntity<MistralAiApi.ChatCompletion> chatCompletionEntity(MistralAiApi.ChatCompletionRequest chatRequest)

Creates a model response for the given chat conversation.

Parameters:

chatRequest - The chat completion request.

Returns:

Entity response with MistralAiApi.ChatCompletion as a body and HTTP status code and headers.

chatCompletionStream

public reactor.core.publisher.Flux<MistralAiApi.ChatCompletionChunk> chatCompletionStream(MistralAiApi.ChatCompletionRequest chatRequest)

Creates a streaming chat response for the given chat conversation.

Parameters:

chatRequest - The chat completion request. Must have the stream property set to true.

Returns:

Returns a Flux stream from chat completion chunks.