From 42d85b3a2936dc1275558d99907b543a24aa595b Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Wed, 5 Jun 2024 07:42:44 -0400 Subject: [PATCH] align tests and docs on NVIDIA_API_KEY (instead of NVIDIA_CATALOG_API_KEY) (#731) --- integrations/nvidia/tests/test_document_embedder.py | 6 +++--- integrations/nvidia/tests/test_generator.py | 6 +++--- integrations/nvidia/tests/test_text_embedder.py | 6 +++--- 3 files changed, 9 insertions(+), 9 deletions(-) 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()