@agent-relay/dashboard-server 2.0.34 → 2.0.36

package/dist/server.js

@@ -836,8 +836,14 @@ export async function startDashboard(portOrOptions, dataDirArg, teamDirArg, dbPa
             };
             client.onStateChange = (state) => {
                 console.log(`[dashboard] Relay client for ${senderName} state: ${state}`);
-                // Clean up disconnected clients
-                if (state === 'DISCONNECTED') {
+                // Handle state transitions
+                if (state === 'READY') {
+                    // Re-add to map on successful reconnection
+                    relayClients.set(senderName, client);
+                }
+                else if (state === 'DISCONNECTED') {
+                    // Only remove if max reconnect attempts exhausted (client gives up)
+                    // The client will auto-reconnect if configured, so don't remove prematurely
                     relayClients.delete(senderName);
                 }
             };
@@ -1980,10 +1986,12 @@ export async function startDashboard(portOrOptions, dataDirArg, teamDirArg, dbPa
                 return false;
             }
             // If agent is daemon-connected but not yet in spawner's activeWorkers,
-            // poll briefly to handle race condition between spawn API returning and
+            // poll to handle race condition between spawn API returning and
             // WebSocket connection. This is common for setup agents (__setup__*).
+            // Longer timeout for CLI auth flows (Cursor, etc.) which can take time to initialize.
             if (!isSpawned && isDaemon && spawner) {
-                const maxWaitMs = 3000; // Wait up to 3 seconds
+                const isSetupAgent = agentName.startsWith('__setup__');
+                const maxWaitMs = isSetupAgent ? 90000 : 5000; // 90s for setup agents (CLI auth can be slow), 5s otherwise
                 const pollIntervalMs = 100;
                 const startTime = Date.now();
                 while (Date.now() - startTime < maxWaitMs) {
@@ -3233,8 +3241,8 @@ export async function startDashboard(portOrOptions, dataDirArg, teamDirArg, dbPa
         const uptime = process.uptime();
         const memUsage = process.memoryUsage();
         const socketExists = fs.existsSync(socketPath);
-        // Check relay client connectivity (check if default Dashboard client is connected)
-        const defaultClient = relayClients.get('Dashboard');
+        // Check relay client connectivity (check if default _DashboardUI client is connected)
+        const defaultClient = relayClients.get('_DashboardUI');
         const relayConnected = defaultClient?.state === 'READY';
         // If socket doesn't exist, daemon may not be running properly
         if (!socketExists) {
@@ -3260,7 +3268,7 @@ export async function startDashboard(portOrOptions, dataDirArg, teamDirArg, dbPa
         const uptime = process.uptime();
         const memUsage = process.memoryUsage();
         const socketExists = fs.existsSync(socketPath);
-        const defaultClient = relayClients.get('Dashboard');
+        const defaultClient = relayClients.get('_DashboardUI');
         const relayConnected = defaultClient?.state === 'READY';
         if (!socketExists) {
             return res.status(503).json({