Snippet for vector search

VectorSearchController.java

public record SimilaritySearchRequest(String question, Integer limit, Double maximumDistance) {

    public SimilaritySearchRequest {
        if (limit == null) {
            limit = 4; // Default value for limit is set to 4
        }
        if (maximumDistance == null) {
            maximumDistance = 1.0; // Default value for threshold is set to 1.0
        }
    }
}

@PostMapping("/api/vectordb/documents/search")
    public List<Document> handleSimilarSearchQuery(@RequestBody SimilaritySearchRequest similaritySearchRequest) {


        Double threshold = 1- similaritySearchRequest.maximumDistance();

        SearchRequest searchRequest = SearchRequest
                                        .query(similaritySearchRequest.question())
                                        .withTopK(similaritySearchRequest.limit())
                                        .withSimilarityThreshold(threshold);
        List<Document> documents = vectorStore.similaritySearch(searchRequest);
        return documents;
    }