npm - agent-dag - Versions diffs - 1.14.2 → 1.15.1 - Mend

agent-dag 1.14.2 → 1.15.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (4) hide show

package/dist/web/assets/{index-CkZuDTIx.js → index-Bq4SSF-l.js} +10 -10
package/dist/web/index.html +1 -1
package/package.json +1 -1
package/src/server/index.mjs +62 -22

package/dist/web/index.html CHANGED Viewed

@@ -5,7 +5,7 @@
     <meta name="viewport" content="width=device-width, initial-scale=1" />
     <title>agent-dag</title>
     <link rel="icon" href="data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 100 100'%3E%3Ctext y='84' font-size='84'%3E%E2%97%89%3C/text%3E%3C/svg%3E" />
-    <script type="module" crossorigin src="/assets/index-CkZuDTIx.js"></script>
+    <script type="module" crossorigin src="/assets/index-Bq4SSF-l.js"></script>
     <link rel="stylesheet" crossorigin href="/assets/index-Bwj0xSAo.css">
   </head>
   <body>

package/package.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "name": "agent-dag",
-  "version": "1.14.2",
+  "version": "1.15.1",
   "description": "Live DAG of Claude Code agents — watch parallel subagents fork, call tools, and return on one calm canvas.",
   "type": "module",
   "bin": {

package/src/server/index.mjs CHANGED Viewed

@@ -73,32 +73,59 @@ async function maybeRotatePersistFile() {
 // payloads for that session before broadcasting, (b) emit a synthetic
 // `ModelObserved` event so the client backfills agents created before
 // the model was resolved.
-const modelBySession = new Map();         // sessionId -> "claude-…"
+const modelBySession = new Map();         // sessionId -> { rootModel, subsSig }
 const pendingTranscriptReads = new Set(); // sessionId currently being read
+const modelLastReadAt = new Map();        // sessionId -> ms timestamp (re-read throttle)
+const MODEL_READ_THROTTLE_MS = 2500;
 async function readModelFromTranscript(path) {
+  // Returns { rootModel, subagentModels } — root is the most recent
+  // non-sidechain assistant model, subagents are attributed by the Task
+  // tool_use id that owns them. CC transcripts mark subagent messages
+  // with `isSidechain:true` and (for current schemas) a `parentToolUseID`
+  // or `parent_tool_use_id` referencing the Task invocation.
   try {
     const s = await stat(path);
     if (s.size === 0) return null;
-    // Read up to last 128 KB — plenty for the most-recent model
-    // declaration. Reading from the tail handles sessions that switched
-    // model mid-conversation (we want the current one).
-    const TAIL = 128 * 1024;
-    const start = Math.max(0, s.size - TAIL);
     const fh = await open(path, "r");
+    let text;
     try {
-      const len = s.size - start;
-      const buf = Buffer.alloc(len);
-      await fh.read(buf, 0, len, start);
-      const text = buf.toString("utf8");
-      // Scan all matches and return the LAST one — most recent model used.
-      const re = /"model"\s*:\s*"(claude[-_][^"]+)"/gi;
-      let last = null;
-      for (const m of text.matchAll(re)) last = m[1];
-      return last;
+      const buf = Buffer.alloc(s.size);
+      await fh.read(buf, 0, s.size, 0);
+      text = buf.toString("utf8");
     } finally {
       await fh.close();
     }
+    let rootModel = null;
+    const subagentModels = {};
+    let anyModelSeen = null; // regex-style fallback — the last claude-* model
+                             // we encountered, regardless of sidechain flag.
+                             // Keeps the old behavior alive when the JSONL
+                             // schema doesn't expose `isSidechain`.
+    for (const line of text.split("\n")) {
+      if (!line) continue;
+      let obj;
+      try { obj = JSON.parse(line); } catch { continue; }
+      const msg = obj && obj.message;
+      const model = (msg && typeof msg.model === "string" && /^claude[-_]/i.test(msg.model)) ? msg.model
+                  : (typeof obj.model === "string" && /^claude[-_]/i.test(obj.model))         ? obj.model
+                  : null;
+      if (!model) continue;
+      anyModelSeen = model;
+      const isSide = obj.isSidechain === true || obj.is_sidechain === true;
+      const ptid = obj.parentToolUseID || obj.parent_tool_use_id || obj.parentToolUseId || null;
+      if (isSide && ptid) {
+        subagentModels[ptid] = model;
+      } else if (!isSide) {
+        rootModel = model;
+      }
+    }
+    // Fallback: if no `isSidechain:false` entry was found (older schema, or
+    // a transcript that only contains sidechain blocks), use the last-seen
+    // model so the root card still shows SOMETHING instead of going blank.
+    if (!rootModel) rootModel = anyModelSeen;
+    if (!rootModel && Object.keys(subagentModels).length === 0) return null;
+    return { rootModel, subagentModels };
   } catch {
     return null;
   }
@@ -109,16 +136,29 @@ function maybeResolveModel(payload) {
   const sid = payload.session_id;
   const tp = payload.transcript_path;
   if (!sid || !tp) return;
-  if (modelBySession.has(sid)) return;
+  // Re-read on every event for this session — the cache was preventing us
+  // from picking up subagent models that arrive after the root is known.
+  // Throttle so we don't thrash the filesystem.
   if (pendingTranscriptReads.has(sid)) return;
+  const now = Date.now();
+  const last = modelLastReadAt.get(sid) ?? 0;
+  if (now - last < MODEL_READ_THROTTLE_MS) return;
+  modelLastReadAt.set(sid, now);
   pendingTranscriptReads.add(sid);
   readModelFromTranscript(tp)
-    .then(model => {
-      if (!model) return;
-      modelBySession.set(sid, model);
-      // Synthetic enrichment event — reducer applies to every agent in
-      // this session, including ones created before we resolved.
-      pushEvent({ hook_event_name: "ModelObserved", session_id: sid, model }, "internal");
+    .then(result => {
+      if (!result) return;
+      const { rootModel, subagentModels } = result;
+      const prev = modelBySession.get(sid);
+      const subsSig = JSON.stringify(subagentModels);
+      if (prev && prev.rootModel === rootModel && prev.subsSig === subsSig) return;
+      modelBySession.set(sid, { rootModel, subsSig });
+      pushEvent({
+        hook_event_name: "ModelObserved",
+        session_id: sid,
+        model: rootModel,
+        subagentModels,
+      }, "internal");
     })
     .catch(() => {})
     .finally(() => pendingTranscriptReads.delete(sid));