Return recall rate in search result for V2

Signed-off-by: yhmo <[email protected]>
milvus-io · Dec 16, 2024 · 63df4f7 · 63df4f7
1 parent 546f00a
commit 63df4f7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/sdk-core/src/main/java/io/milvus/v2/service/vector/VectorService.java b/sdk-core/src/main/java/io/milvus/v2/service/vector/VectorService.java
@@ -181,6 +181,7 @@ public SearchResp search(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStu
         return SearchResp.builder()
                 .searchResults(convertUtils.getEntities(response))
                 .sessionTs(response.getSessionTs())
+                .recalls(response.getResults().getRecallsList())
                 .build();
     }
 
@@ -197,6 +198,7 @@ public SearchResp hybridSearch(MilvusServiceGrpc.MilvusServiceBlockingStub block
 
         return SearchResp.builder()
                 .searchResults(convertUtils.getEntities(response))
+                .recalls(response.getResults().getRecallsList())
                 .build();
     }
 

diff --git a/sdk-core/src/main/java/io/milvus/v2/service/vector/response/SearchResp.java b/sdk-core/src/main/java/io/milvus/v2/service/vector/response/SearchResp.java
@@ -30,6 +30,7 @@
 public class SearchResp {
     private List<List<SearchResult>> searchResults;
     private long sessionTs = 1L; // default eventually ts
+    private List<Float> recalls;
 
     @Data
     @SuperBuilder