Spring Boot AI Ollama Embedding 嵌入模型解析向量数据

@RestController
public class EmbeddingController {

    @Resource
    EmbeddingModel embeddingModel;

    @GetMapping("/ai/embedding")
    public Map embed(@RequestParam(value = "message",defaultValue = "告诉我一个笑话")String message) {
        EmbeddingResponse embeddingResponse = this.embeddingModel.embedForResponse(List.of(message));
        return Map.of("embedding", embeddingResponse);
    }
}

public class EmbedTests {
    public static void main(String[] args) {
        var ollamaApi = new OllamaApi("http://192.168.31.162:11434");
        OllamaEmbeddingModel embeddingModel = new OllamaEmbeddingModel(
                ollamaApi,
                OllamaOptions.builder()
                        .model(OllamaModel.MISTRAL.id())
                        .build(),
                ObservationRegistry.NOOP,
                ModelManagementOptions.defaults()
        );
        EmbeddingResponse helloWorld = embeddingModel.call(
                new EmbeddingRequest(
                        List.of("Hello World", "World is big and salvation is near"),
                        OllamaOptions.builder().model("chroma/all-minilm-l6-v2-f32")
                                .truncate(false)//truncate | 截断，删节；把……截成平面
                                .build()
                )
        );
        Map<String, EmbeddingResponse> embedding = Map.of("embedding", helloWorld);
        System.out.println(helloWorld);
    }
}