Browse Source

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

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 6 months ago
parent
commit
d580cca392

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

@@ -181,6 +181,7 @@ public class VectorService extends BaseService {
         return SearchResp.builder()
         return SearchResp.builder()
                 .searchResults(convertUtils.getEntities(response))
                 .searchResults(convertUtils.getEntities(response))
                 .sessionTs(response.getSessionTs())
                 .sessionTs(response.getSessionTs())
+                .recalls(response.getResults().getRecallsList())
                 .build();
                 .build();
     }
     }
 
 
@@ -197,6 +198,7 @@ public class VectorService extends BaseService {
 
 
         return SearchResp.builder()
         return SearchResp.builder()
                 .searchResults(convertUtils.getEntities(response))
                 .searchResults(convertUtils.getEntities(response))
+                .recalls(response.getResults().getRecallsList())
                 .build();
                 .build();
     }
     }
 
 

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

@@ -30,6 +30,7 @@ import java.util.Map;
 public class SearchResp {
 public class SearchResp {
     private List<List<SearchResult>> searchResults;
     private List<List<SearchResult>> searchResults;
     private long sessionTs = 1L; // default eventually ts
     private long sessionTs = 1L; // default eventually ts
+    private List<Float> recalls;
 
 
     @Data
     @Data
     @SuperBuilder
     @SuperBuilder

+ 1 - 1
sdk-core/src/main/milvus-proto

@@ -1 +1 @@
-Subproject commit 93892e628c6908f6dfda6d7eaad3ba1d66037a97
+Subproject commit c18fc42493ca2768a81bf2346340325db1e6cab9