@kendoo.agentdesk/agentdesk 0.6.2 → 0.6.4

package/cli/team.mjs

@@ -122,6 +122,7 @@ export async function runTeam(taskId, opts = {}) {
       createInstr += `Create a GitHub issue:\n- Run: gh issue create --title "..." --body "..."\n- After creation, use the returned issue number as the task ID for the rest of the session\n`;
     }
     createInstr += `\nTask description: ${description}\n`;
+    createInstr += `\nIMPORTANT: After creating the task, Jane MUST immediately announce the new task ID on its own line in this exact format:\nTASK_ID: <identifier>\nExample: TASK_ID: KEN-530\nThis is required so the session can be linked to the correct task.\n`;
     prompt += createInstr;
   }
 
@@ -324,12 +325,25 @@ export async function runTeam(taskId, opts = {}) {
   console.log("\n━━━ INTAKE ━━━\n");
   vizSend({ type: "phase:change", phase: "INTAKE" });
 
+  let totalInputTokens = 0;
+  let totalOutputTokens = 0;
+
   const rl = createInterface({ input: child.stdout });
 
   for await (const line of rl) {
     try {
       const event = JSON.parse(line);
 
+      // Track token usage from any event that reports it
+      if (event.usage) {
+        if (event.usage.input_tokens) totalInputTokens = Math.max(totalInputTokens, event.usage.input_tokens);
+        if (event.usage.output_tokens) totalOutputTokens += (event.usage.output_tokens_delta || 0);
+      }
+      if (event.message?.usage) {
+        if (event.message.usage.input_tokens) totalInputTokens = Math.max(totalInputTokens, event.message.usage.input_tokens);
+        if (event.message.usage.output_tokens) totalOutputTokens = Math.max(totalOutputTokens, event.message.usage.output_tokens);
+      }
+
       if (event.type === "assistant" && event.message?.content) {
         for (const block of event.message.content) {
           if (block.type === "text" && block.text.trim()) {
@@ -342,6 +356,24 @@ export async function runTeam(taskId, opts = {}) {
               continue;
             }
 
+            // Detect TASK_ID announcement from Jane
+            const taskIdMatch = text.match(/TASK_ID:\s*(\S+)/);
+            if (taskIdMatch && createTask) {
+              const newTaskId = taskIdMatch[1];
+              taskId = newTaskId;
+              // Build task link for the real task
+              if (tracker === "linear" && config.linear?.workspace) {
+                taskLink = `https://linear.app/${config.linear.workspace}/issue/${newTaskId}`;
+              } else if (tracker === "jira" && config.jira?.baseUrl) {
+                taskLink = `${config.jira.baseUrl}/browse/${newTaskId}`;
+              } else if (tracker === "github") {
+                const repo = config.github?.repo || "";
+                if (repo) taskLink = `https://github.com/${repo}/issues/${newTaskId}`;
+              }
+              console.log(`\n  # Task: ${newTaskId}${taskLink ? ` (${taskLink})` : ""}\n`);
+              vizSend({ type: "session:update", taskId: newTaskId, taskLink, title: description || newTaskId });
+            }
+
             const textLines = text.split("\n");
             let i = 0;
             while (i < textLines.length) {
@@ -418,10 +450,16 @@ export async function runTeam(taskId, opts = {}) {
       }
 
       if (event.type === "result") {
+        // Final usage from result event
+        if (event.usage) {
+          if (event.usage.input_tokens) totalInputTokens = Math.max(totalInputTokens, event.usage.input_tokens);
+          if (event.usage.output_tokens) totalOutputTokens = Math.max(totalOutputTokens, event.usage.output_tokens);
+        }
+        const totalTokens = totalInputTokens + totalOutputTokens;
         const totalTime = ((Date.now() - startTime) / 1000).toFixed(1);
         console.log(`\n━━━ DONE ━━━`);
-        console.log(`  ${totalTime}s  |  ${stepCount} steps\n`);
-        vizSend({ type: "session:end", duration: `${totalTime}s`, steps: stepCount });
+        console.log(`  ${totalTime}s  |  ${stepCount} steps${totalTokens ? `  |  ${totalTokens.toLocaleString()} tokens` : ""}\n`);
+        vizSend({ type: "session:end", duration: `${totalTime}s`, steps: stepCount, inputTokens: totalInputTokens, outputTokens: totalOutputTokens });
         clearInterval(inboxTimer);
         setTimeout(() => { try { vizWs?.close(); } catch {} }, 500);
       }

package/package.json

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