admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: fix connection errror of BulkInsert nightly (#42796)

Browse Source 
pr: #42805

Signed-off-by: nico <cheng.yuan@zilliz.com>
This commit is contained in:
nico 2025-06-17 15:28:39 +08:00 committed by GitHub
parent 98fe28d0ef
commit f19349e514
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/python_client/base/client_base.py
View File

@ -150,10 +150,13 @@ class TestcaseBase(Base):
uri = cf.param_info.param_uri
else:
uri = "http://" + cf.param_info.param_host + ":" + str(cf.param_info.param_port)
self.connection_wrap.connect(alias=DefaultConfig.DEFAULT_USING,uri=uri,token=cf.param_info.param_token)
res, is_succ = self.connection_wrap.MilvusClient(uri=uri,
token=cf.param_info.param_token)
self.client = MilvusClient(uri=uri, token=cf.param_info.param_token)
if cf.param_info.param_token:
token = cf.param_info.param_token
else:
token = f"{ct.default_user}:{ct.default_password}"
self.connection_wrap.connect(alias=DefaultConfig.DEFAULT_USING, uri=uri, token=token)
res, is_succ = self.connection_wrap.MilvusClient(uri=uri, token=token)
self.client = MilvusClient(uri=uri, token=token)
else:
if cf.param_info.param_user and cf.param_info.param_password:
res, is_succ = self.connection_wrap.connect(alias=DefaultConfig.DEFAULT_USING,
@ -168,27 +171,29 @@ class TestcaseBase(Base):
port=cf.param_info.param_port)
uri = "http://" + cf.param_info.param_host + ":" + str(cf.param_info.param_port)
self.client = MilvusClient(uri=uri, token=cf.param_info.param_token)
if cf.param_info.param_token:
token = cf.param_info.param_token
else:
token = f"{ct.default_user}:{ct.default_password}"
self.client = MilvusClient(uri=uri, token=token)
server_version = utility.get_server_version()
log.info(f"server version: {server_version}")
return res
def get_tokens_by_analyzer(self, text, analyzer_params):
if cf.param_info.param_uri:
uri = cf.param_info.param_uri
else:
uri = "http://" + cf.param_info.param_host + ":" + str(cf.param_info.param_port)
client = MilvusClient(
uri = uri,
if cf.param_info.param_token:
token = cf.param_info.param_token
)
else:
token = f"{ct.default_user}:{ct.default_password}"
self.client = MilvusClient(uri=uri, token=token)
res = client.run_analyzer(text, analyzer_params, with_detail=True, with_hash=True)
tokens = [r['token'] for r in res.tokens]
return tokens
# def init_async_milvus_client(self):
# uri = cf.param_info.param_uri or f"http://{cf.param_info.param_host}:{cf.param_info.param_port}"
# kwargs = {