@kendoo.agentdesk/agentdesk 0.5.0 → 0.5.2

package/cli/agents.mjs

@@ -148,6 +148,7 @@ export const BUILT_IN_AGENTS = {
  */
 export function resolveTeam(config) {
   const teamConfig = config.team || null;
+  const customAgents = config.projectAgents || [];
 
   // Default: all 7 agents
   if (!teamConfig) {
@@ -162,7 +163,29 @@ export function resolveTeam(config) {
       if (agent) {
         team.push({ name: entry, ...agent });
       } else {
-        console.warn(`Warning: Unknown agent "${entry}" — skipping`);
+        // Check if it's a custom project agent
+        const custom = customAgents.find(a => a.name === entry);
+        if (custom) {
+          team.push({
+            name: custom.name,
+            badge: `** ${custom.name.toUpperCase()} **`,
+            role: custom.role || "Team Member",
+            description: custom.role || "Custom team member",
+            groundRules: custom.when ? `${custom.name} is invoked: ${custom.when}. How to use: ${custom.how || "as needed"}.` : "",
+            brainstorm: { tag: "SAY", focus: custom.role || "general input" },
+            planning: `${custom.role || custom.name} Review`,
+            execution: custom.when ? {
+              step: `${custom.name} reviews`,
+              tasks: [
+                `Review changes relevant to ${custom.role || "this agent's scope"}.`,
+                custom.how ? `Interaction: ${custom.how}` : "Provide feedback on the changes.",
+              ],
+              order: 2.5,
+            } : null,
+          });
+        } else {
+          console.warn(`Warning: Unknown agent "${entry}" — skipping`);
+        }
       }
     } else if (typeof entry === "object" && entry.name) {
       // Custom agent

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.2",
   "description": "AI team orchestrator for Claude Code — run collaborative agent sessions from your terminal",
   "type": "module",
   "bin": {