Browse Source

Return recall rate in search result for V2 (#1237)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 4 months ago
parent
commit
65040cc669

+ 2 - 0
src/main/java/io/milvus/v2/service/vector/VectorService.java

@@ -178,6 +178,7 @@ public class VectorService extends BaseService {
 
         return SearchResp.builder()
                 .searchResults(convertUtils.getEntities(response))
+                .recalls(response.getResults().getRecallsList())
                 .build();
     }
 
@@ -194,6 +195,7 @@ public class VectorService extends BaseService {
 
         return SearchResp.builder()
                 .searchResults(convertUtils.getEntities(response))
+                .recalls(response.getResults().getRecallsList())
                 .build();
     }
 

+ 1 - 0
src/main/java/io/milvus/v2/service/vector/response/SearchResp.java

@@ -29,6 +29,7 @@ import java.util.Map;
 @SuperBuilder
 public class SearchResp {
     private List<List<SearchResult>> searchResults;
+    private List<Float> recalls;
 
     @Data
     @SuperBuilder

+ 1 - 1
src/main/milvus-proto

@@ -1 +1 @@
-Subproject commit a1c2ac2fd2c12e0e6ff73ada3f260966f1cb65e0
+Subproject commit 7f744ed2ad119db58a53d82aeaa61142a0213b72