Support HNSW_PQ/HNSW_SQ/HNSW_PRQ (#1300)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
milvus-io · Feb 8, 2025 · 977de37 · 977de37
1 parent daf567a
commit 977de37
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 5 deletions.
diff --git a/docker-compose.yml b/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

diff --git a/sdk-bulkwriter/src/test/java/io/milvus/bulkwriter/BulkWriterTest.java b/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() {

diff --git a/sdk-core/src/main/java/io/milvus/param/IndexType.java b/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),

diff --git a/sdk-core/src/main/java/io/milvus/v2/common/IndexParam.java b/sdk-core/src/main/java/io/milvus/v2/common/IndexParam.java
@@ -70,6 +70,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),

diff --git a/sdk-core/src/test/java/io/milvus/client/MilvusClientDockerTest.java b/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() {

diff --git a/sdk-core/src/test/java/io/milvus/v2/client/MilvusClientV2DockerTest.java b/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() {