Update _nim_backend.py (#744)

TuanaCelik · web-flow · commit 31e61b72571d · 2024-05-22T12:30:26.000+02:00
* Update _nim_backend.py

* Update _nim_backend.py
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 = {
