align tests and docs on NVIDIA_API_KEY (instead of NVIDIA_CATALOG_API…

…_KEY) (#731)
deepset-ai · Jun 5, 2024 · 42d85b3 · 42d85b3
1 parent e118b92
commit 42d85b3
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git 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
@@ -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
@@ -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()