diff --git a/sdk-core/src/main/java/io/milvus/common/clientenum/FunctionType.java b/sdk-core/src/main/java/io/milvus/common/clientenum/FunctionType.java index 3de64fdcb..e582457d2 100644 --- a/sdk-core/src/main/java/io/milvus/common/clientenum/FunctionType.java +++ b/sdk-core/src/main/java/io/milvus/common/clientenum/FunctionType.java @@ -19,7 +19,19 @@ package io.milvus.common.clientenum; +import lombok.Getter; + public enum FunctionType { - UNKNOWN, - BM25, + UNKNOWN(0), + BM25(1), + TextEmbedding(2), + Rerank(3), + ; + + @Getter + private final int code; + + FunctionType(int i) { + code = i; + } } diff --git a/sdk-core/src/main/java/io/milvus/v2/service/collection/request/CreateCollectionReq.java b/sdk-core/src/main/java/io/milvus/v2/service/collection/request/CreateCollectionReq.java index ea3d7b58b..989ddb7b7 100644 --- a/sdk-core/src/main/java/io/milvus/v2/service/collection/request/CreateCollectionReq.java +++ b/sdk-core/src/main/java/io/milvus/v2/service/collection/request/CreateCollectionReq.java @@ -231,7 +231,8 @@ public static class Function { private String name; @Builder.Default private String description = ""; - private FunctionType functionType; + @Builder.Default + private FunctionType functionType = FunctionType.UNKNOWN; @Builder.Default private List inputFieldNames = new ArrayList<>(); @Builder.Default diff --git a/sdk-core/src/main/java/io/milvus/v2/utils/SchemaUtils.java b/sdk-core/src/main/java/io/milvus/v2/utils/SchemaUtils.java index 60ceeb094..becf8cd47 100644 --- a/sdk-core/src/main/java/io/milvus/v2/utils/SchemaUtils.java +++ b/sdk-core/src/main/java/io/milvus/v2/utils/SchemaUtils.java @@ -129,7 +129,7 @@ public static FunctionSchema convertToGrpcFunction(CreateCollectionReq.Function FunctionSchema.Builder builder = FunctionSchema.newBuilder() .setName(function.getName()) .setDescription(function.getDescription()) - .setType(FunctionType.valueOf(function.getFunctionType().name())); + .setType(FunctionType.forNumber(function.getFunctionType().getCode())); for (String name : function.getInputFieldNames()) { builder.addInputFieldNames(name);