We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6e23658 commit 90340c5Copy full SHA for 90340c5
tests/integration/test_rerankers.py
@@ -13,14 +13,15 @@
13
@pytest.fixture(
14
params=[
15
CohereReranker,
16
- VoyageAIReranker,
+ (VoyageAIReranker, "rerank-lite-1"),
17
+ (VoyageAIReranker, "rerank-2.5"),
18
]
19
)
20
def reranker(request):
21
if request.param == CohereReranker:
22
return request.param()
- elif request.param == VoyageAIReranker:
23
- return request.param(model="rerank-lite-1")
+ elif isinstance(request.param, tuple) and request.param[0] == VoyageAIReranker:
24
+ return request.param[0](model=request.param[1])
25
26
27
@pytest.fixture
0 commit comments