feat: add model nvidia/llama-3.2-nv-rerankqa-1b-v2 to `_MODEL_ENDPO…

…INT_MAP` (#1260) * fix: add `nvidia/llama-3.2-nv-rerankqa-1b-v2` endpoint to `_MODEL_ENDPOINT_MAP` * chore: lint
deepset-ai · Jan 2, 2025 · bbf7417 · bbf7417
1 parent 6b8d02d
commit bbf7417
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/integrations/nvidia/src/haystack_integrations/components/rankers/nvidia/ranker.py b/integrations/nvidia/src/haystack_integrations/components/rankers/nvidia/ranker.py
@@ -19,6 +19,7 @@
 _MODEL_ENDPOINT_MAP = {
     "nvidia/nv-rerankqa-mistral-4b-v3": "https://ai.api.nvidia.com/v1/retrieval/nvidia/nv-rerankqa-mistral-4b-v3/reranking",
     "nvidia/llama-3.2-nv-rerankqa-1b-v1": "https://ai.api.nvidia.com/v1/retrieval/nvidia/llama-3_2-nv-rerankqa-1b-v1/reranking",
+    "nvidia/llama-3.2-nv-rerankqa-1b-v2": "https://ai.api.nvidia.com/v1/retrieval/nvidia/llama-3_2-nv-rerankqa-1b-v2/reranking",
 }