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 39c53e724..c49944911 100644 --- a/src/main/java/io/milvus/v2/service/collection/CollectionService.java +++ b/src/main/java/io/milvus/v2/service/collection/CollectionService.java @@ -123,7 +123,8 @@ public Void createCollectionWithSchema(MilvusServiceGrpc.MilvusServiceBlockingSt CreateCollectionRequest.Builder builder = CreateCollectionRequest.newBuilder() .setCollectionName(request.getCollectionName()) .setSchema(grpcSchema.toByteString()) - .setShardsNum(request.getNumShards()); + .setShardsNum(request.getNumShards()) + .setConsistencyLevelValue(request.getConsistencyLevel().getCode()); List propertiesList = ParamUtils.AssembleKvPair(request.getProperties()); if (CollectionUtils.isNotEmpty(propertiesList)) { propertiesList.forEach(builder::addProperties); diff --git a/src/main/java/io/milvus/v2/service/collection/response/DescribeCollectionResp.java b/src/main/java/io/milvus/v2/service/collection/response/DescribeCollectionResp.java index 2f0ea8478..fb6d71ee8 100644 --- a/src/main/java/io/milvus/v2/service/collection/response/DescribeCollectionResp.java +++ b/src/main/java/io/milvus/v2/service/collection/response/DescribeCollectionResp.java @@ -46,6 +46,7 @@ public class DescribeCollectionResp { private CreateCollectionReq.CollectionSchema collectionSchema; private Long createTime; private ConsistencyLevel consistencyLevel; + private Integer shardsNum; @Builder.Default private final Map properties = new HashMap<>(); } diff --git a/src/main/java/io/milvus/v2/utils/ConvertUtils.java b/src/main/java/io/milvus/v2/utils/ConvertUtils.java index 86613926b..1f12d41f3 100644 --- a/src/main/java/io/milvus/v2/utils/ConvertUtils.java +++ b/src/main/java/io/milvus/v2/utils/ConvertUtils.java @@ -138,6 +138,7 @@ public DescribeCollectionResp convertDescCollectionResp(DescribeCollectionRespon .primaryFieldName(response.getSchema().getFieldsList().stream().filter(FieldSchema::getIsPrimaryKey).map(FieldSchema::getName).collect(java.util.stream.Collectors.toList()).get(0)) .createTime(response.getCreatedTimestamp()) .consistencyLevel(io.milvus.v2.common.ConsistencyLevel.valueOf(response.getConsistencyLevel().name().toUpperCase())) + .shardsNum(response.getShardsNum()) .properties(properties) .build(); return describeCollectionResp;