milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[skip ci] Add gt of query func on benchmark (#12597) 

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
pull/12601/head
wt 2021-12-02 13:13:32 +08:00 committed by GitHub
parent e8442247f0
commit 487bf9483e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/benchmark/milvus_benchmark/client.py
View File

@ -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)