Add vector search test

Author: Strift

src/main/java/com/meilisearch/sdk/SearchRequest.java

@@ -44,7 +44,7 @@ public class SearchRequest {
     protected String[] locales;
     protected String distinct;
     protected Hybrid hybrid;
-
+    protected Double[] vector;
     /**
      * Constructor for SearchRequest for building search queries with the default values: offset: 0,
      * limit: 20, attributesToRetrieve: ["*"], attributesToCrop: null, cropLength: 200,
@@ -106,7 +106,8 @@ public String toString() {
                         .putOpt("showRankingScoreDetails", this.showRankingScoreDetails)
                         .putOpt("rankingScoreThreshold", this.rankingScoreThreshold)
                         .putOpt("locales", this.locales)
-                        .putOpt("distinct", this.distinct);
+                        .putOpt("distinct", this.distinct)
+                        .putOpt("vector", this.vector);
 
         // Add hybrid parameter if it exists
         if (this.hybrid != null) {

src/test/java/com/meilisearch/integration/SearchTest.java

@@ -1116,6 +1116,36 @@ public void testSimilarDocuments() throws Exception {
         assertThat(hits.get(3).get("title"), is("Shazam!"));
     }
 
+    /** Test vector search */
+    @Test
+    public void testVectorSearch() throws Exception {
+        String indexUid = "testVectorSearch";
+        Index index = client.index(indexUid);
+        HashMap<String, Embedder> embedders = new HashMap<>();
+        embedders.put(
+                "manual", new Embedder().setSource(EmbedderSource.USER_PROVIDED).setDimensions(3));
+
+        Settings settings = new Settings();
+        settings.setEmbedders(embedders);
+
+        index.updateSettings(settings);
+
+        TestData<Movie> testData = this.getTestData(VECTOR_MOVIES, Movie.class);
+        TaskInfo task = index.addDocuments(testData.getRaw());
+
+        index.waitForTask(task.getTaskUid());
+
+        SearchRequest searchRequest =
+                SearchRequest.builder().vector(new Double[] {0.1, 0.6, 0.8}).build();
+
+        SearchResult searchResult = (SearchResult) index.search(searchRequest);
+
+        assertThat(searchResult.getHits(), hasSize(5));
+        // The most similar document should be "Escape Room" since its vector [0.1, 0.6, 0.8]
+        assertThat(searchResult.getHits().get(0).get("id"), is("522681"));
+        assertThat(searchResult.getHits().get(0).get("title"), is("Escape Room"));
+    }
+
     /** Test Search with locales */
     @Test
     public void testSearchWithLocales() throws Exception {