From c2939118bc686a4fe4395d2fc4b3ad5e5db4bc79 Mon Sep 17 00:00:00 2001 From: ofekshenawa Date: Mon, 15 Jul 2024 10:49:42 +0300 Subject: [PATCH] Add tests case with FLOAT16 and BFLOAT16 vectors --- search_test.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/search_test.go b/search_test.go index 60888ef5..344e6bd9 100644 --- a/search_test.go +++ b/search_test.go @@ -1043,6 +1043,16 @@ var _ = Describe("RediSearch commands", Label("search"), func() { Expect(err).NotTo(HaveOccurred()) Expect(res2.Total).To(BeEquivalentTo(int64(2))) }) + + It("should create search index with FLOAT16 and BFLOAT16 vectors", Label("search", "ftcreate"), func() { + val, err := client.FTCreate(ctx, "index", &redis.FTCreateOptions{}, + &redis.FieldSchema{FieldName: "float16", FieldType: redis.SearchFieldTypeVector, VectorArgs: &redis.FTVectorArgs{FlatOptions: &redis.FTFlatOptions{Type: "FLOAT16", Dim: 768, DistanceMetric: "COSINE"}}}, + &redis.FieldSchema{FieldName: "bfloat16", FieldType: redis.SearchFieldTypeVector, VectorArgs: &redis.FTVectorArgs{FlatOptions: &redis.FTFlatOptions{Type: "BFLOAT16", Dim: 768, DistanceMetric: "COSINE"}}}, + ).Result() + Expect(err).NotTo(HaveOccurred()) + Expect(val).To(BeEquivalentTo("OK")) + WaitForIndexing(client, "index") + }) }) // It("should FTProfile Search and Aggregate", Label("search", "ftprofile"), func() {