Merge #683

683: Changes related to the next Meilisearch release (v1.11.0) r=brunoocasali a=meili-bot Related to this issue: meilisearch/integration-guides#303 This PR: - gathers the changes related to the next Meilisearch release (v1.11.0) so that this package is ready when the official release is out. - should pass the tests against the [latest pre-release of Meilisearch](https://github.com/meilisearch/meilisearch/releases). - might eventually contain test failures until the Meilisearch v1.11.0 is out. ⚠️ This PR should NOT be merged until the next release of Meilisearch (v1.11.0) is out. _This PR is auto-generated for the [pre-release week](https://github.com/meilisearch/integration-guides/blob/main/resources/pre-release-week.md) purpose._ Co-authored-by: meili-bot <74670311+meili-bot@users.noreply.github.com> Co-authored-by: Clémentine <clementine@meilisearch.com> Co-authored-by: ManyTheFish <many@meilisearch.com> Co-authored-by: Many the fish <many@meilisearch.com> Co-authored-by: curquiza <clementine@meilisearch.com>
meilisearch · Oct 28, 2024 · c575143 · c575143
2 parents 0d4c933 + 39d3cbd
commit c575143
Show file tree

Hide file tree

Showing 9 changed files with 160 additions and 59 deletions.
diff --git a/.code-samples.meilisearch.yaml b/.code-samples.meilisearch.yaml
@@ -722,7 +722,14 @@ search_parameter_reference_retrieve_vectors_1: |-
   $client->index('INDEX_NAME')->search('kitchen utensils', [
     'retrieveVectors' => true,
     'hybrid' => [
-      'embedder': 'default'
+      'embedder': 'EMBEDDER_NAME'
+    ]
+  ]);
+search_parameter_guide_hybrid_1: |-
+  $client->index('INDEX_NAME')->search('kitchen utensils', [
+    'hybrid' => [
+      'semanticRatio' => 0.9,
+      'embedder' => 'EMBEDDER_NAME'
     ]
   ]);
 search_parameter_reference_ranking_score_threshold_1: |-
@@ -742,7 +749,7 @@ distinct_attribute_guide_distinct_parameter_1: |-
 search_parameter_guide_matching_strategy_3: |-
   $client->index('movies')->search('white shirt', ['matchingStrategy' => 'frequency']);
 get_similar_post_1: |-
-  $similarQuery = new SimilarDocumentsQuery('TARGET_DOCUMENT_ID');
+  $similarQuery = new SimilarDocumentsQuery('TARGET_DOCUMENT_ID', 'default');
   $client->index('INDEX_NAME')->searchSimilarDocuments($similarQuery);
 multi_search_federated_1: |-
   $client->multiSearch([
@@ -756,7 +763,7 @@ multi_search_federated_1: |-
     (new MultiSearchFederation())
   );
 search_parameter_reference_locales_1: |-
-  $client->index('INDEX_NAME')->search('進撃の巨人', [
+  $client->index('INDEX_NAME')->search('QUERY TEXT IN JAPANESE', [
     'locales' => ['jpn']
   ]);
 get_localized_attribute_settings_1: |-

diff --git a/src/Contracts/MultiSearchFederation.php b/src/Contracts/MultiSearchFederation.php
@@ -16,6 +16,16 @@ class MultiSearchFederation
      */
     private ?int $offset = null;
 
+    /**
+     * @var array<non-empty-string, list<non-empty-string>>|null
+     */
+    private ?array $facetsByIndex = null;
+
+    /**
+     * @var array{maxValuesPerFacet: positive-int}|null
+     */
+    private ?array $mergeFacets = null;
+
     /**
      * @param non-negative-int $limit
      *
@@ -40,17 +50,45 @@ public function setOffset(int $offset): self
         return $this;
     }
 
+    /**
+     * @param array<non-empty-string, list<non-empty-string>> $facetsByIndex
+     *
+     * @return $this
+     */
+    public function setFacetsByIndex(array $facetsByIndex): self
+    {
+        $this->facetsByIndex = $facetsByIndex;
+
+        return $this;
+    }
+
+    /**
+     * @param array{maxValuesPerFacet: positive-int} $mergeFacets
+     *
+     * @return $this
+     */
+    public function setMergeFacets(array $mergeFacets): self
+    {
+        $this->mergeFacets = $mergeFacets;
+
+        return $this;
+    }
+
     /**
      * @return array{
      *     limit?: non-negative-int,
-     *     offset?: non-negative-int
+     *     offset?: non-negative-int,
+     *     facetsByIndex?: array<non-empty-string, list<non-empty-string>>,
+     *     mergeFacets?: array{maxValuesPerFacet: positive-int},
      * }
      */
     public function toArray(): array
     {
         return array_filter([
             'limit' => $this->limit,
             'offset' => $this->offset,
+            'facetsByIndex' => $this->facetsByIndex,
+            'mergeFacets' => $this->mergeFacets,
         ], static function ($item) { return null !== $item; });
     }
 }
diff --git a/src/Contracts/SimilarDocumentsQuery.php b/src/Contracts/SimilarDocumentsQuery.php
@@ -11,6 +11,11 @@ class SimilarDocumentsQuery
      */
     private $id;
 
+    /**
+     * @var non-empty-string
+     */
+    private string $embedder;
+
     /**
      * @var non-negative-int|null
      */
@@ -21,11 +26,6 @@ class SimilarDocumentsQuery
      */
     private ?int $limit = null;
 
-    /**
-     * @var non-empty-string|null
-     */
-    private ?string $embedder = null;
-
     /**
      * @var list<non-empty-string>|null
      */
@@ -48,11 +48,13 @@ class SimilarDocumentsQuery
     private $rankingScoreThreshold;
 
     /**
-     * @param int|string $id
+     * @param int|string       $id
+     * @param non-empty-string $embedder
      */
-    public function __construct($id)
+    public function __construct($id, string $embedder)
     {
         $this->id = $id;
+        $this->embedder = $embedder;
     }
 
     /**
@@ -91,18 +93,6 @@ public function setFilter(array $filter): self
         return $this;
     }
 
-    /**
-     * @param non-empty-string $embedder
-     *
-     * @return $this
-     */
-    public function setEmbedder(string $embedder): self
-    {
-        $this->embedder = $embedder;
-
-        return $this;
-    }
-
     /**
      * @param list<non-empty-string> $attributesToRetrieve an array of attribute names to retrieve
      *
@@ -166,10 +156,10 @@ public function setRankingScoreThreshold($rankingScoreThreshold): self
     /**
      * @return array{
      *     id: int|string,
+     *     embedder: non-empty-string,
      *     offset?: non-negative-int,
      *     limit?: positive-int,
      *     filter?: array<int, array<int, string>|string>,
-     *     embedder?: non-empty-string,
      *     attributesToRetrieve?: list<non-empty-string>,
      *     showRankingScore?: bool,
      *     showRankingScoreDetails?: bool,
@@ -181,10 +171,10 @@ public function toArray(): array
     {
         return array_filter([
             'id' => $this->id,
+            'embedder' => $this->embedder,
             'offset' => $this->offset,
             'limit' => $this->limit,
             'filter' => $this->filter,
-            'embedder' => $this->embedder,
             'attributesToRetrieve' => $this->attributesToRetrieve,
             'showRankingScore' => $this->showRankingScore,
             'showRankingScoreDetails' => $this->showRankingScoreDetails,

diff --git a/tests/Contracts/MultiSearchFederationTest.php b/tests/Contracts/MultiSearchFederationTest.php
@@ -29,4 +29,18 @@ public function testSetOffset(): void
 
         self::assertSame(['offset' => 5], $data->toArray());
     }
+
+    public function testSetFacetsByIndex(): void
+    {
+        $data = (new MultiSearchFederation())->setFacetsByIndex(['books' => ['author', 'genre']]);
+
+        self::assertSame(['facetsByIndex' => ['books' => ['author', 'genre']]], $data->toArray());
+    }
+
+    public function testSetMergeFacets(): void
+    {
+        $data = (new MultiSearchFederation())->setMergeFacets(['maxValuesPerFacet' => 10]);
+
+        self::assertSame(['mergeFacets' => ['maxValuesPerFacet' => 10]], $data->toArray());
+    }
 }
diff --git a/tests/Contracts/SimilarDocumentsQueryTest.php b/tests/Contracts/SimilarDocumentsQueryTest.php
@@ -10,54 +10,48 @@
 final class SimilarDocumentsQueryTest extends TestCase
 {
     /**
-     * @param int|string $id
+     * @param int|string       $id
+     * @param non-empty-string $embedder
      *
-     * @testWith [123]
-     *           ["test"]
+     * @testWith [123, "default"]
+     *           ["test", "manual"]
      */
-    public function testConstruct($id): void
+    public function testConstruct($id, string $embedder): void
     {
-        $data = new SimilarDocumentsQuery($id);
+        $data = new SimilarDocumentsQuery($id, $embedder);
 
-        self::assertSame(['id' => $id], $data->toArray());
+        self::assertSame(['id' => $id, 'embedder' => $embedder], $data->toArray());
     }
 
     public function testSetOffset(): void
     {
-        $data = (new SimilarDocumentsQuery('test'))->setOffset(66);
+        $data = (new SimilarDocumentsQuery('test', 'default'))->setOffset(66);
 
-        self::assertSame(['id' => 'test', 'offset' => 66], $data->toArray());
+        self::assertSame(['id' => 'test', 'embedder' => 'default', 'offset' => 66], $data->toArray());
     }
 
     public function testSetLimit(): void
     {
-        $data = (new SimilarDocumentsQuery('test'))->setLimit(50);
+        $data = (new SimilarDocumentsQuery('test', 'default'))->setLimit(50);
 
-        self::assertSame(['id' => 'test', 'limit' => 50], $data->toArray());
+        self::assertSame(['id' => 'test', 'embedder' => 'default', 'limit' => 50], $data->toArray());
     }
 
     public function testSetFilter(): void
     {
-        $data = (new SimilarDocumentsQuery('test'))->setFilter([
+        $data = (new SimilarDocumentsQuery('test', 'default'))->setFilter([
             ['genres = horror', 'genres = mystery'],
             "director = 'Jordan Peele'",
         ]);
 
-        self::assertSame(['id' => 'test', 'filter' => [['genres = horror', 'genres = mystery'], "director = 'Jordan Peele'"]], $data->toArray());
-    }
-
-    public function testSetEmbedder(): void
-    {
-        $data = (new SimilarDocumentsQuery('test'))->setEmbedder('default');
-
-        self::assertSame(['id' => 'test', 'embedder' => 'default'], $data->toArray());
+        self::assertSame(['id' => 'test', 'embedder' => 'default', 'filter' => [['genres = horror', 'genres = mystery'], "director = 'Jordan Peele'"]], $data->toArray());
     }
 
     public function testSetAttributesToRetrieve(): void
     {
-        $data = (new SimilarDocumentsQuery('test'))->setAttributesToRetrieve(['name', 'price']);
+        $data = (new SimilarDocumentsQuery('test', 'default'))->setAttributesToRetrieve(['name', 'price']);
 
-        self::assertSame(['id' => 'test', 'attributesToRetrieve' => ['name', 'price']], $data->toArray());
+        self::assertSame(['id' => 'test', 'embedder' => 'default', 'attributesToRetrieve' => ['name', 'price']], $data->toArray());
     }
 
     /**
@@ -66,9 +60,9 @@ public function testSetAttributesToRetrieve(): void
      */
     public function testSetShowRankingScore(bool $showRankingScore): void
     {
-        $data = (new SimilarDocumentsQuery('test'))->setShowRankingScore($showRankingScore);
+        $data = (new SimilarDocumentsQuery('test', 'default'))->setShowRankingScore($showRankingScore);
 
-        self::assertSame(['id' => 'test', 'showRankingScore' => $showRankingScore], $data->toArray());
+        self::assertSame(['id' => 'test', 'embedder' => 'default', 'showRankingScore' => $showRankingScore], $data->toArray());
     }
 
     /**
@@ -77,9 +71,9 @@ public function testSetShowRankingScore(bool $showRankingScore): void
      */
     public function testSetShowRankingScoreDetails(bool $showRankingScoreDetails): void
     {
-        $data = (new SimilarDocumentsQuery('test'))->setShowRankingScoreDetails($showRankingScoreDetails);
+        $data = (new SimilarDocumentsQuery('test', 'default'))->setShowRankingScoreDetails($showRankingScoreDetails);
 
-        self::assertSame(['id' => 'test', 'showRankingScoreDetails' => $showRankingScoreDetails], $data->toArray());
+        self::assertSame(['id' => 'test', 'embedder' => 'default', 'showRankingScoreDetails' => $showRankingScoreDetails], $data->toArray());
     }
 
     /**
@@ -88,9 +82,9 @@ public function testSetShowRankingScoreDetails(bool $showRankingScoreDetails): v
      */
     public function testSetRetrieveVectors(bool $retrieveVectors): void
     {
-        $data = (new SimilarDocumentsQuery('test'))->setRetrieveVectors($retrieveVectors);
+        $data = (new SimilarDocumentsQuery('test', 'default'))->setRetrieveVectors($retrieveVectors);
 
-        self::assertSame(['id' => 'test', 'retrieveVectors' => $retrieveVectors], $data->toArray());
+        self::assertSame(['id' => 'test', 'embedder' => 'default', 'retrieveVectors' => $retrieveVectors], $data->toArray());
     }
 
     /**
@@ -101,8 +95,8 @@ public function testSetRetrieveVectors(bool $retrieveVectors): void
      */
     public function testSetRankingScoreThreshold($rankingScoreThreshold): void
     {
-        $data = (new SimilarDocumentsQuery('test'))->setRankingScoreThreshold($rankingScoreThreshold);
+        $data = (new SimilarDocumentsQuery('test', 'default'))->setRankingScoreThreshold($rankingScoreThreshold);
 
-        self::assertSame(['id' => 'test', 'rankingScoreThreshold' => $rankingScoreThreshold], $data->toArray());
+        self::assertSame(['id' => 'test', 'embedder' => 'default', 'rankingScoreThreshold' => $rankingScoreThreshold], $data->toArray());
     }
 }
diff --git a/tests/Endpoints/MultiSearchTest.php b/tests/Endpoints/MultiSearchTest.php
@@ -89,14 +89,15 @@ public function testFederation(): void
                 // By setting the weight to 0.9 this query should appear second
                 ->setFederationOptions((new FederationOptions())->setWeight(0.9)),
         ],
-            (new MultiSearchFederation())->setLimit(2)
+            (new MultiSearchFederation())->setLimit(2)->setFacetsByIndex([$this->booksIndex->getUid() => ['genre'], $this->songsIndex->getUid() => ['duration-float']])->setMergeFacets(['maxValuesPerFacet' => 10])
         );
 
         self::assertArrayHasKey('hits', $response);
         self::assertArrayHasKey('processingTimeMs', $response);
         self::assertArrayHasKey('limit', $response);
         self::assertArrayHasKey('offset', $response);
         self::assertArrayHasKey('estimatedTotalHits', $response);
+        self::assertArrayHasKey('facetDistribution', $response);
         self::assertCount(2, $response['hits']);
         self::assertSame(2, $response['limit']);
         self::assertSame(0, $response['offset']);

diff --git a/tests/Endpoints/SearchTest.php b/tests/Endpoints/SearchTest.php
@@ -701,12 +701,12 @@ public function testVectorSearch(): void
         $this->assertIsValidPromise($promise);
         $index->waitForTask($promise['taskUid']);
 
-        $response = $index->search('', ['vector' => [-0.5, 0.3, 0.85], 'hybrid' => ['semanticRatio' => 1.0]]);
+        $response = $index->search('', ['vector' => [-0.5, 0.3, 0.85], 'hybrid' => ['semanticRatio' => 1.0, 'embedder' => 'manual']]);
 
         self::assertSame(5, $response->getSemanticHitCount());
         self::assertArrayNotHasKey('_vectors', $response->getHit(0));
 
-        $response = $index->search('', ['vector' => [-0.5, 0.3, 0.85], 'hybrid' => ['semanticRatio' => 1.0], 'retrieveVectors' => true]);
+        $response = $index->search('', ['vector' => [-0.5, 0.3, 0.85], 'hybrid' => ['semanticRatio' => 1.0, 'embedder' => 'manual'], 'retrieveVectors' => true]);
 
         self::assertSame(5, $response->getSemanticHitCount());
         self::assertArrayHasKey('_vectors', $response->getHit(0));

diff --git a/tests/Endpoints/SimilarDocumentsTest.php b/tests/Endpoints/SimilarDocumentsTest.php
@@ -30,14 +30,14 @@ public function testBasicSearchWithSimilarDocuments(): void
         self::assertSame(1, $response->getHitsCount());
 
         $documentId = $response->getHit(0)['id'];
-        $response = $this->index->searchSimilarDocuments(new SimilarDocumentsQuery($documentId));
+        $response = $this->index->searchSimilarDocuments(new SimilarDocumentsQuery($documentId, 'manual'));
 
         self::assertGreaterThanOrEqual(4, $response->getHitsCount());
         self::assertArrayNotHasKey('_vectors', $response->getHit(0));
         self::assertArrayHasKey('id', $response->getHit(0));
         self::assertSame($documentId, $response->getId());
 
-        $similarQuery = new SimilarDocumentsQuery($documentId);
+        $similarQuery = new SimilarDocumentsQuery($documentId, 'manual');
         $response = $this->index->searchSimilarDocuments($similarQuery->setRetrieveVectors(true));
         self::assertGreaterThanOrEqual(4, $response->getHitsCount());
         self::assertArrayHasKey('_vectors', $response->getHit(0));