Przeglądaj źródła

More parameters for RunAnalyzer (#1429)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 3 miesięcy temu
rodzic
commit
53ff76d534

+ 2 - 2
docker-compose.yml

@@ -32,7 +32,7 @@ services:
 
   standalone:
     container_name: milvus-javasdk-test-standalone
-    image: milvusdb/milvus:v2.5.11
+    image: milvusdb/milvus:v2.5.13
     command: ["milvus", "run", "standalone"]
     environment:
       ETCD_ENDPOINTS: etcd:2379
@@ -77,7 +77,7 @@ services:
 
   standaloneslave:
     container_name: milvus-javasdk-test-slave-standalone
-    image: milvusdb/milvus:v2.5.11
+    image: milvusdb/milvus:v2.5.13
     command: ["milvus", "run", "standalone"]
     environment:
       ETCD_ENDPOINTS: etcdslave:2379

+ 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<>();
 }

+ 1 - 1
sdk-core/src/test/java/io/milvus/TestUtils.java

@@ -11,7 +11,7 @@ public class TestUtils {
     private int dimension = 256;
     private static final Random RANDOM = new Random();
 
-    public static final String MilvusDockerImageID = "milvusdb/milvus:v2.5.11";
+    public static final String MilvusDockerImageID = "milvusdb/milvus:v2.5.13";
 
     public TestUtils(int dimension) {
         this.dimension = dimension;