test: add model rerank testcases (#42657)

/kind improvement

---------

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
This commit is contained in:
zhuwenxing 2025-06-27 10:48:43 +08:00 committed by GitHub
parent 8bda337f52
commit 76004eb6cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1121 additions and 2 deletions

View File

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