diff --git a/internal/proxy/search_util.go b/internal/proxy/search_util.go index c4e4b992fb771..e717416eab175 100644 --- a/internal/proxy/search_util.go +++ b/internal/proxy/search_util.go @@ -103,7 +103,7 @@ func parseSearchIteratorV2Info(searchParamsPair []*commonpb.KeyValuePair, groupB } // parse token, generate if not exist - token, _ := funcutil.GetAttrByKeyFromRepeatedKV(SearchIterTokenKey, searchParamsPair) + token, _ := funcutil.GetAttrByKeyFromRepeatedKV(SearchIterIdKey, searchParamsPair) if token == "" { generatedToken, err := uuid.NewRandom() if err != nil { diff --git a/internal/proxy/task.go b/internal/proxy/task.go index 42ebc280fa3ce..51c8d4c758bdd 100644 --- a/internal/proxy/task.go +++ b/internal/proxy/task.go @@ -70,7 +70,7 @@ const ( SearchIterV2Key = "search_iter_v2" SearchIterBatchSizeKey = "search_iter_batch_size" SearchIterLastBoundKey = "search_iter_last_bound" - SearchIterTokenKey = "search_iter_token" + SearchIterIdKey = "search_iter_id" InsertTaskName = "InsertTask" CreateCollectionTaskName = "CreateCollectionTask" diff --git a/internal/proxy/task_search_test.go b/internal/proxy/task_search_test.go index 686884a8cc9fb..aff17affd3af7 100644 --- a/internal/proxy/task_search_test.go +++ b/internal/proxy/task_search_test.go @@ -2780,7 +2780,7 @@ func TestTaskSearch_parseSearchInfo(t *testing.T) { t.Run("iteratorV2 invalid token", func(t *testing.T) { param := generateValidParamsForSearchIteratorV2() param = append(param, &commonpb.KeyValuePair{ - Key: SearchIterTokenKey, + Key: SearchIterIdKey, Value: "invalid_token", }) searchInfo := parseSearchInfo(param, nil, nil) @@ -2793,7 +2793,7 @@ func TestTaskSearch_parseSearchInfo(t *testing.T) { assert.NoError(t, err) param := generateValidParamsForSearchIteratorV2() param = append(param, &commonpb.KeyValuePair{ - Key: SearchIterTokenKey, + Key: SearchIterIdKey, Value: token.String(), }) searchInfo := parseSearchInfo(param, nil, nil)