Spring AI学习笔记四十二:嵌入模型之Mistral AI

spring.ai.mistralai.api-key=<your-mistralai-api-key>

# In application.ymlspring:  ai:    mistralai:      api-key: ${MISTRALAI_API_KEY}

# In your environment or .env fileexport MISTRALAI_API_KEY=<your-mistralai-api-key>

// Retrieve API key from a secure source or environment variableString apiKey = System.getenv("MISTRALAI_API_KEY");

<dependency>    <groupId>org.springframework.ai</groupId>    <artifactId>spring-ai-starter-model-mistral-ai</artifactId></dependency>

dependencies {    implementation 'org.springframework.ai:spring-ai-starter-model-mistral-ai'}

// Using mistral-embed for general textEmbeddingResponse textEmbeddingResponse = embeddingModel.call(    new EmbeddingRequest(List.of("Hello World", "World is big and salvation is near"),        MistralAiEmbeddingOptions.builder()            .withModel("mistral-embed")        .build()));// Using codestral-embed for codeEmbeddingResponse codeEmbeddingResponse = embeddingModel.call(    new EmbeddingRequest(List.of("public class HelloWorld {}", "def hello_world():"),        MistralAiEmbeddingOptions.builder()            .withModel("codestral-embed")        .build()));

spring.ai.mistralai.api-key=YOUR_API_KEYspring.ai.mistralai.embedding.options.model=mistral-embed

@RestControllerpublic class EmbeddingController {    private final EmbeddingModel embeddingModel;    @Autowired    public EmbeddingController(EmbeddingModel embeddingModel) {        this.embeddingModel = embeddingModel;    }    @GetMapping("/ai/embedding")    public Map embed(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {        var embeddingResponse = this.embeddingModel.embedForResponse(List.of(message));        return Map.of("embedding", embeddingResponse);    }}

<dependency>    <groupId>org.springframework.ai</groupId>    <artifactId>spring-ai-mistral-ai</artifactId></dependency>

dependencies {    implementation 'org.springframework.ai:spring-ai-mistral-ai'}

var mistralAiApi = new MistralAiApi(System.getenv("MISTRAL_AI_API_KEY"));var embeddingModel = new MistralAiEmbeddingModel(this.mistralAiApi,        MistralAiEmbeddingOptions.builder()                .withModel("mistral-embed")                .withEncodingFormat("float")                .build());EmbeddingResponse embeddingResponse = this.embeddingModel        .embedForResponse(List.of("Hello World", "World is big and salvation is near"));