@arhen/pi-core-subagent 1.2.0 → 1.2.2

package/README.md

@@ -90,10 +90,19 @@ Chain — `{previous}` is replaced with the prior agent's output:
 }
 ```
 
+The call line renders the graph in §2 notation as the model types it:
+
 ```
-wave 1: api ∥ db     →  gate  →  wave 2: doc
+subagent graph 3
+  6 at a time
+  wave1[api ∥ db] → gate → wave2[doc]
+  api api-mapper Map every route in src/api/
+  db  db-mapper Map the schema in src/db/
+  doc writer ✎ ← api, db Write ARCHITECTURE.md from the maps above. Verify: test -s ARCHI…
 ```
 
+`✎` marks a write-toolset task; `←` lists its edges. With no `needs` anywhere the wave line is omitted entirely.
+
 What the edge does:
 
 - **Gates** — `doc` starts only after both `api` and `db` finish.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arhen/pi-core-subagent",
-  "version": "1.2.0",
+  "version": "1.2.2",
   "type": "module",
   "description": "pi extension: fast in-process subagents with a dependency-graph scheduler (needs edges gate tasks and carry upstream output into dependent prompts), plus background runs, intercom and agent-to-agent mailbox. Leader defines agents inline.",
   "license": "MIT",

package/src/index.ts

@@ -472,6 +472,33 @@ export function resolveNeeds(inputs: { id?: string; needs?: string[] }[], mode:
 	return edges;
 }
 
+/**
+ * Graph Protocol §2 notation: `wave1[api ∥ db] → gate → wave2[doc]`.
+ *
+ * Tolerates half-streamed args: a need pointing at an id that has not arrived yet
+ * keeps its task out of the ready set, so the layout settles as the model types.
+ * Returns "" when there are no edges — flat fan-out gets no graph vocabulary.
+ */
+export function waveNotation(tasks: { id?: string; needs?: string[] }[]): string {
+	if (!tasks.some((t) => t.needs?.length)) return "";
+	const ids = tasks.map((t, i) => t.id ?? `task_${i + 1}`);
+	const settled = new Set<string>();
+	let remaining = tasks.map((t, i) => ({ id: ids[i] as string, needs: t.needs ?? [] }));
+	const waves: string[][] = [];
+	while (remaining.length > 0) {
+		const ready = remaining.filter((t) => t.needs.every((n) => settled.has(n)));
+		if (ready.length === 0) break; // cycle, or an upstream id not typed yet
+		waves.push(ready.map((t) => t.id));
+		for (const t of ready) settled.add(t.id);
+		remaining = remaining.filter((t) => !settled.has(t.id));
+	}
+	if (remaining.length > 0) waves.push(remaining.map((t) => t.id)); // show them rather than drop them
+	if (waves.length < 2) return "";
+	const full = waves.map((w, i) => `wave${i + 1}[${w.join(" ∥ ")}]`).join(" → gate → ");
+	// Long graphs: keep the shape, drop the names.
+	return full.length <= 100 ? full : waves.map((w, i) => `wave${i + 1}[${w.length}]`).join(" → gate → ");
+}
+
 /**
  * Graph Protocol §6: the edge carries the upstream output, not just ordering.
  * Upstream results are prepended verbatim; `{previous}` stays supported so old
@@ -1402,7 +1429,23 @@ export default function (pi: ExtensionAPI) {
 			if (args.concurrency) parts.push(`${args.concurrency} at a time`);
 			if (args.maxRuntimeMs) parts.push(`${Math.round(args.maxRuntimeMs / 60000)}m limit`);
 			const params = parts.length > 0 ? `\n  ${theme.fg("dim", parts.join(" · "))}` : "";
-			return new Text(`${theme.fg("toolTitle", theme.bold("subagent"))} ${theme.fg("accent", mode)}${flags ? ` ${theme.fg("muted", `[${flags}]`)}` : ""}${params}`, 0, 0);
+			const notation = waveNotation(tasks);
+			const graphLine = notation ? `\n  ${theme.fg("muted", notation)}` : "";
+			// The plan the model actually wrote: ids, edges, toolset. Streams in as args arrive,
+			// so a graph is visible before the first child spawns.
+			const plan = tasks
+				.filter((t: any) => t.agent || t.id)
+				.map((t: any, i: number) => {
+					const id = t.id ?? `task_${i + 1}`;
+					const edge = t.needs?.length ? theme.fg("muted", ` ← ${t.needs.join(", ")}`) : "";
+					const mark = t.write ? theme.fg("warning", " ✎") : "";
+					// Plain clip, not truncateText — that one appends a multi-line session-file notice.
+					const flat = String(t.task ?? "").replace(/\s+/g, " ").trim();
+					const what = flat ? theme.fg("dim", ` ${flat.length > 64 ? `${flat.slice(0, 64)}…` : flat}`) : "";
+					return `\n  ${theme.fg("muted", id)} ${theme.fg("accent", t.agent ?? "…")}${mark}${edge}${what}`;
+				})
+				.join("");
+			return new Text(`${theme.fg("toolTitle", theme.bold("subagent"))} ${theme.fg("accent", mode)}${flags ? ` ${theme.fg("muted", `[${flags}]`)}` : ""}${params}${graphLine}${plan}`, 0, 0);
 		},
 		renderResult(result, { expanded }, theme) {
 			const run = result.details?.run;