enhance: Return segment id hint in QueryStream response (#36487)

Related to #36482 This PR reuses `SealedSegmentIDsRetrieved` field in `RetrieveResults` struct to store segment id hint. Signed-off-by: Congqi Xia <[email protected]>
milvus-io · Sep 26, 2024 · 4fd9b0a · 4fd9b0a
1 parent 9e8cafc
commit 4fd9b0a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/querynodev2/segments/retrieve.go b/internal/querynodev2/segments/retrieve.go
@@ -124,7 +124,8 @@ func retrieveOnSegmentsWithStream(ctx context.Context, mgr *Manager, segments []
 					CostAggregation: &internalpb.CostAggregation{
 						TotalRelatedDataSize: GetSegmentRelatedDataSize(segment),
 					},
-					AllRetrieveCount: result.GetAllRetrieveCount(),
+					SealedSegmentIDsRetrieved: []int64{segment.ID()},
+					AllRetrieveCount:          result.GetAllRetrieveCount(),
 				}); err != nil {
 					errs[i] = err
 				}