Jelajahi Sumber

Support HNSW_PQ/HNSW_SQ/HNSW_PRQ (#1301)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 2 bulan lalu
induk
melakukan
7accd52b23

+ 2 - 2
docker-compose.yml

@@ -32,7 +32,7 @@ services:
 
   standalone:
     container_name: milvus-javasdk-test-standalone
-    image: milvusdb/milvus:v2.5.2
+    image: milvusdb/milvus:v2.5.4
     command: ["milvus", "run", "standalone"]
     environment:
       ETCD_ENDPOINTS: etcd:2379
@@ -77,7 +77,7 @@ services:
 
   standaloneslave:
     container_name: milvus-javasdk-test-slave-standalone
-    image: milvusdb/milvus:v2.5.2
+    image: milvusdb/milvus:v2.5.4
     command: ["milvus", "run", "standalone"]
     environment:
       ETCD_ENDPOINTS: etcdslave:2379

+ 1 - 1
sdk-bulkwriter/src/test/java/io/milvus/bulkwriter/BulkWriterTest.java

@@ -54,7 +54,7 @@ public class BulkWriterTest {
     private static final TestUtils utils = new TestUtils(DIMENSION);
 
     @Container
-    private static final MilvusContainer milvus = new MilvusContainer("milvusdb/milvus:v2.5.0-beta");
+    private static final MilvusContainer milvus = new MilvusContainer("milvusdb/milvus:v2.5.4");
 
     @BeforeAll
     public static void setUp() {

+ 3 - 0
sdk-core/src/main/java/io/milvus/param/IndexType.java

@@ -33,6 +33,9 @@ public enum IndexType {
     IVF_SQ8(3),
     IVF_PQ(4),
     HNSW(5),
+    HNSW_SQ(6),
+    HNSW_PQ(7),
+    HNSW_PRQ(8),
     DISKANN(10),
     AUTOINDEX(11),
     SCANN(12),

+ 3 - 0
sdk-core/src/main/java/io/milvus/v2/common/IndexParam.java

@@ -70,6 +70,9 @@ public class IndexParam {
         IVF_SQ8(3),
         IVF_PQ(4),
         HNSW(5),
+        HNSW_SQ(6),
+        HNSW_PQ(7),
+        HNSW_PRQ(8),
         DISKANN(10),
         AUTOINDEX(11),
         SCANN(12),

+ 1 - 1
sdk-core/src/test/java/io/milvus/client/MilvusClientDockerTest.java

@@ -78,7 +78,7 @@ class MilvusClientDockerTest {
     private static final TestUtils utils = new TestUtils(DIMENSION);
 
     @Container
-    private static final MilvusContainer milvus = new MilvusContainer("milvusdb/milvus:v2.5.2");
+    private static final MilvusContainer milvus = new MilvusContainer("milvusdb/milvus:v2.5.4");
 
     @BeforeAll
     public static void setUp() {

+ 1 - 1
sdk-core/src/test/java/io/milvus/v2/client/MilvusClientV2DockerTest.java

@@ -79,7 +79,7 @@ class MilvusClientV2DockerTest {
     private static final TestUtils utils = new TestUtils(DIMENSION);
 
     @Container
-    private static final MilvusContainer milvus = new MilvusContainer("milvusdb/milvus:v2.5.2");
+    private static final MilvusContainer milvus = new MilvusContainer("milvusdb/milvus:v2.5.4");
 
     @BeforeAll
     public static void setUp() {