@howarewoo/pi-subagents 0.1.1 → 0.2.1

package/src/delegation-types.ts

@@ -0,0 +1,111 @@
+import type {
+  AgentSession,
+  AgentSessionEvent,
+  ExtensionToolContext,
+  Skill,
+  ToolDefinition,
+} from "@earendil-works/pi-coding-agent";
+import type { CapabilityDenial, ToolMetadata } from "./capabilities.ts";
+import type { StructuredOutcome } from "./outcomes.ts";
+import type { ModelSelection } from "./model-policy.ts";
+import type { ExecutionLimits } from "./scheduling.ts";
+import type { BuiltinTool, Profile } from "./profiles.ts";
+
+export type DelegationMessage = {
+  id: string;
+  sender: ChildRef | "root";
+  recipient: ChildRef | "root";
+  text: string;
+  mode: "inbox" | "steer" | "followUp";
+  status: "accepted" | "queued" | "handled" | "delivered" | "not-running" | "error";
+  error?: string;
+};
+export type CoordinationWait =
+  | { status: "timeout" | "cancelled" | "shutdown" }
+  | { status: "messages"; messages: DelegationMessage[] }
+  | { status: "message"; message: DelegationMessage }
+  | {
+      status: "outcome";
+      outcome: Pick<ChildSnapshot, "id" | "generation" | "status" | "limitReason" | "pauseReason">;
+    };
+export type ChildRef = Readonly<{ id: string; generation: number }>;
+export type ChildSnapshot = Readonly<
+  ChildRef & {
+    parentId: string;
+    parentGeneration?: number;
+    depth: number;
+    writerLease: boolean;
+    limits: ExecutionLimits;
+    limitReason?: string;
+    toolCalls: number;
+    spawned?: number;
+    pauseReason?: string;
+    retainedContext?: boolean;
+    window?: number;
+    status:
+      | "queued"
+      | "setup"
+      | "running"
+      | "pausing"
+      | "paused"
+      | "draining"
+      | "settled"
+      | "failed"
+      | "cancelled";
+    profile: string;
+    tools: readonly BuiltinTool[];
+    leafTools?: readonly string[];
+    skills?: readonly string[];
+    capabilityDenials?: readonly CapabilityDenial[];
+    cwd?: string;
+    assignment: string;
+    activity: string;
+    progress: string;
+    result: string;
+    model: { provider: string; id: string };
+    effort: string | null;
+    prewalk?: {
+      target: ModelSelection;
+      status: "armed" | "handed-off" | "suppressed-exact" | "no-op";
+    };
+    telemetry: {
+      requests: number | null;
+      tokens: ReturnType<AgentSession["getSessionStats"]>["tokens"] | null;
+      cost: number | null;
+      context: ReturnType<AgentSession["getContextUsage"]> | null;
+    };
+    outcome?: StructuredOutcome;
+    artifacts?: { text?: string; structured?: string; error?: string };
+    error?: string;
+    transcript?: string;
+    disposed: boolean;
+    createdAt: number;
+    endedAt: number | null;
+  }
+>;
+export type RegistryEvent = Readonly<{
+  type: "changed" | "sdk";
+  child: ChildSnapshot;
+  event?: AgentSessionEvent;
+}>;
+export type StartTask = {
+  cwd?: string;
+  model?: ModelSelection;
+  resultSchema?: unknown;
+  assignment: string;
+  profile: Profile;
+  skills: Skill[];
+  context: ExtensionToolContext;
+  assertTools: (names: readonly string[]) => void;
+  toolMetadata?: () => readonly ToolMetadata[];
+  /** Live parent codemode availability; never a grant to forward its dispatcher. */
+  codemodeEnabled?: () => boolean;
+  availableSkills?: () => readonly Skill[];
+  capabilityDenials?: CapabilityDenial[];
+  leafProvenance?: Readonly<Record<string, string>>;
+  callerId: string;
+  parent?: ChildRef;
+  limits?: Partial<ExecutionLimits>;
+  coordinationTools?: (actor: ChildRef) => ToolDefinition[];
+  nestedTool?: (parent: ChildRef) => ToolDefinition;
+};

package/src/hub.ts

@@ -3,7 +3,6 @@ import {
   Input,
   matchesKey,
   oklchColor,
-  sliceByColumn,
   truncateToWidth,
   visibleWidth,
   wrapTextWithAnsi,
@@ -15,229 +14,24 @@ import type { Theme } from "@earendil-works/pi-coding-agent";
 import type { ChildRef, ChildSnapshot, DelegationCoordinator } from "./coordinator.ts";
 import { createNativeThread, safe, type LiveTool, type ThreadDependencies } from "./thread.ts";
 export { safe } from "./thread.ts";
+export { createChildWidget } from "./subagent-widget.ts";
+export {
+  agentLabel,
+  fleetSummary,
+  taskCard,
+  transcriptLines,
+  writerWarning,
+} from "./task-presentation.ts";
+import {
+  agentLabel,
+  assignmentHeadline,
+  lineText,
+  taskDetails,
+  taskTree,
+  transcriptLines,
+} from "./task-presentation.ts";
 
 type Palette = Pick<Theme, "fg" | "style"> & { colors: Pick<Theme["colors"], "customMessageBg"> };
-const lineText = (text: string) => safe(text).replaceAll("\n", " ");
-const assignmentHeadline = (child: ChildSnapshot) =>
-  safe(child.assignment)
-    .split("\n")
-    .map((line) => line.trim())
-    .find(Boolean)
-    ?.replace(/^#{1,6}\s+/, "") || "Assignment";
-export const agentLabel = (child: ChildSnapshot) =>
-  `${assignmentHeadline(child)} · ${lineText(child.profile)}`;
-const active = (child: ChildSnapshot) => !child.disposed;
-const activity = (child: ChildSnapshot) =>
-  child.error ||
-  (active(child) ? child.activity || child.progress : child.result) ||
-  child.progress ||
-  "Waiting for SDK activity";
-export const writerWarning = "Model-tool writer only: no conflicting manual !/!! or external edits";
-const statuses = [
-  "queued",
-  "setup",
-  "running",
-  "draining",
-  "settled",
-  "failed",
-  "cancelled",
-] as const;
-export function fleetSummary(children: ChildSnapshot[]): string {
-  return children.length
-    ? statuses
-        .map((status) => `${status} ${children.filter((child) => child.status === status).length}`)
-        .join(" · ")
-    : "No tasks · ask the model to delegate_task";
-}
-
-/** Event-driven, bounded current-agent tree; history and full metrics remain in the Hub. */
-export function createChildWidget(
-  snapshot: () => ChildSnapshot[],
-  theme: Pick<Theme, "fg"> & Partial<Pick<Theme, "bold">>,
-  availableRows = () => 30,
-): Component {
-  return {
-    render(width) {
-      width = Math.max(0, width);
-      const children = snapshot();
-      if (!children.length) return [];
-      const queued = children.filter((child) => child.status === "queued").length;
-      const priority = (child: ChildSnapshot) =>
-        child.status === "draining"
-          ? 0
-          : child.status === "running"
-            ? 1
-            : child.status === "setup"
-              ? 2
-              : 3;
-      const live = children
-        .filter(
-          (child) =>
-            !child.disposed &&
-            (child.status === "setup" || child.status === "running" || child.status === "draining"),
-        )
-        .sort((a, b) => priority(a) - priority(b));
-      const writer = children.some((child) => child.writerLease);
-      const budget = Math.max(1, Math.min(5, Math.floor(availableRows() / 4)));
-      const counts =
-        width < 60
-          ? `${live.length} active · ${queued} queued`
-          : statuses
-              .flatMap((status) => {
-                const count = children.filter((child) => child.status === status).length;
-                return count ? [`${count} ${status}`] : [];
-              })
-              .join(" · ");
-      const rows = [
-        theme.fg("accent", `● Subagents ${children.length}`) + theme.fg("muted", ` · ${counts}`),
-      ];
-      if (writer && budget === 1)
-        rows[0] = theme.fg(
-          "warning",
-          `● Subagents ${children.length} · Model writer: !/!! + external unsafe · ${queued} queued`,
-        );
-      const room = Math.max(0, budget - 1 - (writer && budget > 1 ? 1 : 0));
-      const activityRows = live.length && room >= (live.length > 1 ? 3 : 2) ? 1 : 0;
-      const slots = room - activityRows;
-      const overflow = live.length > slots;
-      const shown = live.slice(0, Math.max(0, slots - (overflow ? 1 : 0)));
-      shown.forEach((child, index) => {
-        const last = index === shown.length - 1 && shown.length === live.length;
-        const color = child.status === "draining" ? "warning" : "accent";
-        const role = lineText(child.profile);
-        const indent = "  ".repeat(Math.min(3, Math.max(0, child.depth - 1)));
-        const prefix =
-          theme.fg("muted", `${indent}${last ? "└─" : "├─"} `) +
-          theme.fg(color, "● ") +
-          (theme.bold?.(role) ?? role) +
-          " ";
-        const elapsed = Math.max(
-          0,
-          Math.floor(((child.endedAt ?? Date.now()) - child.createdAt) / 1000),
-        );
-        const stats = [
-          child.status,
-          `${elapsed}s`,
-          `${child.toolCalls} tools`,
-          ...(child.telemetry.requests === null ? [] : [`${child.telemetry.requests} req`]),
-          ...(child.telemetry.tokens ? [`${child.telemetry.tokens.total} tok`] : []),
-          ...(child.telemetry.cost === null ? [] : [`$${child.telemetry.cost.toFixed(3)}`]),
-        ].join(" · ");
-        const metric = truncateToWidth(stats, Math.max(8, Math.floor(width / 2)));
-        const headline = assignmentHeadline(child);
-        rows.push(
-          prefix +
-            theme.fg(
-              "text",
-              truncateToWidth(
-                headline,
-                Math.max(4, width - visibleWidth(prefix) - visibleWidth(metric) - 3),
-              ),
-            ) +
-            theme.fg("muted", " · " + metric),
-        );
-        if (!index && activityRows) {
-          const prefix = `${indent}${last ? "   " : "│  "} └ `;
-          const text = lineText(activity(child));
-          const columns = Math.max(0, width - visibleWidth(prefix));
-          const length = visibleWidth(text);
-          // Progress is an accumulated tail; keep its newest text, not a frozen opening paragraph.
-          const preview =
-            length > columns
-              ? "…" +
-                sliceByColumn(text, length - Math.max(0, columns - 1), Math.max(0, columns - 1))
-              : text;
-          rows.push(theme.fg("muted", prefix + preview));
-        }
-      });
-      if (shown.length < live.length && room)
-        rows.push(theme.fg("muted", `└─ ${live.length - shown.length} more · /subagent-hub`));
-      if (writer && budget > 1)
-        rows.push(
-          theme.fg("warning", width < 75 ? "Model writer: !/!! + external unsafe" : writerWarning),
-        );
-      return rows.slice(0, budget).map((line) => truncateToWidth(line, width));
-    },
-    invalidate() {},
-  };
-}
-
-export function taskCard(child: ChildSnapshot | undefined, expanded = false): string[] {
-  if (!child)
-    return [
-      "Task identity unavailable",
-      "See tool error; /subagent-hub lists current runtime tasks",
-    ];
-  return [
-    `${truncateToWidth(agentLabel(child), 240, "… [preview truncated]")} #${child.generation} · ${child.status}`,
-    `Identity ${child.id} #${child.generation}`,
-    `${child.model.provider}/${child.model.id} · effort ${child.effort ?? "unavailable"}`,
-    truncateToWidth(lineText(activity(child)), 240, "… [preview truncated]"),
-    `Prewalk ${child.prewalk ? JSON.stringify(child.prewalk) : "off"} · advisor unavailable`,
-    `Parent ${child.parentId}${child.parentGeneration ? ` #${child.parentGeneration}` : ""} · depth ${child.depth} · writer lease ${child.writerLease}`,
-    ...(child.outcome
-      ? [
-          `Structured ${child.outcome.validation} · ${child.outcome.finalized ? "finalized" : "provisional"} · attempts ${child.outcome.attempts}`,
-        ]
-      : []),
-    ...(child.limitReason ? [`Limit: ${child.limitReason}`] : []),
-    ...(child.artifacts ? [`Artifacts: ${JSON.stringify(child.artifacts)}`] : []),
-    ...(expanded
-      ? [
-          "Assignment: " +
-            truncateToWidth(lineText(child.assignment), 480, "… [preview truncated]"),
-          "Result: " +
-            truncateToWidth(
-              lineText(child.result || child.error || "Not available yet"),
-              480,
-              "… [preview truncated]",
-            ),
-        ]
-      : []),
-    child.transcript
-      ? `Full transcript: ${child.transcript} · /subagent-hub`
-      : "Transcript unavailable (setup) · /subagent-hub",
-  ].map(lineText);
-}
-
-/** Read-only content projection: never dumps provider/auth objects, image payloads or opaque signatures. */
-export function transcriptLines(
-  messages: ReturnType<DelegationCoordinator["messages"]>,
-  toolsOnly = false,
-): string[] {
-  const lines: string[] = [];
-  for (const message of messages) {
-    if (message.role === "system") continue;
-    const start = lines.length;
-    const blocks = "content" in message ? message.content : undefined;
-    if (!toolsOnly || message.role === "toolResult") {
-      lines.push(
-        `[${message.role}${message.role === "toolResult" ? ` ${message.toolName} ${message.toolCallId} ${message.isError ? "ERROR" : "result"}` : ""}]`,
-      );
-      if (typeof blocks === "string") lines.push(safe(blocks));
-    }
-    if (Array.isArray(blocks))
-      for (const block of blocks) {
-        if (block.type === "toolCall")
-          lines.push(
-            `Tool ${block.name} ${block.id}\nArguments: ${safe(JSON.stringify(block.arguments, null, 2))}`,
-          );
-        else if (!toolsOnly || message.role === "toolResult") {
-          if (block.type === "text") lines.push(safe(block.text));
-          else if (block.type === "thinking") lines.push("Thinking: " + safe(block.thinking));
-          else if (block.type === "image")
-            lines.push(`[image ${safe(block.mimeType)}; payload omitted]`);
-        }
-      }
-    if (message.role === "assistant" && message.errorMessage)
-      lines.push("Error: " + safe(message.errorMessage));
-    if (start && lines.length > start) lines.splice(start, 0, "");
-  }
-  return lines.length
-    ? lines.map(safe)
-    : [toolsOnly ? "No tool calls yet" : "Waiting for SDK messages"];
-}
-
 /** Observation only: closing or disposing never touches child execution. */
 export function createHub(
   coordinator: DelegationCoordinator,
@@ -289,26 +83,28 @@ export function createHub(
   });
   const identity = (ref: ChildRef) => `${ref.id}/${ref.generation}`;
   const roster = () =>
-    coordinator
-      .list()
-      .flatMap((child) => coordinator.generations(child.id))
-      .filter((child) =>
-        lineText(
-          [
-            child.id,
-            child.parentId,
-            child.profile,
-            child.status,
-            child.assignment,
-            child.progress,
-            child.result,
-            child.error,
-            child.model.id,
-          ].join(" "),
-        )
-          .toLowerCase()
-          .includes(query.toLowerCase()),
-      );
+    taskTree(
+      coordinator
+        .list()
+        .flatMap((child) => coordinator.generations(child.id))
+        .filter((child) =>
+          lineText(
+            [
+              child.id,
+              child.parentId,
+              child.profile,
+              child.status,
+              child.assignment,
+              child.progress,
+              child.result,
+              child.error,
+              child.model.id,
+            ].join(" "),
+          )
+            .toLowerCase()
+            .includes(query.toLowerCase()),
+        ),
+    ).map(({ child }) => child);
   const selection = () => {
     const children = roster();
     if (!children.some((child) => identity(child) === selected)) {
@@ -337,6 +133,7 @@ export function createHub(
         !child ||
         child.disposed ||
         child.status === "draining" ||
+        child.status === "pausing" ||
         (text !== undefined && child.status !== "running")
       )
         throw new Error(
@@ -452,8 +249,6 @@ export function createHub(
         footer.push(
           theme.fg("muted", `Filter: ${lineText(query)} · ${children.length} generations`),
         );
-      if (current.some((item) => item.writerLease))
-        footer.push(theme.fg("warning", "WRITER: model tools only; no manual/external writes"));
       if (notice) {
         const lines = wrapTextWithAnsi(lineText(notice), Math.max(1, inner));
         footer.push(
@@ -517,40 +312,13 @@ export function createHub(
       let nativeLines: string[] | undefined;
       if (child) {
         if (tab === "details") {
-          const metric = (value: unknown) =>
-            value === null || value === undefined ? "unavailable" : JSON.stringify(value);
-          detail = [
-            `Identity ${child.id} #${child.generation}`,
-            "",
-            "Assignment",
-            safe(child.assignment),
-            "",
-            "Result",
-            safe(child.result || "Not available yet"),
-            "",
-            ...(child.error ? ["Error", safe(child.error), ""] : []),
-            ...(child.outcome
-              ? ["Structured outcome", JSON.stringify(child.outcome, null, 2), ""]
-              : []),
-            ...(child.artifacts
-              ? ["Saved artifacts", JSON.stringify(child.artifacts, null, 2), ""]
-              : []),
-            `Workspace: ${safe(child.cwd ?? "parent workspace")}`,
-            `Effective builtin tools: ${child.tools.join(", ")}`,
-            `${child.model.provider}/${child.model.id} · effort ${child.effort ?? "unavailable"}`,
-            `Prewalk ${child.prewalk ? JSON.stringify(child.prewalk) : "off"} · advisor unavailable`,
-            `Presence ${coordinator.agents().find((agent) => agent.id === child.id && agent.generation === child.generation)?.presence ?? "historical"} · resources ${child.disposed ? "disposed" : "live"}`,
-            `Elapsed ${Math.max(0, Math.floor(((child.endedAt ?? Date.now()) - child.createdAt) / 1000))}s`,
-            `Direct requests ${metric(child.telemetry.requests)} · cost ${metric(child.telemetry.cost)}`,
-            `Direct tokens ${metric(child.telemetry.tokens)}`,
-            `Context ${metric(child.telemetry.context)}`,
-            `Parent ${child.parentId} #${child.parentGeneration ?? "root"} · depth ${child.depth} · direct generation only`,
-            `Limits ${JSON.stringify(child.limits)} · calls ${child.toolCalls} · reason ${child.limitReason ?? "none"}`,
-            "Activity: " + safe(activity(child)),
-            "Full transcript: " + safe(child.transcript ?? "unavailable (queued/setup)"),
-            "Followup/revive: native delegate_coordinate only (Hub has no tool dispatcher)",
-            ...(child.writerLease ? [writerWarning] : []),
-          ];
+          detail = taskDetails(
+            child,
+            coordinator
+              .agents()
+              .find((agent) => agent.id === child.id && agent.generation === child.generation)
+              ?.presence,
+          );
         } else {
           try {
             const messages = coordinator.messages(child);
@@ -626,26 +394,33 @@ export function createHub(
               ? "success"
               : "accent";
       const selectedLabel = child
-        ? `${agentLabel(child)} #${child.generation} · ${child.status}${historical(child) ? " · history" : ""}`
+        ? `${agentLabel(child)} · ${child.status}${historical(child) ? " · history" : ""}`
         : "No selection";
       const viewLabel = `${tab.toUpperCase()} · Tab: ${tab === "thread" ? "details" : tab === "details" ? "tools" : "thread"}`;
       const position = `${tab === "thread" ? (following ? "LIVE · " : "READING · ") : ""}${offset + 1}–${Math.min(wrapped.length, offset + viewport)}/${wrapped.length} lines${tab === "thread" ? (following ? " · k/PgUp read" : " · End: live") : offset < maxOffset ? " · more ↓" : ""}`;
       const pane: string[] = [];
       if (wide) {
-        const slots = Math.max(1, Math.floor((paneHeight - 2) / 3));
+        const slots = Math.max(1, Math.floor((paneHeight - 2) / 2));
         const start = Math.max(0, index - slots + 1);
         const left = [
           theme.fg("accent", `SUBAGENTS · ${children.length} generations`),
-          ...children
+          ...taskTree(children)
             .slice(start, start + slots)
-            .flatMap((item) => [
-              theme.fg(
-                identity(item) === selected ? "accent" : "text",
-                `${identity(item) === selected ? ">" : " "} ${" ".repeat(Math.min(4, Math.max(0, item.depth - 1)))}└ ${agentLabel(item)} #${item.generation}${historical(item) ? " · history" : ""}`,
-              ),
-              theme.fg(statusColor(item), `  ${item.profile} · ${item.status}`),
-              theme.fg("muted", `  #${item.generation}${historical(item) ? " · history" : ""}`),
-            ]),
+            .flatMap(({ child: item, indent, last, continuation }) => {
+              const prefix = `${identity(item) === selected ? ">" : " "} ${indent}${last ? "└" : "├"} `;
+              return [
+                theme.fg(identity(item) === selected ? "accent" : "text", prefix[0]) +
+                  theme.fg("text", prefix.slice(1)) +
+                  theme.fg(statusColor(item), "● ") +
+                  theme.fg(
+                    identity(item) === selected ? "accent" : "text",
+                    `${assignmentHeadline(item)}${historical(item) ? " · history" : ""}`,
+                  ),
+                theme.fg("text", `  ${continuation}`) +
+                  theme.fg(statusColor(item), lineText(item.profile)) +
+                  theme.fg("text", ` · ${item.status}`),
+              ];
+            }),
         ];
         const right = [
           theme.fg("accent", lineText(selectedLabel)),
@@ -674,21 +449,25 @@ export function createHub(
               ),
           );
       } else {
-        const suffix = child ? ` #${child.generation} · ${index + 1}/${children.length}` : "";
+        const suffix = child ? ` · ${index + 1}/${children.length}` : "";
         pane.push(
-          theme.fg(
-            "accent",
-            child
-              ? `> └ ${truncateToWidth(agentLabel(child), Math.max(0, inner - 4 - visibleWidth(suffix)))}${suffix}`
-              : "No selection",
-          ),
+          child
+            ? theme.fg("accent", ">") +
+                theme.fg("text", " └ ") +
+                theme.fg(statusColor(child), "● ") +
+                theme.fg(
+                  "accent",
+                  `${truncateToWidth(agentLabel(child), Math.max(0, inner - 6 - visibleWidth(suffix)))}${suffix}`,
+                )
+            : theme.fg("accent", "No selection"),
         );
         if (paneHeight > 1)
           pane.push(
-            theme.fg(
-              child ? statusColor(child) : "muted",
-              `${tab.toUpperCase()}${child ? ` · ${child.profile} · ${child.status}` : ""}`,
-            ),
+            child
+              ? theme.fg("text", `${tab.toUpperCase()} · `) +
+                  theme.fg(statusColor(child), lineText(child.profile)) +
+                  theme.fg("text", ` · ${child.status}`)
+              : theme.fg("muted", tab.toUpperCase()),
           );
         if (paneHeight > 2)
           pane.push(
@@ -798,6 +577,7 @@ export function createHub(
           child &&
           !child.disposed &&
           child.status !== "draining" &&
+          child.status !== "pausing" &&
           (data === "x" || child.status === "running")
         ) {
           mode = data === "s" ? "steer" : data === "m" ? "message" : "stop";