File tree 1 file changed +42
-0
lines changed
1 file changed +42
-0
lines changed Original file line number Diff line number Diff line change @@ -1063,6 +1063,48 @@ def test_config_with_generative(
1063
1063
"reranker-cohere" : {},
1064
1064
},
1065
1065
),
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
+ ),
1066
1108
(
1067
1109
Configure .Reranker .transformers (),
1068
1110
{
You can’t perform that action at this time.
0 commit comments