Auto add ssh missing host keys

viaacode · Dec 12, 2024 · cb14cb4 · cb14cb4
1 parent cc92dfa
commit cb14cb4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/prefect_meemoo/ssh/credentials.py b/prefect_meemoo/ssh/credentials.py
@@ -5,7 +5,7 @@
 from prefect import get_run_logger
 from pydantic import SecretStr
 from pydantic import Field
-from paramiko import SSHClient
+from paramiko import SSHClient, AutoAddPolicy
 
 
 class SSHCredentials(Block):
@@ -56,7 +56,7 @@ def get_client(self) -> SSHClient:
         """
 
         client = SSHClient()
-        client.load_system_host_keys()
+        client.set_missing_host_key_policy(AutoAddPolicy())
         try:
             client.connect(
                 hostname=self.hostname,