feat: support group_size parameter for search_group_by (#2130)

related: #2129

Signed-off-by: MrPresent-Han <[email protected]>
Co-authored-by: MrPresent-Han <[email protected]>

Author: MrPresent-Han

pymilvus/client/constants.py

@@ -10,6 +10,7 @@
 DEFAULT_RESOURCE_GROUP = "__default_resource_group"
 REDUCE_STOP_FOR_BEST = "reduce_stop_for_best"
 GROUP_BY_FIELD = "group_by_field"
+GROUP_SIZE = "group_size"
 ITERATOR_FIELD = "iterator"
 
 RANKER_TYPE_RRF = "rrf"

pymilvus/client/prepare.py

@@ -15,6 +15,7 @@
 from .constants import (
     DEFAULT_CONSISTENCY_LEVEL,
     GROUP_BY_FIELD,
+    GROUP_SIZE,
     ITERATOR_FIELD,
     REDUCE_STOP_FOR_BEST,
 )
@@ -667,6 +668,10 @@ def search_requests_with_expr(
         if group_by_field is not None:
             search_params[GROUP_BY_FIELD] = group_by_field
 
+        group_size = kwargs.get(GROUP_SIZE)
+        if group_size is not None:
+            search_params[GROUP_SIZE] = group_size
+
         if param.get("metric_type") is not None:
             search_params["metric_type"] = param["metric_type"]