Merge branch 'v2.x' into vv-geoshape-queries

vladvildanov · web-flow · commit f6f0dec12833 · 2024-07-01T14:46:03.000+03:00
diff --git a/tests/Predis/Command/Redis/Search/FTCREATE_Test.php b/tests/Predis/Command/Redis/Search/FTCREATE_Test.php
@@ -16,6 +16,7 @@
 use Predis\Command\Argument\Search\SchemaFields\NumericField;
 use Predis\Command\Argument\Search\SchemaFields\TagField;
 use Predis\Command\Argument\Search\SchemaFields\TextField;
+use Predis\Command\Argument\Search\SchemaFields\VectorField;
 use Predis\Command\Redis\PredisCommandTestCase;
 
 /**
@@ -86,6 +87,32 @@ public function testCreatesSearchIndexWithGivenArgumentsAndSchema(): void
         $this->assertEquals('OK', $actualResponse);
     }
 
+    /**
+     * @group connected
+     * @group relay-resp3
+     * @return void
+     * @requiresRediSearchVersion >= 2.9.0
+     */
+    public function testCreatesSearchIndexWithFloat16Vector(): void
+    {
+        $redis = $this->getClient();
+
+        $schema = [
+            new VectorField('float16',
+                'FLAT',
+                ['TYPE', 'FLOAT16', 'DIM', 768, 'DISTANCE_METRIC', 'COSINE']
+            ),
+            new VectorField('bfloat16',
+                'FLAT',
+                ['TYPE', 'BFLOAT16', 'DIM', 768, 'DISTANCE_METRIC', 'COSINE']
+            ),
+        ];
+
+        $actualResponse = $redis->ftcreate('index', $schema);
+
+        $this->assertEquals('OK', $actualResponse);
+    }
+
     public function argumentsProvider(): array
     {
         return [
