support search group-by(#1809) (#1810)

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

pymilvus/client/constants.py

@@ -9,3 +9,4 @@
 DEFAULT_CONSISTENCY_LEVEL = ConsistencyLevel.Bounded
 DEFAULT_RESOURCE_GROUP = "__default_resource_group"
 REDUCE_STOP_FOR_BEST = "reduce_stop_for_best"
+GROUP_BY_FIELD = "group_by_field"

pymilvus/client/prepare.py

@@ -13,7 +13,7 @@
 
 from . import blob, entity_helper, ts_utils
 from .check import check_pass_param, is_legal_collection_properties
-from .constants import DEFAULT_CONSISTENCY_LEVEL, REDUCE_STOP_FOR_BEST
+from .constants import DEFAULT_CONSISTENCY_LEVEL, GROUP_BY_FIELD, REDUCE_STOP_FOR_BEST
 from .types import DataType, PlaceholderType, get_consistency_level
 from .utils import traverse_info, traverse_rows_info
 
@@ -612,6 +612,9 @@ def search_requests_with_expr(
         if param.get("metric_type", None) is not None:
             search_params["metric_type"] = param["metric_type"]
 
+        if param.get(GROUP_BY_FIELD, None) is not None:
+            search_params[GROUP_BY_FIELD] = param[GROUP_BY_FIELD]
+
         if anns_field:
             search_params["anns_field"] = anns_field
 

pymilvus/orm/iterator.py

@@ -168,7 +168,7 @@ def __check_reached_limit(self, ret: List):
     def __setup__pk_prop(self):
         fields = self._schema[FIELDS]
         for field in fields:
-            if field[IS_PRIMARY]:
+            if IS_PRIMARY in field and field[IS_PRIMARY]:
                 if field["type"] == DataType.VARCHAR:
                     self._pk_str = True
                 else: