nvidia-haystack 1.1.1__tar.gz → 1.2.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (39) hide show
  1. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/CHANGELOG.md +12 -0
  2. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/PKG-INFO +1 -1
  3. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/generators/nvidia/generator.py +6 -0
  4. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/test_base_url.py +2 -1
  5. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/test_generator.py +5 -5
  6. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/.gitignore +0 -0
  7. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/LICENSE.txt +0 -0
  8. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/README.md +0 -0
  9. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/examples/chat_generator_with_structured_outputs.py +0 -0
  10. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/pydoc/config_docusaurus.yml +0 -0
  11. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/pyproject.toml +0 -0
  12. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/embedders/nvidia/__init__.py +0 -0
  13. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/embedders/nvidia/document_embedder.py +0 -0
  14. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/embedders/nvidia/text_embedder.py +0 -0
  15. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/embedders/nvidia/truncate.py +0 -0
  16. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/embedders/py.typed +0 -0
  17. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/generators/nvidia/__init__.py +0 -0
  18. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/generators/nvidia/chat/__init__.py +0 -0
  19. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/generators/nvidia/chat/chat_generator.py +0 -0
  20. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/generators/py.typed +0 -0
  21. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/rankers/nvidia/__init__.py +0 -0
  22. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/rankers/nvidia/py.typed +0 -0
  23. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/rankers/nvidia/ranker.py +0 -0
  24. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/components/rankers/nvidia/truncate.py +0 -0
  25. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/utils/nvidia/__init__.py +0 -0
  26. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/utils/nvidia/client.py +0 -0
  27. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/utils/nvidia/models.py +0 -0
  28. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/utils/nvidia/nim_backend.py +0 -0
  29. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/utils/nvidia/utils.py +0 -0
  30. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/src/haystack_integrations/utils/py.typed +0 -0
  31. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/__init__.py +0 -0
  32. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/conftest.py +0 -0
  33. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/test_document_embedder.py +0 -0
  34. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/test_embedding_truncate_mode.py +0 -0
  35. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/test_nim_backend.py +0 -0
  36. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/test_nvidia_chat_generator.py +0 -0
  37. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/test_ranker.py +0 -0
  38. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/test_text_embedder.py +0 -0
  39. {nvidia_haystack-1.1.1 → nvidia_haystack-1.2.0}/tests/test_utils.py +0 -0
@@ -1,5 +1,17 @@
1
1
  # Changelog
2
2
 
3
+ ## [integrations/nvidia-v1.1.1] - 2026-04-14
4
+
5
+ ### 🐛 Bug Fixes
6
+
7
+ - Replace in-place dataclass mutations with dataclasses.replace() (#3112)
8
+
9
+ ### 🧪 Testing
10
+
11
+ - Track test coverage for all integrations (#3065)
12
+ - Add unit tests for Nvidia integration (#3162)
13
+
14
+
3
15
  ## [integrations/nvidia-v1.1.0] - 2026-03-30
4
16
 
5
17
  ### 🐛 Bug Fixes
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nvidia-haystack
3
- Version: 1.1.1
3
+ Version: 1.2.0
4
4
  Project-URL: Documentation, https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/nvidia#readme
5
5
  Project-URL: Issues, https://github.com/deepset-ai/haystack-core-integrations/issues
6
6
  Project-URL: Source, https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/nvidia
@@ -75,6 +75,12 @@ class NvidiaGenerator:
75
75
  Timeout for request calls, if not set it is inferred from the `NVIDIA_TIMEOUT` environment variable
76
76
  or set to 60 by default.
77
77
  """
78
+ warnings.warn(
79
+ "The `NvidiaGenerator` component is deprecated and will be removed in a future version. "
80
+ "Use `NvidiaChatGenerator` instead, which now also supports string inputs.",
81
+ FutureWarning,
82
+ stacklevel=2,
83
+ )
78
84
  self._model = model
79
85
  self.api_url = url_validation(api_url)
80
86
  self._api_key = api_key
@@ -28,7 +28,8 @@ from haystack_integrations.components.rankers.nvidia import NvidiaRanker
28
28
  def test_base_url_invalid_not_hosted(base_url: str, component) -> None:
29
29
  with pytest.warns(UserWarning) as msg:
30
30
  component(api_url=base_url, model="x")
31
- assert "you may have inference and listing issues" in str(msg[0].message)
31
+ user_warnings = [w for w in msg if issubclass(w.category, UserWarning)]
32
+ assert any("you may have inference and listing issues" in str(w.message) for w in user_warnings)
32
33
 
33
34
 
34
35
  @pytest.mark.parametrize(
@@ -193,7 +193,7 @@ class TestNvidiaGenerator:
193
193
  @pytest.mark.integration
194
194
  def test_run_integration_with_api_catalog(self):
195
195
  generator = NvidiaGenerator(
196
- model="meta/llama3-8b-instruct",
196
+ model="meta/llama-3.1-8b-instruct",
197
197
  api_url="https://integrate.api.nvidia.com/v1",
198
198
  api_key=Secret.from_env_var("NVIDIA_API_KEY"),
199
199
  model_arguments={
@@ -256,7 +256,7 @@ class TestNvidiaGenerator:
256
256
 
257
257
  def test_warm_up_is_idempotent(self, monkeypatch):
258
258
  monkeypatch.setenv("NVIDIA_API_KEY", "fake-api-key")
259
- generator = NvidiaGenerator("meta/llama3-8b-instruct")
259
+ generator = NvidiaGenerator("meta/llama-3.1-8b-instruct")
260
260
  generator.warm_up()
261
261
  backend = generator.backend
262
262
  generator.warm_up()
@@ -264,7 +264,7 @@ class TestNvidiaGenerator:
264
264
 
265
265
  def test_available_models_without_backend(self, monkeypatch):
266
266
  monkeypatch.setenv("NVIDIA_API_KEY", "fake-api-key")
267
- generator = NvidiaGenerator("meta/llama3-8b-instruct")
267
+ generator = NvidiaGenerator("meta/llama-3.1-8b-instruct")
268
268
  assert generator.available_models == []
269
269
 
270
270
  @pytest.mark.usefixtures("mock_local_models")
@@ -283,12 +283,12 @@ class TestNvidiaGenerator:
283
283
  "init_parameters": {
284
284
  "api_key": {"env_vars": ["NVIDIA_API_KEY"], "strict": True, "type": "env_var"},
285
285
  "api_url": "https://my.url.com/v1",
286
- "model": "meta/llama3-8b-instruct",
286
+ "model": "meta/llama-3.1-8b-instruct",
287
287
  "model_arguments": {"temperature": 0.5},
288
288
  },
289
289
  }
290
290
  generator = NvidiaGenerator.from_dict(data)
291
- assert generator._model == "meta/llama3-8b-instruct"
291
+ assert generator._model == "meta/llama-3.1-8b-instruct"
292
292
  assert generator.api_url == "https://my.url.com/v1"
293
293
  assert generator._model_arguments == {"temperature": 0.5}
294
294