Update _nim_backend.py (#744)

* Update _nim_backend.py * Update _nim_backend.py
deepset-ai · May 22, 2024 · 31e61b7 · 31e61b7
1 parent 7141c68
commit 31e61b7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/_nim_backend.py b/integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/_nim_backend.py
@@ -13,7 +13,7 @@ def __init__(
         self,
         model: str,
         api_url: str,
-        api_key: Optional[Secret] = None,
+        api_key: Optional[Secret] = Secret.from_env_var("NVIDIA_API_KEY"),
         model_kwargs: Optional[Dict[str, Any]] = None,
     ):
         headers = {

diff --git a/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/_nim_backend.py b/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/_nim_backend.py
@@ -13,7 +13,7 @@ def __init__(
         self,
         model: str,
         api_url: str,
-        api_key: Optional[Secret] = None,
+        api_key: Optional[Secret] = Secret.from_env_var("NVIDIA_API_KEY"),
         model_kwargs: Optional[Dict[str, Any]] = None,
     ):
         headers = {