2
0
Эх сурвалжийг харах

Function parameters (#1428)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 3 долоо хоног өмнө
parent
commit
bc51a10e7b

+ 2 - 2
sdk-core/src/main/java/io/milvus/common/clientenum/FunctionType.java

@@ -24,8 +24,8 @@ import lombok.Getter;
 public enum FunctionType {
     UNKNOWN(0),
     BM25(1),
-    TextEmbedding(2),
-    Rerank(3),
+    TEXTEMBEDDING(2),
+    RERANK(3),
     ;
 
     @Getter

+ 13 - 0
sdk-core/src/main/java/io/milvus/v2/service/collection/request/CreateCollectionReq.java

@@ -200,5 +200,18 @@ public class CreateCollectionReq {
         private List<String> inputFieldNames = new ArrayList<>();
         @Builder.Default
         private List<String> outputFieldNames = new ArrayList<>();
+        @Builder.Default
+        private Map<String, String> params = new HashMap<>();
+
+        public static abstract class FunctionBuilder<C extends Function, B extends Function.FunctionBuilder<C, B>> {
+            public B param(String key, String value) {
+                if(null == this.params$value ){
+                    this.params$value = new HashMap<>();
+                }
+                this.params$value.put(key, value);
+                this.params$set = true;
+                return self();
+            }
+        }
     }
 }

+ 5 - 0
sdk-core/src/main/java/io/milvus/v2/utils/SchemaUtils.java

@@ -139,6 +139,11 @@ public class SchemaUtils {
             builder.addOutputFieldNames(name);
         }
 
+        List<KeyValuePair> params = ParamUtils.AssembleKvPair(function.getParams());
+        if (CollectionUtils.isNotEmpty(params)) {
+            params.forEach(builder::addParams);
+        }
+
         return builder.build();
     }