@kendoo.agentdesk/agentdesk 0.5.0 → 0.5.1

package/cli/team.mjs

@@ -174,48 +174,66 @@ export async function runTeam(taskId, opts = {}) {
     }
   }
 
-  try {
-    vizWs = new WebSocket(AGENTDESK_URL);
-    vizWs.on("open", () => {
-      // Authenticate via first message (proxy-safe — query params can be stripped)
-      if (apiKey) {
-        vizWs.send(JSON.stringify({ type: "auth", apiKey }));
-      }
-    });
-    vizWs.on("message", (raw) => {
-      let msg;
-      try { msg = JSON.parse(raw.toString()); } catch { return; }
-      if (msg.type === "auth:ok") {
-        vizConnected = true;
-        console.log("AgentDesk: connected");
-        vizSend({
-          type: "session:start",
-          taskId,
-          taskLink,
-          title: description || taskId,
-          project: project.name || null,
-          sessionNumber: 1,
-          agents: ["Jane", "Dennis", "Bart", "Vera", "Sam", "Luna", "Mark"],
-        });
-        while (vizQueue.length > 0 && vizWs.readyState === WebSocket.OPEN) {
-          vizWs.send(vizQueue.shift());
+  let sessionStartSent = false;
+  let reconnectTimer = null;
+
+  function connectWs() {
+    try {
+      vizWs = new WebSocket(AGENTDESK_URL);
+      vizWs.on("open", () => {
+        if (apiKey) {
+          vizWs.send(JSON.stringify({ type: "auth", apiKey }));
+        }
+      });
+      vizWs.on("message", (raw) => {
+        let msg;
+        try { msg = JSON.parse(raw.toString()); } catch { return; }
+        if (msg.type === "auth:ok") {
+          vizConnected = true;
+          if (!sessionStartSent) {
+            console.log("AgentDesk: connected");
+          } else {
+            console.log("AgentDesk: reconnected");
+          }
+          // Always re-send session:start on (re)connect so server knows about the session
+          vizSend({
+            type: "session:start",
+            taskId,
+            taskLink,
+            title: description || taskId,
+            project: project.name || null,
+            sessionNumber: 1,
+            agents: ["Jane", "Dennis", "Bart", "Vera", "Sam", "Luna", "Mark"],
+          });
+          sessionStartSent = true;
+          while (vizQueue.length > 0 && vizWs.readyState === WebSocket.OPEN) {
+            vizWs.send(vizQueue.shift());
+          }
+        } else if (msg.type === "auth:error") {
+          console.log("AgentDesk: authentication failed — run 'agentdesk login'");
+          vizConnected = false;
         }
-      } else if (msg.type === "auth:error") {
-        console.log("AgentDesk: authentication failed — run 'agentdesk login'");
+      });
+      vizWs.on("error", () => { vizConnected = false; });
+      vizWs.on("close", (code) => {
         vizConnected = false;
-      }
-    });
-    vizWs.on("error", () => { vizConnected = false; });
-    vizWs.on("close", (code) => {
-      vizConnected = false;
-      if (code === 4001) {
-        console.log("AgentDesk: authentication required — run 'agentdesk login'");
-      }
-    });
-  } catch {
-    // AgentDesk not running
+        if (code === 4001) {
+          console.log("AgentDesk: authentication required — run 'agentdesk login'");
+          return; // Don't reconnect on auth failure
+        }
+        // Auto-reconnect after 5 seconds
+        clearTimeout(reconnectTimer);
+        reconnectTimer = setTimeout(connectWs, 5000);
+      });
+    } catch {
+      // AgentDesk not running — retry in 10s
+      clearTimeout(reconnectTimer);
+      reconnectTimer = setTimeout(connectWs, 10000);
+    }
   }
 
+  connectWs();
+
   // --- Spawn Claude ---
   const child = spawn(
     "claude",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kendoo.agentdesk/agentdesk",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "description": "AI team orchestrator for Claude Code — run collaborative agent sessions from your terminal",
   "type": "module",
   "bin": {