redo

cmd/runner/main.go

@@ -81,11 +81,18 @@ func main() {
 		}
 	}
 
-	// Start heartbeat loop
+	// Start WebSocket connection with reconnection
+	go client.ConnectWebSocketWithReconnect()
+
+	// Start heartbeat loop (for WebSocket ping/pong and HTTP fallback)
 	go client.HeartbeatLoop()
 
-	// Start task processing loop
-	client.ProcessTasks()
+	// ProcessTasks is now handled via WebSocket, but kept for HTTP fallback
+	// WebSocket will handle task assignment automatically
+	log.Printf("Runner started, connecting to manager via WebSocket...")
+	
+	// Block forever
+	select {}
 }
 
 func loadSecrets(path string) (*SecretsFile, error) {