feat: add missing params in SearchQuery

Author: bpolaszek

src/Contracts/SearchQuery.php

@@ -111,6 +111,13 @@ class SearchQuery
 
     private ?FederationOptions $federationOptions = null;
 
+    private ?bool $retrieveVectors = null;
+
+    /**
+     * @var array<string, mixed>|null
+     */
+    private ?array $media = null;
+
     /**
      * @return $this
      */
@@ -429,6 +436,36 @@ public function setAttributesToSearchOn(array $attributesToSearchOn): self
         return $this;
     }
 
+    public function getRetrieveVectors(): ?bool
+    {
+        return $this->retrieveVectors;
+    }
+
+    public function setRetrieveVectors(?bool $retrieveVectors): self
+    {
+        $this->retrieveVectors = $retrieveVectors;
+
+        return $this;
+    }
+
+    /**
+     * @return array<string, mixed>|null
+     */
+    public function getMedia(): ?array
+    {
+        return $this->media;
+    }
+
+    /**
+     * @param array<string, mixed>|null $media
+     */
+    public function setMedia(?array $media): self
+    {
+        $this->media = $media;
+
+        return $this;
+    }
+
     /**
      * @return array{
      *     indexUid?: non-empty-string,
@@ -457,7 +494,9 @@ public function setAttributesToSearchOn(array $attributesToSearchOn): self
      *     showRankingScoreDetails?: bool,
      *     rankingScoreThreshold?: float,
      *     distinct?: non-empty-string,
-     *     federationOptions?: array<mixed>
+     *     federationOptions?: array<mixed>,
+     *     retrieveVectors?: bool,
+     *     media?: array<string, mixed>,
      * }
      */
     public function toArray(): array
@@ -490,6 +529,8 @@ public function toArray(): array
             'rankingScoreThreshold' => $this->rankingScoreThreshold,
             'distinct' => $this->distinct,
             'federationOptions' => null !== $this->federationOptions ? $this->federationOptions->toArray() : null,
+            'retrieveVectors' => $this->retrieveVectors,
+            'media' => $this->media,
         ], static function ($item) { return null !== $item; });
     }
 }

tests/Contracts/SearchQueryTest.php

@@ -236,4 +236,23 @@ public function testSetFederationOptions(): void
 
         self::assertSame(['federationOptions' => ['weight' => 0.5]], $data->toArray());
     }
+
+    /**
+     * @testWith [true]
+     *           [false]
+     */
+    public function testSetRetrieveVectors(bool $retrieveVectors): void
+    {
+        $data = (new SearchQuery())->setRetrieveVectors($retrieveVectors);
+
+        self::assertSame(['retrieveVectors' => $retrieveVectors], $data->toArray());
+    }
+
+    public function testSetMedia(): void
+    {
+        $media = ['image' => ['mime' => 'image/jpeg', 'data' => 'data://foo:bar']];
+        $data = (new SearchQuery())->setMedia($media);
+
+        self::assertSame(['media' => $media], $data->toArray());
+    }
 }