From f3c7cd4148a67f4b6b89d9d0eb7cb56cbd3eff82 Mon Sep 17 00:00:00 2001 From: Daniel Kleine <53251018+d-kleine@users.noreply.github.com> Date: Wed, 8 Jan 2025 11:55:21 +0100 Subject: [PATCH 1/2] added mistral-4b v1 reranker (#1278) --- .../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 ec1af7ab8..67215d753 100644 --- a/integrations/nvidia/src/haystack_integrations/components/rankers/nvidia/ranker.py +++ b/integrations/nvidia/src/haystack_integrations/components/rankers/nvidia/ranker.py @@ -17,6 +17,7 @@ _DEFAULT_MODEL = "nvidia/nv-rerankqa-mistral-4b-v3" _MODEL_ENDPOINT_MAP = { + "nv-rerank-qa-mistral-4b:1": "https://ai.api.nvidia.com/v1/retrieval/nvidia/reranking", "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", From f5c3aee5478b310d9fee785ba2ade196753fc78a Mon Sep 17 00:00:00 2001 From: HaystackBot Date: Wed, 8 Jan 2025 10:57:56 +0000 Subject: [PATCH 2/2] Update the changelog --- integrations/nvidia/CHANGELOG.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/integrations/nvidia/CHANGELOG.md b/integrations/nvidia/CHANGELOG.md index 2523907d6..a3de61119 100644 --- a/integrations/nvidia/CHANGELOG.md +++ b/integrations/nvidia/CHANGELOG.md @@ -1,5 +1,11 @@ # Changelog +## [integrations/nvidia-v0.1.4] - 2025-01-08 + +### 🌀 Miscellaneous + +- Feat: add nv-rerank-qa-mistral-4b:1 reranker (#1278) + ## [integrations/nvidia-v0.1.3] - 2025-01-02 ### 🚀 Features