@mu-crew/mu 3.3.0 → 3.4.0

package/dist/extension/mu-pi.js

@@ -55,6 +55,11 @@ function delegateStatus(n) {
   return n > 0 ? `${n} delegate${n === 1 ? "" : "s"} running` : void 0;
 }
 var DELEGATE_TIMEOUT_S = 3600;
+var DELEGATE_MAX_DEFAULT = 16;
+function delegateMax(env = process.env) {
+  const n = Number(env.MU_DELEGATE_MAX);
+  return Number.isInteger(n) && n > 0 ? n : DELEGATE_MAX_DEFAULT;
+}
 var W = DELEGATE_WORKSTREAM;
 function delegateEnabled(env = process.env) {
   if (env.MU_MANAGED_AGENT) return false;
@@ -141,6 +146,7 @@ function labelStem(label) {
 function registerDelegate(pi, run = defaultRunner()) {
   if (!delegateEnabled()) return;
   const inflight = /* @__PURE__ */ new Map();
+  let starting = 0;
   const reserved = /* @__PURE__ */ new Set();
   let ui;
   const showStatus = () => {
@@ -253,7 +259,7 @@ function registerDelegate(pi, run = defaultRunner()) {
   pi.registerTool({
     name: DELEGATE_TOOL,
     label: "mu delegate",
-    description: `Subagent: run one self-contained task in a fresh pi agent (its own mux pane in mu's ${W} workstream) while you keep working. Use it whenever you would reach for a subagent: research, a review, a draft, an investigation; call it several times to fan out in parallel. The subagent starts with no context, in your working directory: put everything it needs in task. The call returns at once and the answer arrives later as a follow-up message: carry on with other work, or end your turn if your next step needs the answer (the follow-up resumes you). The pane closes after a clean finish; keep: true leaves it open.`,
+    description: `Subagent: run one self-contained task in a fresh pi agent (its own mux pane in mu's ${W} workstream) while you keep working. Use it whenever you would reach for a subagent: research, a review, a draft, an investigation; call it several times to fan out in parallel (at most ${delegateMax()} in flight). The subagent starts with no context, in your working directory: put everything it needs in task. The call returns at once and the answer arrives later as a follow-up message: carry on with other work, or end your turn if your next step needs the answer (the follow-up resumes you). The pane closes after a clean finish; keep: true leaves it open.`,
     promptSnippet: "Subagent: delegate a self-contained task to a background pi agent; its answer arrives later as a follow-up",
     parameters: {
       type: "object",
@@ -296,83 +302,106 @@ function registerDelegate(pi, run = defaultRunner()) {
       track(ctx);
       const task = typeof params.task === "string" ? params.task.trim() : "";
       if (!task) throw new Error("mu_delegate needs a non-empty task");
-      const brief = typeof params.brief === "string" ? params.brief.trim() : "";
-      const keep = params.keep === true;
-      const t = params.timeout;
-      if (t !== void 0 && !(typeof t === "number" && Number.isFinite(t) && t > 0))
+      const max = delegateMax();
+      if (inflight.size + starting >= max)
         throw new Error(
-          `mu_delegate: timeout must be a positive number of seconds (got ${JSON.stringify(t)})`
+          `mu_delegate: ${inflight.size + starting} delegates already running (MU_DELEGATE_MAX=${max}). End your turn; issue the rest as answers arrive (each answer resumes you).`
         );
-      const timeoutS = typeof t === "number" ? Math.ceil(t) : DELEGATE_TIMEOUT_S;
-      let cwd;
-      if (params.workspace !== true) {
-        cwd = typeof params.cwd === "string" && params.cwd ? params.cwd : ctx?.cwd ?? process.cwd();
-        if (!isDir(cwd)) throw new Error(`mu_delegate: cwd ${cwd} is not a directory`);
-      }
-      const { name, attach, workspace } = await spawn(params, cwd, signal);
-      const entry = { abort: new AbortController(), cancelling: false };
-      inflight.set(name, entry);
-      showStatus();
-      const started = Date.now();
-      const waiting = mu(
-        ["agent", "wait", name, "-w", W, "--json", "--timeout", String(timeoutS)],
-        entry.abort.signal
-      );
-      const sent = await mu(["agent", "send", name, brief ? `${brief}
-
-${task}` : task, "-w", W]);
-      if (sent.code !== 0) {
-        entry.abort.abort();
-        forget(name);
-        const c = await mu(["agent", "close", name, "-w", W]);
-        const pane = c.code === 0 ? "Pane closed." : `Pane kept; close it with: mu agent close ${name} -w ${W}`;
-        throw new Error(`${failure(`mu agent send to ${W}/${name}`, sent).message}. ${pane}`);
-      }
-      const settle = async (r) => {
-        forget(name);
-        try {
-          await deliver(name, r, keep, {
-            timeoutS,
-            elapsedMs: Date.now() - started,
-            ...attach ? { attach } : {},
-            ...workspace ? { workspace } : {}
-          });
-        } catch (e) {
-          await Promise.resolve(
-            pi.sendMessage(
-              {
-                customType: DELEGATE_MESSAGE_TYPE,
-                content: `Delegate ${W}/${name} settled, but delivering its answer failed: ${e instanceof Error ? e.message : String(e)}. Read it with: mu agent read ${name} -n 50 -w ${W}`,
-                display: true,
-                details: { name, workstream: W, outcome: null, closed: false }
-              },
-              { deliverAs: "followUp", triggerTurn: true }
-            )
-          ).catch(() => {
-          });
-        }
+      starting++;
+      let slotHeld = true;
+      const freeSlot = () => {
+        if (slotHeld) starting--;
+        slotHeld = false;
       };
-      void waiting.then(async (r) => {
-        if (inflight.get(name) !== entry) return;
-        if (entry.cancelling) entry.parked = () => settle(r);
-        else await settle(r);
-      });
-      const lines = [
-        `Delegated to ${W}/${name}. Its answer arrives as a follow-up message when it finishes; do not wait or poll.`,
-        workspace ? `Workspace: ${workspace}` : void 0,
-        attach ? `Attach: ${attach}` : void 0,
-        `Cancel: ${DELEGATE_CANCEL_TOOL} { name: "${name}" }`
-      ].filter((x) => x !== void 0);
-      return {
-        content: [{ type: "text", text: lines.join("\n") }],
-        details: {
+      try {
+        const brief = typeof params.brief === "string" ? params.brief.trim() : "";
+        const keep = params.keep === true;
+        const t = params.timeout;
+        if (t !== void 0 && !(typeof t === "number" && Number.isFinite(t) && t > 0))
+          throw new Error(
+            `mu_delegate: timeout must be a positive number of seconds (got ${JSON.stringify(t)})`
+          );
+        const timeoutS = typeof t === "number" ? Math.ceil(t) : DELEGATE_TIMEOUT_S;
+        let cwd;
+        if (params.workspace !== true) {
+          cwd = typeof params.cwd === "string" && params.cwd ? params.cwd : ctx?.cwd ?? process.cwd();
+          if (!isDir(cwd)) throw new Error(`mu_delegate: cwd ${cwd} is not a directory`);
+        }
+        const { name, attach, workspace } = await spawn(params, cwd, signal);
+        const entry = { abort: new AbortController(), cancelling: false };
+        inflight.set(name, entry);
+        freeSlot();
+        showStatus();
+        const started = Date.now();
+        const waiting = mu(
+          ["agent", "wait", name, "-w", W, "--json", "--timeout", String(timeoutS)],
+          entry.abort.signal
+        );
+        const sent = await mu([
+          "agent",
+          "send",
           name,
-          workstream: W,
-          attach: attach ?? null,
-          workspace: workspace ?? null,
-          keep
+          brief ? `${brief}
+
+${task}` : task,
+          "-w",
+          W
+        ]);
+        if (sent.code !== 0) {
+          entry.abort.abort();
+          forget(name);
+          const c = await mu(["agent", "close", name, "-w", W]);
+          const pane = c.code === 0 ? "Pane closed." : `Pane kept; close it with: mu agent close ${name} -w ${W}`;
+          throw new Error(`${failure(`mu agent send to ${W}/${name}`, sent).message}. ${pane}`);
         }
-      };
+        const settle = async (r) => {
+          forget(name);
+          try {
+            await deliver(name, r, keep, {
+              timeoutS,
+              elapsedMs: Date.now() - started,
+              ...attach ? { attach } : {},
+              ...workspace ? { workspace } : {}
+            });
+          } catch (e) {
+            await Promise.resolve(
+              pi.sendMessage(
+                {
+                  customType: DELEGATE_MESSAGE_TYPE,
+                  content: `Delegate ${W}/${name} settled, but delivering its answer failed: ${e instanceof Error ? e.message : String(e)}. Read it with: mu agent read ${name} -n 50 -w ${W}`,
+                  display: true,
+                  details: { name, workstream: W, outcome: null, closed: false }
+                },
+                { deliverAs: "followUp", triggerTurn: true }
+              )
+            ).catch(() => {
+            });
+          }
+        };
+        void waiting.then(async (r) => {
+          if (inflight.get(name) !== entry) return;
+          if (entry.cancelling) entry.parked = () => settle(r);
+          else await settle(r);
+        });
+        const lines = [
+          `Delegated to ${W}/${name}. Its answer arrives as a follow-up message when it finishes; do not wait or poll.`,
+          workspace ? `Workspace: ${workspace}` : void 0,
+          attach ? `Attach: ${attach}` : void 0,
+          `Cancel: ${DELEGATE_CANCEL_TOOL} { name: "${name}" }`
+        ].filter((x) => x !== void 0);
+        return {
+          content: [{ type: "text", text: lines.join("\n") }],
+          details: {
+            name,
+            workstream: W,
+            attach: attach ?? null,
+            workspace: workspace ?? null,
+            keep
+          }
+        };
+      } finally {
+        freeSlot();
+      }
     }
   });
   pi.registerTool({
@@ -561,7 +590,7 @@ function registerNudge(pi, run = defaultRunner(), skillMd = readSkill()) {
 }
 
 // extension/mu-pi.ts
-var EXT_VERSION = true ? "3.3.0" : void 0;
+var EXT_VERSION = true ? "3.4.0" : void 0;
 var FRESH_COMMAND = "mu-fresh";
 var FRESH_TIMEOUT_MS = 3e4;
 function commandName(name) {