Update crates/rmcp/src/transport/common/client_side_sse.rs

Co-authored-by: Dale Seo <5466341+DaleSeo@users.noreply.github.com>

Author: alexhancock

crates/rmcp/src/transport/common/client_side_sse.rs

@@ -258,19 +258,12 @@ where
                         // Per SEP-1699, a graceful stream close is
                         // reconnectable.  If the server sent a `retry` field
                         // we MUST wait that long before reconnecting.
-                        if let Some(interval) = this.server_retry_interval.take() {
+                        let interval = this.server_retry_interval.take()
+                            .or_else(|| this.retry_policy.retry(0));
+                        if let Some(interval) = interval {
                             tracing::debug!(
                                 ?interval,
-                                "sse stream ended gracefully, waiting server retry interval before reconnecting"
-                            );
-                            SseAutoReconnectStreamState::WaitingNextRetry {
-                                sleep: tokio::time::sleep(interval),
-                                retry_times: 0,
-                            }
-                        } else if let Some(interval) = this.retry_policy.retry(0) {
-                            tracing::debug!(
-                                ?interval,
-                                "sse stream ended gracefully, reconnecting with policy interval"
+                                "sse stream ended gracefully, reconnecting"
                             );
                             SseAutoReconnectStreamState::WaitingNextRetry {
                                 sleep: tokio::time::sleep(interval),