Browse Source

Group by for V2 (#918)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 1 year ago
parent
commit
9c84c0ff5a

+ 1 - 0
src/main/java/io/milvus/v2/service/vector/request/SearchReq.java

@@ -56,4 +56,5 @@ public class SearchReq {
     @Builder.Default
     private ConsistencyLevel consistencyLevel = ConsistencyLevel.BOUNDED;
     private boolean ignoreGrowing;
+    private String groupByFieldName;
 }

+ 8 - 0
src/main/java/io/milvus/v2/utils/VectorUtils.java

@@ -166,6 +166,14 @@ public class VectorUtils {
             }
         }
 
+        if (request.getGroupByFieldName() != null && !request.getGroupByFieldName().isEmpty()) {
+            builder.addSearchParams(
+                    KeyValuePair.newBuilder()
+                            .setKey(Constant.GROUP_BY_FIELD)
+                            .setValue(request.getGroupByFieldName())
+                            .build());
+        }
+
         if (!request.getOutputFields().isEmpty()) {
             request.getOutputFields().forEach(builder::addOutputFields);
         }