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

Signed-off-by: yhmo <[email protected]>
milvus-io · Dec 16, 2024 · 65040cc · 65040cc
1 parent a0cced0
commit 65040cc
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/src/main/java/io/milvus/v2/service/vector/VectorService.java b/src/main/java/io/milvus/v2/service/vector/VectorService.java
@@ -178,6 +178,7 @@ public SearchResp search(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStu
 
         return SearchResp.builder()
                 .searchResults(convertUtils.getEntities(response))
+                .recalls(response.getResults().getRecallsList())
                 .build();
     }
 
@@ -194,6 +195,7 @@ public SearchResp hybridSearch(MilvusServiceGrpc.MilvusServiceBlockingStub block
 
         return SearchResp.builder()
                 .searchResults(convertUtils.getEntities(response))
+                .recalls(response.getResults().getRecallsList())
                 .build();
     }
 

diff --git a/src/main/java/io/milvus/v2/service/vector/response/SearchResp.java b/src/main/java/io/milvus/v2/service/vector/response/SearchResp.java
@@ -29,6 +29,7 @@
 @SuperBuilder
 public class SearchResp {
     private List<List<SearchResult>> searchResults;
+    private List<Float> recalls;
 
     @Data
     @SuperBuilder

diff --git a/src/main/milvus-proto b/src/main/milvus-proto
+546 −527		go-api/commonpb/common.pb.go
+4,031 −3,859		go-api/milvuspb/milvus.pb.go
+37 −0		go-api/milvuspb/milvus_grpc.pb.go
+136 −32		go-api/msgpb/msg.pb.go
+269 −249		go-api/schemapb/schema.pb.go
+3 −0		proto/common.proto
+24 −4		proto/milvus.proto
+7 −0		proto/msg.proto
+2 −0		proto/schema.proto