Address feedbacks

Author: sfc-gh-zzhu

src/snowflake/cli/_plugins/remote/constants.py

@@ -99,7 +99,7 @@ class ServiceResult(enum.Enum):
 RAY_DASHBOARD_ENDPOINT_NAME = "ray-dashboard"
 
 # SSH Configuration
-DEFAULT_SSH_REFRESH_INTERVAL = 300  # 5 minutes
+DEFAULT_SSH_REFRESH_INTERVAL = 2700  # 45 minutes
 SSH_RETRY_INTERVAL = 30  # 30 seconds
 SSH_COUNTDOWN_INTERVAL = 30  # Show countdown every 30 seconds
 SSH_DIR_NAME = ".ssh"

src/snowflake/cli/_plugins/remote/manager.py

@@ -877,7 +877,7 @@ def _ssh_token_refresh_loop(
             raise CliError(f"SSH error occurred: {str(e)}")
 
     def _get_fresh_token(self) -> str:
-        """Get a fresh authentication token with natural session expiration.
+        """Get a fresh authentication token with session keep-alive enabled.
 
         Creates a connection that will expire naturally according to Snowflake's
         default session timeout, allowing proper token lifecycle management
@@ -890,12 +890,12 @@ def _get_fresh_token(self) -> str:
 
         current_context = get_cli_context().connection_context
 
-        # Create connection with natural session expiration for SSH token refresh
+        # Create connection with session keep-alive enabled for SSH token refresh
         fresh_connection = connect_to_snowflake(
             connection_name=current_context.connection_name,
             temporary_connection=current_context.temporary_connection,
-            # Allow session to expire naturally - don't keep it alive artificially
-            using_session_keep_alive=False,
+            # Enable session keep-alive to prevent session expiry
+            using_session_keep_alive=True,
         )
 
         # Track the connection and proactively cap the number of live temp connections

tests/remote/test_manager.py

@@ -629,12 +629,11 @@ def test_get_fresh_token_success(self, mock_get_context, mock_connect):
         mock_connect.assert_called_once_with(
             connection_name="test_conn",
             temporary_connection=False,
-            using_session_keep_alive=False,
+            using_session_keep_alive=True,
         )
         mock_cursor.execute.assert_called_once_with(
             "ALTER SESSION SET python_connector_query_result_format = 'JSON'"
         )
-        # Connection is designed to expire naturally, no explicit close
 
     @patch("snowflake.cli._app.snow_connector.connect_to_snowflake")
     @patch("snowflake.cli._plugins.remote.manager.get_cli_context")