Skip to content

Commit 90340c5

Browse files
committed
VoyageAI refactoring:
- contextual model - removing the model default value - token counting, ie. more effective use of batches
1 parent 6e23658 commit 90340c5

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/integration/test_rerankers.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,14 +13,15 @@
1313
@pytest.fixture(
1414
params=[
1515
CohereReranker,
16-
VoyageAIReranker,
16+
(VoyageAIReranker, "rerank-lite-1"),
17+
(VoyageAIReranker, "rerank-2.5"),
1718
]
1819
)
1920
def reranker(request):
2021
if request.param == CohereReranker:
2122
return request.param()
22-
elif request.param == VoyageAIReranker:
23-
return request.param(model="rerank-lite-1")
23+
elif isinstance(request.param, tuple) and request.param[0] == VoyageAIReranker:
24+
return request.param[0](model=request.param[1])
2425

2526

2627
@pytest.fixture

0 commit comments

Comments
 (0)