Browse Source

add new index type for v2 client: INVERTED, GPU_CAGRA, sparse vector related, etc. (#843)

Signed-off-by: Nian Liu <nian.liu@zilliz.com>
Nian Liu 1 year ago
parent
commit
2f9b2c462d
1 changed files with 15 additions and 8 deletions
  1. 15 8
      src/main/java/io/milvus/v2/common/IndexParam.java

+ 15 - 8
src/main/java/io/milvus/v2/common/IndexParam.java

@@ -41,8 +41,8 @@ public class IndexParam {
 
 
     @Getter
     @Getter
     public enum IndexType {
     public enum IndexType {
-        INVALID,
-        //Only supported for float vectors
+        None(0),
+        // Only supported for float vectors
         FLAT(1),
         FLAT(1),
         IVF_FLAT(2),
         IVF_FLAT(2),
         IVF_SQ8(3),
         IVF_SQ8(3),
@@ -52,20 +52,27 @@ public class IndexParam {
         AUTOINDEX(11),
         AUTOINDEX(11),
         SCANN(12),
         SCANN(12),
 
 
-        // GPU index
+        // GPU indexes only for float vectors
         GPU_IVF_FLAT(50),
         GPU_IVF_FLAT(50),
         GPU_IVF_PQ(51),
         GPU_IVF_PQ(51),
+        GPU_BRUTE_FORCE(52),
+        GPU_CAGRA(53),
 
 
-        //Only supported for binary vectors
+        // Only supported for binary vectors
         BIN_FLAT(80),
         BIN_FLAT(80),
         BIN_IVF_FLAT(81),
         BIN_IVF_FLAT(81),
 
 
-        //Scalar field index start from here
-        //Only for varchar type field
+        // Only for varchar type field
         TRIE("Trie", 100),
         TRIE("Trie", 100),
-        //Only for scalar type field
-        STL_SORT(200),
+        // Only for scalar type field
+        STL_SORT(200), // only for numeric type field
+        INVERTED(201), // works for all scalar fields except JSON type field
+
+        // Only for sparse vectors
+        SPARSE_INVERTED_INDEX(300),
+        SPARSE_WAND(301)
         ;
         ;
+
         private final String name;
         private final String name;
         private final int code;
         private final int code;