Add async option to qdrant store

Author: natewiebe13

src/store/src/Bridge/Qdrant/Store.php

@@ -33,6 +33,7 @@ public function __construct(
         private readonly string $collectionName,
         private readonly int $embeddingsDimension = 1536,
         private readonly string $embeddingsDistance = 'Cosine',
+        private readonly bool $async = true,
     ) {
     }
 
@@ -60,7 +61,8 @@ public function add(VectorDocument ...$documents): void
     {
         $this->request('PUT', \sprintf('collections/%s/points', $this->collectionName), [
             'points' => array_map($this->convertToIndexableArray(...), $documents),
-        ]);
+        ],
+        !$this->async ? ['wait' => 'true'] : []);
     }
 
     /**
@@ -105,14 +107,15 @@ public function drop(): void
      *
      * @return array<string, mixed>
      */
-    private function request(string $method, string $endpoint, array $payload = []): array
+    private function request(string $method, string $endpoint, array $payload = [], array $queryParameters = []): array
     {
         $url = \sprintf('%s/%s', $this->endpointUrl, $endpoint);
 
         $response = $this->httpClient->request($method, $url, [
             'headers' => [
                 'api-key' => $this->apiKey,
             ],
+            'query' => $queryParameters,
             'json' => $payload,
         ]);
 

src/store/tests/Bridge/Qdrant/StoreTest.php

@@ -172,6 +172,39 @@ public function testStoreCanAdd()
         $this->assertSame(1, $httpClient->getRequestsCount());
     }
 
+    public function testStoreCanAddSynchronously()
+    {
+        $document = new VectorDocument(Uuid::v4(), new Vector([0.1, 0.2, 0.3]));
+
+        $httpClient = new MockHttpClient(static function (string $method, string $url, array $options) use ($document): JsonMockResponse {
+            self::assertArrayHasKey('wait', $options['query']);
+            self::assertEquals('true', $options['query']['wait']);
+
+            return new JsonMockResponse([
+                'time' => 0.002,
+                'status' => 'ok',
+                'result' => [
+                    'points' => [
+                        [
+                            'id' => (string) $document->id,
+                            'payload' => (array) $document->metadata,
+                            'vector' => $document->vector->getData(),
+                        ]
+                    ],
+                ],
+            ], [
+                'http_code' => 200,
+            ]);
+        }, 'http://127.0.0.1:6333');
+
+        $store = new Store($httpClient, 'http://127.0.0.1:6333', 'test', 'test', async: false);
+
+        $store->add($document);
+
+        $this->assertSame(1, $httpClient->getRequestsCount());
+        $this->assertSame(1, $httpClient->getRequestsCount());
+    }
+
     public function testStoreCannotQueryOnInvalidResponse()
     {
         $httpClient = new MockHttpClient([