Browse Source

More parameters for RunAnalyzer (#1430)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 3 weeks ago
parent
commit
7fb9b0266b

+ 6 - 0
sdk-core/src/main/java/io/milvus/v2/service/vector/VectorService.java

@@ -318,12 +318,18 @@ public class VectorService extends BaseService {
             byteStrings.add(ByteString.copyFrom(text.getBytes()));
         }
 
+        List<String> analyzerNames = request.getAnalyzerNames();
+        builder.addAllAnalyzerNames(analyzerNames);
+
         String params = JsonUtils.toJson(request.getAnalyzerParams());
         System.out.println(params);
         RunAnalyzerRequest runRequest = builder.addAllPlaceholder(byteStrings)
                 .setAnalyzerParams(params)
                 .setWithDetail(request.getWithDetail())
                 .setWithHash(request.getWithHash())
+                .setDbName(request.getDatabaseName())
+                .setCollectionName(request.getCollectionName())
+                .setFieldName(request.getFieldName())
                 .build();
         RunAnalyzerResponse response = blockingStub.runAnalyzer(runRequest);
         rpcUtils.handleResponse(title, response.getStatus());

+ 8 - 0
sdk-core/src/main/java/io/milvus/v2/service/vector/request/RunAnalyzerReq.java

@@ -36,4 +36,12 @@ public class RunAnalyzerReq {
     private Boolean withDetail = Boolean.FALSE;
     @Builder.Default
     private Boolean withHash = Boolean.FALSE;
+    @Builder.Default
+    private String databaseName = "";
+    @Builder.Default
+    private String collectionName = "";
+    @Builder.Default
+    private String fieldName = "";
+    @Builder.Default
+    private List<String> analyzerNames = new ArrayList<>();
 }