diff --git a/integrations/nvidia/tests/test_document_embedder.py b/integrations/nvidia/tests/test_document_embedder.py index 28858dcf7..ed31df65e 100644 --- a/integrations/nvidia/tests/test_document_embedder.py +++ b/integrations/nvidia/tests/test_document_embedder.py @@ -388,15 +388,15 @@ def test_run_integration_with_nim_backend(self): assert isinstance(doc.embedding[0], float) @pytest.mark.skipif( - not os.environ.get("NVIDIA_CATALOG_API_KEY", None), - reason="Export an env var called NVIDIA_CATALOG_API_KEY containing the Nvidia API key to run this test.", + not os.environ.get("NVIDIA_API_KEY", None), + reason="Export an env var called NVIDIA_API_KEY containing the NVIDIA API key to run this test.", ) @pytest.mark.integration def test_run_integration_with_api_catalog(self): embedder = NvidiaDocumentEmbedder( model="NV-Embed-QA", api_url="https://ai.api.nvidia.com/v1/retrieval/nvidia", - api_key=Secret.from_env_var("NVIDIA_CATALOG_API_KEY"), + api_key=Secret.from_env_var("NVIDIA_API_KEY"), ) embedder.warm_up() diff --git a/integrations/nvidia/tests/test_generator.py b/integrations/nvidia/tests/test_generator.py index a30ccc9cc..882bcdc88 100644 --- a/integrations/nvidia/tests/test_generator.py +++ b/integrations/nvidia/tests/test_generator.py @@ -208,15 +208,15 @@ def test_run_integration_with_nim_backend(self): assert result["meta"] @pytest.mark.skipif( - not os.environ.get("NVIDIA_CATALOG_API_KEY", None), - reason="Export an env var called NVIDIA_CATALOG_API_KEY containing the Nvidia API key to run this test.", + not os.environ.get("NVIDIA_API_KEY", None), + reason="Export an env var called NVIDIA_API_KEY containing the NVIDIA API key to run this test.", ) @pytest.mark.integration def test_run_integration_with_api_catalog(self): generator = NvidiaGenerator( model="meta/llama3-70b-instruct", api_url="https://integrate.api.nvidia.com/v1", - api_key=Secret.from_env_var("NVIDIA_CATALOG_API_KEY"), + api_key=Secret.from_env_var("NVIDIA_API_KEY"), model_arguments={ "temperature": 0.2, }, diff --git a/integrations/nvidia/tests/test_text_embedder.py b/integrations/nvidia/tests/test_text_embedder.py index b85ac39bf..206f79c10 100644 --- a/integrations/nvidia/tests/test_text_embedder.py +++ b/integrations/nvidia/tests/test_text_embedder.py @@ -175,15 +175,15 @@ def test_run_integration_with_nim_backend(self): assert "usage" in meta @pytest.mark.skipif( - not os.environ.get("NVIDIA_CATALOG_API_KEY", None), - reason="Export an env var called NVIDIA_CATALOG_API_KEY containing the Nvidia API key to run this test.", + not os.environ.get("NVIDIA_API_KEY", None), + reason="Export an env var called NVIDIA_API_KEY containing the NVIDIA API key to run this test.", ) @pytest.mark.integration def test_run_integration_with_api_catalog(self): embedder = NvidiaTextEmbedder( model="NV-Embed-QA", api_url="https://ai.api.nvidia.com/v1/retrieval/nvidia", - api_key=Secret.from_env_var("NVIDIA_CATALOG_API_KEY"), + api_key=Secret.from_env_var("NVIDIA_API_KEY"), ) embedder.warm_up()