mirror
/
redis
mirror of https://github.com/go-redis/redis.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Add tests case with FLOAT16 and BFLOAT16 vectors

This commit is contained in:
ofekshenawa 2024-07-15 10:49:42 +03:00
parent 67824ebf86
commit c2939118bc
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
search_test.go
View File

@ -1043,6 +1043,16 @@ var _ = Describe("RediSearch commands", Label("search"), func() {
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
Expect(res2.Total).To(BeEquivalentTo(int64(2))) 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() { // It("should FTProfile Search and Aggregate", Label("search", "ftprofile"), func() {