From 8d0d58a73dda776cdc0924f2604dc26d0458917f Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Tue, 12 Nov 2024 13:49:40 -0500 Subject: [PATCH] add nvidia/llama-3.2-nv-rerankqa-1b-v1 to set of known ranking models --- .../haystack_integrations/components/rankers/nvidia/ranker.py | 1 + 1 file changed, 1 insertion(+) diff --git a/integrations/nvidia/src/haystack_integrations/components/rankers/nvidia/ranker.py b/integrations/nvidia/src/haystack_integrations/components/rankers/nvidia/ranker.py index 46c736883..1553d1ac3 100644 --- a/integrations/nvidia/src/haystack_integrations/components/rankers/nvidia/ranker.py +++ b/integrations/nvidia/src/haystack_integrations/components/rankers/nvidia/ranker.py @@ -12,6 +12,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", }