Skip to content

Commit

Permalink
Formatting.
Browse files Browse the repository at this point in the history
  • Loading branch information
Adversarian committed Dec 21, 2024
1 parent 643cedc commit 39e5bad
Showing 1 changed file with 5 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -61,8 +61,11 @@ def test_rerank_construction_with_no_optional_kwargs():
assert reranker.top_n is None
assert reranker.model == "rerank-2"


def test_rerank_construction_with_optional_kwargs():
reranker = VoyageAIRerank(model="rerank-2", api_key="mock_api_key", top_n=10, truncation=False)
reranker = VoyageAIRerank(
model="rerank-2", api_key="mock_api_key", top_n=10, truncation=False
)
assert not reranker.truncation
assert reranker.top_n == 10
assert reranker.model == "rerank-2"
assert reranker.model == "rerank-2"

0 comments on commit 39e5bad

Please sign in to comment.