enhance: support recalls for milvus_client

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

pymilvus/client/abstract.py

@@ -649,7 +649,13 @@ 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 ""
-        recall_msg = f", recalls: {list(map(str, self.recalls))}" if len(self.recalls) > 0 else ""
+        recall_msg = (
+            f", recalls: {list(map(str, self.recalls[:10]))}" if len(self.recalls) > 0 else ""
+        ) + (
+            f" ... and {len(self.recalls) - 10} recall results remaining"
+            if len(self.recalls) > 10
+            else ""
+        )
         cost_msg = f", cost: {self.cost}" if self.cost else ""
         return f"data: {list(map(str, self[:10]))}{reminder}{recall_msg}{cost_msg}"
 

pymilvus/client/types.py

@@ -967,15 +967,27 @@ class ExtraList(list):
         ExtraList([1, 2, 3], extra={"total": 3})
     """
 
-    def __init__(self, *args, extra: Optional[Dict] = None, **kwargs) -> None:
+    def __init__(
+        self, *args, extra: Optional[Dict] = None, recalls: Optional[List[float]] = None, **kwargs
+    ) -> None:
         super().__init__(*args, **kwargs)
         self.extra = OmitZeroDict(extra or {})
+        self.recalls = recalls
 
     def __str__(self) -> str:
         """Only print at most 10 query results"""
+        recall_msg = (
+            f", recalls: {list(map(str, self.recalls[:10]))}"
+            if self.recalls is not None and len(self.recalls) > 0
+            else ""
+        ) + (
+            f" ... and {len(self.recalls) - 10} recall results remaining"
+            if self.recalls is not None and len(self.recalls) > 10
+            else ""
+        )
         if self.extra and self.extra.omit_zero_len() != 0:
-            return f"data: {list(map(str, self[:10]))} {'...' if len(self) > 10 else ''}, extra_info: {self.extra}"
-        return f"data: {list(map(str, self[:10]))} {'...' if len(self) > 10 else ''}"
+            return f"data: {list(map(str, self[:10]))}{' ...' if len(self) > 10 else ''}{recall_msg}, extra_info: {self.extra}"
+        return f"data: {list(map(str, self[:10]))}{' ...' if len(self) > 10 else ''}{recall_msg}"
 
     __repr__ = __str__
 

pymilvus/milvus_client/async_milvus_client.py

@@ -400,7 +400,7 @@ async def search(
                 query_result.append(hit.to_dict())
             ret.append(query_result)
 
-        return ExtraList(ret, extra=construct_cost_extra(res.cost))
+        return ExtraList(ret, extra=construct_cost_extra(res.cost), recalls=res.recalls)
 
     async def query(
         self,

pymilvus/milvus_client/milvus_client.py

@@ -421,7 +421,7 @@ def search(
                 query_result.append(hit.to_dict())
             ret.append(query_result)
 
-        return ExtraList(ret, extra=construct_cost_extra(res.cost))
+        return ExtraList(ret, extra=construct_cost_extra(res.cost), recalls=res.recalls)
 
     def query(
         self,

pyproject.toml

@@ -199,4 +199,4 @@ builtins-ignorelist = [
     "dict", # TODO
     "filter",
 ]
-
+builtins-allowed-modules = ["types"]