From 487bf9483e61cf851d474cc7de98f23bb51d9a7b Mon Sep 17 00:00:00 2001 From: wt Date: Thu, 2 Dec 2021 13:13:32 +0800 Subject: [PATCH] [skip ci] Add gt of query func on benchmark (#12597) Signed-off-by: wangting0128 --- tests/benchmark/milvus_benchmark/client.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/benchmark/milvus_benchmark/client.py b/tests/benchmark/milvus_benchmark/client.py index 327015f1ab..ae954c965d 100644 --- a/tests/benchmark/milvus_benchmark/client.py +++ b/tests/benchmark/milvus_benchmark/client.py @@ -299,12 +299,16 @@ class MilvusClient(object): return self._milvus.drop_index(self._collection_name, field_name) @time_wrapper - def query(self, vector_query, filter_query=None, collection_name=None, timeout=300): + def query(self, vector_query, filter_query=None, collection_name=None, guarantee_timestamp=None, timeout=300): """ This method corresponds to the search method of milvus """ tmp_collection_name = self._collection_name if collection_name is None else collection_name params = util.search_param_analysis(vector_query, filter_query) params.update({"timeout": timeout}) + + if guarantee_timestamp is not None: + params.update({"guarantee_timestamp": guarantee_timestamp}) + logger.debug("Params of search : %s" % str(params)) result = self._milvus.search(tmp_collection_name, **params)