feat: [2.4] support recalls field in SearchResult (#2389)

milvus issue: milvus-io/milvus#37899

Signed-off-by: chasingegg <[email protected]>

pymilvus/client/abstract.py

@@ -402,6 +402,7 @@ def __init__(
     ):
         self._nq = res.num_queries
         all_topks = res.topks
+        self.recalls = res.recalls
 
         self.cost = int(status.extra_info["report_value"] if status and status.extra_info else "0")
 
@@ -534,9 +535,9 @@ def __iter__(self) -> SequenceIterator:
     def __str__(self) -> str:
         """Only print at most 10 query results"""
         reminder = f" ... and {len(self) - 10} results remaining" if len(self) > 10 else ""
-        if self.cost:
-            return f"data: {list(map(str, self[:10]))}{reminder}, cost: {self.cost}"
-        return f"data: {list(map(str, self[:10]))}{reminder}"
+        recall_msg = f", recalls: {list(map(str, self.recalls))}" if len(self.recalls) > 0 else ""
+        cost_msg = f", cost: {self.cost}" if self.cost else ""
+        return f"data: {list(map(str, self[:10]))}{reminder}{recall_msg}{cost_msg}"
 
     __repr__ = __str__
 

pymilvus/grpc_gen/common_pb2.pyi

@@ -193,6 +193,7 @@ class MsgType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     Connect: _ClassVar[MsgType]
     ListClientInfos: _ClassVar[MsgType]
     AllocTimestamp: _ClassVar[MsgType]
+    Replicate: _ClassVar[MsgType]
     CreateCredential: _ClassVar[MsgType]
     GetCredential: _ClassVar[MsgType]
     DeleteCredential: _ClassVar[MsgType]
@@ -515,6 +516,7 @@ DataNodeTt: MsgType
 Connect: MsgType
 ListClientInfos: MsgType
 AllocTimestamp: MsgType
+Replicate: MsgType
 CreateCredential: MsgType
 GetCredential: MsgType
 DeleteCredential: MsgType
@@ -742,12 +744,14 @@ class MsgBase(_message.Message):
     def __init__(self, msg_type: _Optional[_Union[MsgType, str]] = ..., msgID: _Optional[int] = ..., timestamp: _Optional[int] = ..., sourceID: _Optional[int] = ..., targetID: _Optional[int] = ..., properties: _Optional[_Mapping[str, str]] = ..., replicateInfo: _Optional[_Union[ReplicateInfo, _Mapping]] = ...) -> None: ...
 
 class ReplicateInfo(_message.Message):
-    __slots__ = ("isReplicate", "msgTimestamp")
+    __slots__ = ("isReplicate", "msgTimestamp", "replicateID")
     ISREPLICATE_FIELD_NUMBER: _ClassVar[int]
     MSGTIMESTAMP_FIELD_NUMBER: _ClassVar[int]
+    REPLICATEID_FIELD_NUMBER: _ClassVar[int]
     isReplicate: bool
     msgTimestamp: int
-    def __init__(self, isReplicate: bool = ..., msgTimestamp: _Optional[int] = ...) -> None: ...
+    replicateID: str
+    def __init__(self, isReplicate: bool = ..., msgTimestamp: _Optional[int] = ..., replicateID: _Optional[str] = ...) -> None: ...
 
 class MsgHeader(_message.Message):
     __slots__ = ("base",)

pymilvus/grpc_gen/msg_pb2.pyi

@@ -182,3 +182,17 @@ class DataNodeTtMsg(_message.Message):
     timestamp: int
     segments_stats: _containers.RepeatedCompositeFieldContainer[_common_pb2.SegmentStats]
     def __init__(self, base: _Optional[_Union[_common_pb2.MsgBase, _Mapping]] = ..., channel_name: _Optional[str] = ..., timestamp: _Optional[int] = ..., segments_stats: _Optional[_Iterable[_Union[_common_pb2.SegmentStats, _Mapping]]] = ...) -> None: ...
+
+class ReplicateMsg(_message.Message):
+    __slots__ = ("base", "is_end", "is_cluster", "database", "collection")
+    BASE_FIELD_NUMBER: _ClassVar[int]
+    IS_END_FIELD_NUMBER: _ClassVar[int]
+    IS_CLUSTER_FIELD_NUMBER: _ClassVar[int]
+    DATABASE_FIELD_NUMBER: _ClassVar[int]
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    base: _common_pb2.MsgBase
+    is_end: bool
+    is_cluster: bool
+    database: str
+    collection: str
+    def __init__(self, base: _Optional[_Union[_common_pb2.MsgBase, _Mapping]] = ..., is_end: bool = ..., is_cluster: bool = ..., database: _Optional[str] = ..., collection: _Optional[str] = ...) -> None: ...