Merge branch 'deepset-ai:main' into main

deepset-ai · Jan 9, 2025 · a5bc364 · a5bc364
2 parents 5804c5f + f5c3aee
commit a5bc364
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git 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

diff --git 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",