nvidia-haystack 0.1.2__tar.gz → 0.1.4__tar.gz

{nvidia_haystack-0.1.2 → nvidia_haystack-0.1.4}/CHANGELOG.md

@@ -1,11 +1,24 @@
 # Changelog
 
+## [integrations/nvidia-v0.1.3] - 2025-01-02
+
+### 🚀 Features
+
+- Improvements to NvidiaRanker and adding user input timeout (#1193)
+- Add model `nvidia/llama-3.2-nv-rerankqa-1b-v2` to `_MODEL_ENDPOINT_MAP` (#1260)
+
+### 🧹 Chores
+
+- Fix linting/isort (#1215)
+
+
 ## [integrations/nvidia-v0.1.1] - 2024-11-14
 
 ### 🐛 Bug Fixes
 
 - Fixes to NvidiaRanker (#1191)
 
+
 ## [integrations/nvidia-v0.1.0] - 2024-11-13
 
 ### 🚀 Features
@@ -31,16 +44,25 @@
 
 - Do not retry tests in `hatch run test` command (#954)
 
-### ⚙️ Miscellaneous Tasks
+### ⚙️ CI
 
 - Retry tests to reduce flakyness (#836)
+- Adopt uv as installer (#1142)
+
+### 🧹 Chores
+
 - Update ruff invocation to include check parameter (#853)
 - Update ruff linting scripts and settings (#1105)
-- Adopt uv as installer (#1142)
 
-### Docs
+### 🌀 Miscellaneous
 
+- Fix: make hosted nim default (#734)
+- Fix: align tests and docs on NVIDIA_API_KEY (instead of NVIDIA_CATALOG_API_KEY) (#731)
+- Ci: install `pytest-rerunfailures` where needed; add retry config to `test-cov` script (#845)
+- Raise warning for base_url ../embeddings .../completions .../rankings (#922)
 - Update NvidiaGenerator docstrings (#966)
+-  Add default model for NVIDIA HayStack local NIM endpoints (#915)
+- Feat: add nvidia/llama-3.2-nv-rerankqa-1b-v1 to set of known ranking models (#1183)
 
 ## [integrations/nvidia-v0.0.3] - 2024-05-22
 
@@ -48,16 +70,30 @@
 
 - Update docstrings of Nvidia integrations (#599)
 
-### ⚙️ Miscellaneous Tasks
+### ⚙️ CI
 
 - Add generate docs to Nvidia workflow (#603)
 
+### 🌀 Miscellaneous
+
+- Remove references to Python 3.7 (#601)
+- Chore: add license classifiers (#680)
+- Chore: change the pydoc renderer class (#718)
+- Update Nvidia integration to support new endpoints (#701)
+- Docs: add missing api references (#728)
+- Update _nim_backend.py (#744)
+
 ## [integrations/nvidia-v0.0.2] - 2024-03-18
 
 ### 📚 Documentation
 
 - Disable-class-def (#556)
 
+### 🌀 Miscellaneous
+
+- Make tests show coverage (#566)
+- Add NIM backend support (#597)
+
 ## [integrations/nvidia-v0.0.1] - 2024-03-07
 
 ### 🚀 Features
@@ -68,6 +104,15 @@
 
 - `nvidia-haystack`- Handle non-strict env var secrets correctly (#543)
 
+### 🌀 Miscellaneous
+
+- Add `NvidiaGenerator` (#557)
+- Add missing import in NvidiaGenerator docstring (#559)
+
 ## [integrations/nvidia-v0.0.0] - 2024-03-01
 
+### 🌀 Miscellaneous
+
+- Add Nvidia integration scaffold (#515)
+
 <!-- generated by git-cliff -->

{nvidia_haystack-0.1.2 → nvidia_haystack-0.1.4}/PKG-INFO

@@ -1,11 +1,12 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: nvidia-haystack
-Version: 0.1.2
+Version: 0.1.4
 Project-URL: Documentation, https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/nvidia#readme
 Project-URL: Issues, https://github.com/deepset-ai/haystack-core-integrations/issues
 Project-URL: Source, https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/nvidia
 Author-email: deepset GmbH <info@deepset.ai>
-License: Apache-2.0
+License-Expression: Apache-2.0
+License-File: LICENSE.txt
 Classifier: Development Status :: 4 - Beta
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python

{nvidia_haystack-0.1.2 → nvidia_haystack-0.1.4}/src/haystack_integrations/components/rankers/nvidia/ranker.py

@@ -17,8 +17,10 @@ logger = logging.getLogger(__name__)
 _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",
 }