fix: Pass offset param in rerank params for HybridSearch

See also milvus-io/milvus#32562 Signed-off-by: Congqi Xia <[email protected]>
milvus-io · Apr 26, 2024 · ddc5800 · ddc5800
1 parent 41f8c8d
commit ddc5800
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/client/data.go b/client/data.go
@@ -67,8 +67,13 @@ func (c *GrpcClient) HybridSearch(ctx context.Context, collName string, partitio
 		sReqs = append(sReqs, r)
 	}
 
+	opt := &SearchQueryOption{}
+	for _, o := range opts {
+		o(opt)
+	}
 	params := reranker.GetParams()
 	params = append(params, &commonpb.KeyValuePair{Key: limitKey, Value: strconv.FormatInt(int64(limit), 10)})
+	params = append(params, &commonpb.KeyValuePair{Key: offsetKey, Value: strconv.FormatInt(int64(opt.Offset), 10)})
 
 	req := &milvuspb.HybridSearchRequest{
 		CollectionName:   collName,

diff --git a/client/reranker.go b/client/reranker.go
@@ -17,6 +17,7 @@ const (
 )
 
 type Reranker interface {
+	WithOffset(offset int64)
 	GetParams() []*commonpb.KeyValuePair
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -17,6 +17,7 @@ const ( @@
     )
     type Reranker interface {
+    	WithOffset(offset int64)
     	GetParams() []*commonpb.KeyValuePair
     }
@@ Expand Down @@