mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 19:31:51 +08:00
test: add model rerank testcases (#42657)
/kind improvement --------- Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
This commit is contained in:
parent
8bda337f52
commit
76004eb6cf
@ -52,8 +52,10 @@ def pytest_addoption(parser):
|
||||
parser.addoption('--token', action='store', default="root:Milvus", help="token for milvus client")
|
||||
parser.addoption("--request_duration", action="store", default="10m", help="request_duration")
|
||||
# a tei endpoint for text embedding, default is http://text-embeddings-service.milvus-ci.svc.cluster.local:80 which is deployed in house
|
||||
parser.addoption("--tei_endpoint", action="store", default="http://text-embeddings-service.milvus-ci.svc.cluster.local:80", help="tei endpoint")
|
||||
parser.addoption("--tei_endpoint", action="store", default="http://text-embeddings-service.milvus-ci.svc.cluster.local:80", help="tei embedding endpoint")
|
||||
|
||||
parser.addoption("--tei_reranker_endpoint", action="store", default="http://text-rerank-service.milvus-ci.svc.cluster.local:80", help="tei rerank endpoint")
|
||||
parser.addoption("--vllm_reranker_endpoint", action="store", default="http://vllm-rerank-service.milvus-ci.svc.cluster.local:80", help="vllm rerank endpoint")
|
||||
|
||||
@pytest.fixture
|
||||
def host(request):
|
||||
@ -214,6 +216,14 @@ def request_duration(request):
|
||||
def tei_endpoint(request):
|
||||
return request.config.getoption("--tei_endpoint")
|
||||
|
||||
@pytest.fixture
|
||||
def tei_reranker_endpoint(request):
|
||||
return request.config.getoption("--tei_reranker_endpoint")
|
||||
|
||||
@pytest.fixture
|
||||
def vllm_reranker_endpoint(request):
|
||||
return request.config.getoption("--vllm_reranker_endpoint")
|
||||
|
||||
""" fixture func """
|
||||
|
||||
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
Loading…
x
Reference in New Issue
Block a user