diff --git a/src/websocket_client.rs b/src/websocket_client.rs index b2d5e8f..5890143 100644 --- a/src/websocket_client.rs +++ b/src/websocket_client.rs @@ -50,8 +50,14 @@ impl AgentClient { /// ``` pub async fn connect(url: &str, token: &str) -> Result { let ws_url = format!("{}/socket/agent?token={}", url, token); + log::info!("Connecting to WebSocket: {}", ws_url); + let (ws_stream, _) = connect_async(&ws_url) .await + .map_err(|e| { + log::error!("WebSocket connection failed: {}", e); + e + }) .context("Failed to connect to WebSocket")?; log::info!("Connected to Towerops server at {}", url);