diff --git a/mcp_ssh_session/session_manager.py b/mcp_ssh_session/session_manager.py index cef57b2..d9e3a9a 100644 --- a/mcp_ssh_session/session_manager.py +++ b/mcp_ssh_session/session_manager.py @@ -12,6 +12,12 @@ import paramiko +try: + from paramiko import NoValidConnectionsError +except ImportError: + # paramiko 4.0.0+ moved NoValidConnectionsError to ssh_exception + from paramiko.ssh_exception import NoValidConnectionsError + from .command_executor import CommandExecutor from .datastructures import CommandStatus from .file_manager import FileManager @@ -187,7 +193,7 @@ def get_or_create_session( except ( paramiko.AuthenticationException, paramiko.SSHException, - paramiko.NoValidConnectionsError, + NoValidConnectionsError, OSError, TimeoutError, ) as e: