diff --git a/src/main/java/io/milvus/v2/service/collection/CollectionService.java b/src/main/java/io/milvus/v2/service/collection/CollectionService.java index a81f05793..130888f3c 100644 --- a/src/main/java/io/milvus/v2/service/collection/CollectionService.java +++ b/src/main/java/io/milvus/v2/service/collection/CollectionService.java @@ -249,6 +249,7 @@ public Void loadCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingS .setRefresh(request.getRefresh()) .addAllLoadFields(request.getLoadFields()) .setSkipLoadDynamicField(request.getSkipLoadDynamicField()) + .addAllResourceGroups(request.getResourceGroups()) .build(); Status status = blockingStub.loadCollection(loadCollectionRequest); rpcUtils.handleResponse(title, status); diff --git a/src/main/java/io/milvus/v2/service/collection/request/LoadCollectionReq.java b/src/main/java/io/milvus/v2/service/collection/request/LoadCollectionReq.java index 4c6117f43..d6f9e2534 100644 --- a/src/main/java/io/milvus/v2/service/collection/request/LoadCollectionReq.java +++ b/src/main/java/io/milvus/v2/service/collection/request/LoadCollectionReq.java @@ -42,4 +42,6 @@ public class LoadCollectionReq { private List loadFields = new ArrayList<>(); @Builder.Default private Boolean skipLoadDynamicField = Boolean.FALSE; + @Builder.Default + private List resourceGroups = new ArrayList<>(); } diff --git a/src/main/java/io/milvus/v2/service/partition/PartitionService.java b/src/main/java/io/milvus/v2/service/partition/PartitionService.java index c67bd24e5..22c231616 100644 --- a/src/main/java/io/milvus/v2/service/partition/PartitionService.java +++ b/src/main/java/io/milvus/v2/service/partition/PartitionService.java @@ -102,6 +102,7 @@ public Void loadPartitions(MilvusServiceGrpc.MilvusServiceBlockingStub blockingS .setRefresh(request.getRefresh()) .addAllLoadFields(request.getLoadFields()) .setSkipLoadDynamicField(request.getSkipLoadDynamicField()) + .addAllResourceGroups(request.getResourceGroups()) .build(); Status status = blockingStub.loadPartitions(loadPartitionsRequest); rpcUtils.handleResponse(title, status); diff --git a/src/main/java/io/milvus/v2/service/partition/request/LoadPartitionsReq.java b/src/main/java/io/milvus/v2/service/partition/request/LoadPartitionsReq.java index 0323cb549..026108533 100644 --- a/src/main/java/io/milvus/v2/service/partition/request/LoadPartitionsReq.java +++ b/src/main/java/io/milvus/v2/service/partition/request/LoadPartitionsReq.java @@ -38,4 +38,6 @@ public class LoadPartitionsReq { private List loadFields = new ArrayList<>(); @Builder.Default private Boolean skipLoadDynamicField = Boolean.FALSE; + @Builder.Default + private List resourceGroups = new ArrayList<>(); }