Skip to content

Commit d570dc9

Browse files
committed
Post pre-commit :D
1 parent c13c589 commit d570dc9

File tree

1 file changed

+42
-0
lines changed

1 file changed

+42
-0
lines changed

test/collection/test_config.py

+42
Original file line numberDiff line numberDiff line change
@@ -1063,6 +1063,48 @@ def test_config_with_generative(
10631063
"reranker-cohere": {},
10641064
},
10651065
),
1066+
(
1067+
Configure.Reranker.voyageai(),
1068+
{
1069+
"reranker-voyageai": {},
1070+
},
1071+
),
1072+
(
1073+
Configure.Reranker.voyageai(model="rerank-lite-1"),
1074+
{
1075+
"reranker-voyageai": {"model": "rerank-lite-1"},
1076+
},
1077+
),
1078+
(
1079+
Configure.Reranker.jinaai(),
1080+
{
1081+
"reranker-jinaai": {},
1082+
},
1083+
),
1084+
(
1085+
Configure.Reranker.jinaai(model="jina-reranker-v2-base-multilingual"),
1086+
{
1087+
"reranker-jinaai": {"model": "jina-reranker-v2-base-multilingual"},
1088+
},
1089+
),
1090+
(
1091+
Configure.Reranker.nvidia(),
1092+
{
1093+
"reranker-nvidia": {},
1094+
},
1095+
),
1096+
(
1097+
Configure.Reranker.nvidia(
1098+
model="nvidia/llama-3.2-nv-rerankqa-1b-v2",
1099+
base_url="https://integrate.api.nvidia.com/v1",
1100+
),
1101+
{
1102+
"reranker-nvidia": {
1103+
"model": "nvidia/llama-3.2-nv-rerankqa-1b-v2",
1104+
"baseURL": "https://integrate.api.nvidia.com/v1",
1105+
},
1106+
},
1107+
),
10661108
(
10671109
Configure.Reranker.transformers(),
10681110
{

0 commit comments

Comments
 (0)