createCollection request config sets

Author: hkulekci

src/Models/Request/CreateCollection.php

@@ -12,6 +12,7 @@
 use Qdrant\Models\Request\CollectionConfig\HnswConfig;
 use Qdrant\Models\Request\CollectionConfig\OptimizersConfigDiff;
 use Qdrant\Models\Request\CollectionConfig\QuantizationConfig;
+use Qdrant\Models\Request\CollectionConfig\WalConfig;
 
 class CreateCollection implements RequestModel
 {
@@ -32,6 +33,8 @@ class CreateCollection implements RequestModel
 
     protected ?HnswConfig $hnswConfig = null;
 
+    protected ?WalConfig $walConfig = null;
+
     protected ?QuantizationConfig $quantizationConfig = null;
 
     public function addVector(VectorParams $vectorParams, string $name = null): CreateCollection
@@ -87,6 +90,20 @@ public function setOptimizersConfigDiff(OptimizersConfigDiff $optimizersConfig):
         return $this;
     }
 
+    public function setHnswConfig(HnswConfig $hnswConfig): CreateCollection
+    {
+        $this->hnswConfig = $hnswConfig;
+
+        return $this;
+    }
+
+    public function setWalConfig(WalConfig $walConfig): CreateCollection
+    {
+        $this->walConfig = $walConfig;
+
+        return $this;
+    }
+
     public function setQuantizationConfig(QuantizationConfig $quantizationConfig): CreateCollection
     {
         $this->quantizationConfig = $quantizationConfig;
@@ -121,6 +138,9 @@ public function toArray(): array
         if ($this->hnswConfig !== null) {
             $data['hnsw_config'] = $this->hnswConfig->toArray();
         }
+        if ($this->walConfig !== null) {
+            $data['wal_config'] = $this->walConfig->toArray();
+        }
 
         if ($this->quantizationConfig instanceof DisabledQuantization) {
             $data['quantization_config'] = 'Disabled';

tests/Unit/Models/Request/CreateCollectionTest.php

@@ -6,6 +6,9 @@
 namespace Qdrant\Tests\Unit\Models\Request;
 
 use PHPUnit\Framework\TestCase;
+use Qdrant\Models\Request\CollectionConfig\HnswConfig;
+use Qdrant\Models\Request\CollectionConfig\OptimizersConfigDiff;
+use Qdrant\Models\Request\CollectionConfig\WalConfig;
 use Qdrant\Models\Request\CreateCollection;
 use Qdrant\Models\Request\VectorParams;
 
@@ -16,7 +19,6 @@ public function testCreateCollectionWithVector(): void
         $collection = new CreateCollection();
         $collection->addVector(new VectorParams(1024, VectorParams::DISTANCE_COSINE), 'image');
 
-
         $this->assertEquals(
             [
                 'vectors' => [
@@ -140,4 +142,79 @@ public function testCreateCollectionWithOnDiskPayload(): void
             $collection->toArray()
         );
     }
+
+    public function testCreateCollectionWithOptimizersConfigDiff(): void
+    {
+        $collection = new CreateCollection();
+        $collection->addVector(new VectorParams(1024, VectorParams::DISTANCE_COSINE));
+        $diff = (new OptimizersConfigDiff())
+            ->setMaxSegmentSize(1)
+            ->setDefaultSegmentNumber(1);
+
+        $collection->setOptimizersConfigDiff($diff);
+
+        $this->assertEquals(
+            [
+                'vectors' => [
+                    'size' => '1024',
+                    'distance' => 'Cosine'
+                ],
+                'optimizers_config' => [
+                    'max_segment_size' => 1,
+                    'default_segment_number' => 1,
+                ]
+            ],
+            $collection->toArray()
+        );
+    }
+
+    public function testCreateCollectionWithHnswConfig(): void
+    {
+        $collection = new CreateCollection();
+        $collection->addVector(new VectorParams(1024, VectorParams::DISTANCE_COSINE));
+        $diff = (new HnswConfig())
+            ->setM(1)
+            ->setPayloadM(1);
+
+        $collection->setHnswConfig($diff);
+
+        $this->assertEquals(
+            [
+                'vectors' => [
+                    'size' => '1024',
+                    'distance' => 'Cosine'
+                ],
+                'hnsw_config' => [
+                    'm' => 1,
+                    'payload_m' => 1,
+                ]
+            ],
+            $collection->toArray()
+        );
+    }
+
+    public function testCreateCollectionWithWalConfig(): void
+    {
+        $collection = new CreateCollection();
+        $collection->addVector(new VectorParams(1024, VectorParams::DISTANCE_COSINE));
+        $diff = (new WalConfig())
+            ->setWalSegmentsAhead(1)
+            ->setWalCapacityMb(1);
+
+        $collection->setWalConfig($diff);
+
+        $this->assertEquals(
+            [
+                'vectors' => [
+                    'size' => '1024',
+                    'distance' => 'Cosine'
+                ],
+                'wal_config' => [
+                    'wal_segments_ahead' => 1,
+                    'wal_capacity_mb' => 1,
+                ]
+            ],
+            $collection->toArray()
+        );
+    }
 }