From 334d186f5c79e71e39b8ddc3a8b845c335c74e2a Mon Sep 17 00:00:00 2001 From: TheMimikyu Date: Mon, 23 Dec 2024 08:52:54 +0000 Subject: [PATCH] fix: add `nvidia/llama-3.2-nv-rerankqa-1b-v2` endpoint to `_MODEL_ENDPOINT_MAP` --- .../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 66203a490..e078758e5 100644 --- 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" }