From b38abbe7f602e68872d8cc60459af236f8d03892 Mon Sep 17 00:00:00 2001 From: yhmo Date: Sat, 8 Feb 2025 14:39:39 +0800 Subject: [PATCH] Support HNSW_PQ/HNSW_SQ/HNSW_PRQ Signed-off-by: yhmo --- docker-compose.yml | 4 ++-- .../src/test/java/io/milvus/bulkwriter/BulkWriterTest.java | 2 +- sdk-core/src/main/java/io/milvus/param/IndexType.java | 3 +++ sdk-core/src/main/java/io/milvus/v2/common/IndexParam.java | 3 +++ .../test/java/io/milvus/client/MilvusClientDockerTest.java | 2 +- .../java/io/milvus/v2/client/MilvusClientV2DockerTest.java | 2 +- 6 files changed, 11 insertions(+), 5 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index ab02a84ba..4818f3735 100644 --- 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 index 41e38313d..3fe806146 100644 --- 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 index 66701998c..6b59078aa 100644 --- 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 index 1d66c02ea..4658ebedd 100644 --- 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 index dfe606e2e..902fced32 100644 --- 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 index 5c6550dec..eb1738850 100644 --- 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() {