@fradser/pi-agent-teams 0.7.0 → 0.8.2

package/src/ui.ts

@@ -1,24 +1,39 @@
 /** Passive team widget and the /agent-teams management console
  * (session teammates + persistent agent roles + board). */
 
-import { truncateToWidth, Key, matchesKey } from "@earendil-works/pi-tui";
-import { createPiThemeStyle, PI_SPINNER_FRAMES, PI_SPINNER_INTERVAL_MS } from "@fradser/pi-kit";
+import { truncateToWidth, Key, matchesKey, fuzzyFilter } from "@earendil-works/pi-tui";
+import {
+  createPiThemeStyle,
+  createSearchPicker,
+  modelLabel,
+  modelSearchText,
+  PI_SPINNER_FRAMES,
+  PI_SPINNER_INTERVAL_MS,
+  notifyPi,
+  renderPiPanel,
+  renderPiWidgetRow,
+  sortModels,
+  type SearchPicker,
+} from "@fradser/pi-kit";
 import type { ExtensionUIContext } from "@earendil-works/pi-coding-agent";
 import {
   clampConsoleScroll, consoleScrollRange, maxConsoleBody, scrollConsoleDetail, wrapConsoleDetail,
 } from "./console-viewport.ts";
 import { fitTeammateRow, formatTeammateLabel, runningTeammateActivity } from "./activity.ts";
-import { discoverAgents, resolveAgent, type AgentDefinition } from "./agents.ts";
-import { getState, getTeammate, listTasks, listTeammates, livingTeammates } from "./state.ts";
+import { MODEL_INHERIT_ALIAS, discoverAgents, resolveAgent, type AgentDefinition } from "./agents.ts";
+import { getPeerDeliveryState, getState, getTeammate, getTeamDefaultModel, listTasks, listTeammates, livingTeammates, setTeamDefaultModel } from "./state.ts";
 import {
+  currentLeaderModelRef,
   ensureLivePoll,
   formatSilenceDuration,
+  resolveSpawnModel,
   runtimeDirPath,
   shutdownTeammate,
-  STALL_NOTICE_MS,
   stallSilenceMs,
+  stallThresholdMs,
 } from "./team-machine.ts";
 import type { Teammate } from "./types.ts";
+import { mapPickerKey } from "./picker-keys.ts";
 import { inboxPath } from "./statefile.ts";
 import * as fs from "node:fs";
 import * as path from "node:path";
@@ -79,13 +94,13 @@ export function ensureTeamWidget(ctx?: { ui?: ExtensionUIContext; mode?: string
         if (working.length === 0) return [];
         const lines: string[] = [];
         for (const teammate of working) {
-          lines.push(fitTeammateRow(
+          lines.push(renderPiWidgetRow(fitTeammateRow(
             PI_SPINNER_FRAMES[spinnerFrame],
             style.fg(colorFor(teammate.name), teammate.name),
             runningTeammateActivity(teammate) + stallSuffix(teammate),
-            width,
+            Math.max(1, width - 1),
             (activity) => theme.bold(style.fg("accent", activity)),
-          ));
+          ), width, truncateToWidth));
         }
         return lines;
       },
@@ -99,11 +114,12 @@ function isWorking(teammate: { status: string }): boolean {
   return teammate.status === "working" || teammate.status === "starting";
 }
 
-/** Silence marker appended once a working teammate passes the stall notice threshold. */
+/** Silence marker appended once a working teammate passes its stall threshold. */
 function stallSuffix(teammate: Teammate): string {
   if (!isWorking(teammate)) return "";
   const silence = stallSilenceMs(teammate);
-  return silence !== undefined && STALL_NOTICE_MS > 0 && silence >= STALL_NOTICE_MS
+  const threshold = stallThresholdMs(teammate);
+  return silence !== undefined && threshold > 0 && silence >= threshold
     ? ` · stalled for ${formatSilenceDuration(silence)}`
     : "";
 }
@@ -122,6 +138,7 @@ function cap(text: string | undefined, maxBytes = 2000): string {
 // ── Detail builders ───────────────────────────────────────────────
 
 interface PeerMailLine {
+  id: string;
   direction: "sent" | "received";
   counterpart: string;
   subject: string;
@@ -155,6 +172,7 @@ function readPeerMail(teammateName: string): PeerMailLine[] {
         if (!message.id || !message.from || !message.subject) continue;
         if (message.from !== teammateName && recipient !== teammateName) continue;
         lines.push({
+          id: message.id,
           direction: message.from === teammateName ? "sent" : "received",
           counterpart: message.from === teammateName ? recipient : message.from,
           subject: message.subject,
@@ -173,7 +191,8 @@ function indent(text: string): string[] {
   return text.split("\n").map((line) => `  ${line}`);
 }
 
-function displaySource(sourcePath: string): string {
+function displaySource(sourcePath: string | undefined): string {
+  if (!sourcePath) return "(in-memory session role)";
   const relative = path.relative(process.cwd(), sourcePath);
   return relative.startsWith("..") ? sourcePath : `./${relative}`;
 }
@@ -188,6 +207,8 @@ function buildTeammateDetail(name: string): string[] {
     "== teammate ==",
     `  Status: ${teammate.status}${teammate.currentTaskId ? ` | Task: ${teammate.currentTaskId}` : ""}`,
     `  Spawn: ${teammate.pid > 0 ? `pid ${teammate.pid}` : "pid unknown"} | Isolation: ${teammate.isolation}`,
+    ...(teammate.model ? [`  Launch model: ${teammate.model}`] : []),
+    ...(teammate.tools?.length ? [`  Tools: ${teammate.tools.join(", ")}`] : []),
     `  Created: ${new Date(teammate.createdAt).toLocaleString()}`,
   ];
   if (teammate.stoppedAt) lines.push(`  Stopped: ${new Date(teammate.stoppedAt).toLocaleString()}`);
@@ -203,7 +224,8 @@ function buildTeammateDetail(name: string): string[] {
   lines.push("", `== reports to leader (${reports.length}) ==`);
   if (reports.length === 0) lines.push("  (none)");
   for (const report of reports) {
-    lines.push(`  -> [${report.subject}] ${new Date(report.timestamp).toLocaleString()}`, ...indent(cap(report.body)), "");
+    const archive = report.archived ? " · archived after shutdown" : "";
+    lines.push(`  ->${archive} [${report.subject}] ${new Date(report.timestamp).toLocaleString()}`, ...indent(cap(report.body)), "");
   }
 
   const peer = readPeerMail(name);
@@ -211,7 +233,10 @@ function buildTeammateDetail(name: string): string[] {
   if (peer.length === 0) lines.push("  (none)");
   for (const mail of peer) {
     const arrow = mail.direction === "sent" ? `to @${mail.counterpart}` : `from @${mail.counterpart}`;
-    lines.push(`  ${arrow} [${mail.subject}] ${new Date(mail.timestamp).toLocaleString()}`, ...indent(cap(mail.body)), "");
+    const routing = mail.direction === "sent"
+      ? getPeerDeliveryState(mail.id) ?? "written"
+      : "received";
+    lines.push(`  ${arrow} · ${routing} [${mail.subject}] ${new Date(mail.timestamp).toLocaleString()}`, ...indent(cap(mail.body)), "");
   }
   return lines;
 }
@@ -225,7 +250,8 @@ function buildTaskDetail(taskId: string): string[] {
     `[${task.id}] ${task.subject}`,
     "",
     "== task ==",
-    `  Status: ${task.status}${task.claimedBy ? ` (@${task.claimedBy})` : ""}`,
+    `  Status: ${task.status}${task.claimedBy ? ` (@${task.claimedBy})` : ""}${task.supersededBy ? ` → superseded by ${task.supersededBy}` : ""}`,
+    ...((task.resources?.length ?? 0) > 0 ? [`  Resources: ${task.resources.join(", ")}`] : []),
     ...(task.description ? [`  Description:`, ...indent(cap(task.description))] : []),
     ...(task.dependsOn.length > 0
       ? [`  Depends on: ${task.dependsOn.join(", ")} (${task.dependsOn.every((dep) => byId.get(dep)?.status === "completed") ? "met" : "unmet"})`]
@@ -242,13 +268,13 @@ function buildRoleDetail(name: string): string[] {
   const def = resolveAgent(name);
   if (!def) return ["(agent definition not found)"];
   return [
-    `@${def.name} (${def.scope})${def.gitManaged ? " [git-managed]" : " [local]"}`,
+    `@${def.name} (${def.scope})${def.scope === "session" ? " [in-memory]" : def.gitManaged ? " [git-managed]" : " [local]"}`,
     "",
     "== role ==",
     `  Source: ${displaySource(def.source)}`,
     `  Description: ${def.description || "(none)"}`,
     `  Tools: ${def.tools.join(", ") || "(role defaults)"}`,
-    `  Model: ${def.model ?? "(session default)"}`,
+    `  Model: ${describeSpawnModel(def.model)}`,
     `  Verify: ${def.verify ?? "(none)"}`,
     `  Worktree: ${def.worktree ? "true" : "false"}`,
     `  Living instances: ${livingTeammates().filter((teammate) => teammate.agent === def.name).length}`,
@@ -258,6 +284,17 @@ function buildRoleDetail(name: string): string[] {
   ];
 }
 
+/** Human-readable effective launch model for a role definition. */
+function describeSpawnModel(pinned: string | undefined): string {
+  const resolved = resolveSpawnModel(pinned, getTeamDefaultModel(), currentLeaderModelRef());
+  if (pinned && pinned.trim().toLowerCase() !== MODEL_INHERIT_ALIAS) return pinned.trim();
+  if (resolved.model) {
+    const suffix = pinned ? "inherit -> " : "[team default] ";
+    return `${suffix}${resolved.model}`;
+  }
+  return pinned ? "inherit -> (unavailable at spawn)" : "(Pi default)";
+}
+
 // ── Full-screen console ───────────────────────────────────────────
 
 type ConsolePage = "roster" | "board";
@@ -292,11 +329,15 @@ function wheelDelta(data: string): number | undefined {
 const LIST_CHROME_LINES = 5;
 
 /** Full-screen Team Console — owns input via ctx.ui.custom. Pages: roster /
- * board; each row opens a scrolling detail view. */
-export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void> {
+ * board; each row opens a scrolling detail view; m opens the searchable
+ * teammate-model picker. */
+export function openTeamConsole(ctx: {
+  ui: ExtensionUIContext;
+  modelRegistry?: { getAvailable(): Array<{ provider: string; id: string; name?: string }> };
+}): Promise<void> {
   return ctx.ui.custom<void>((tui, theme, _keybindings, done) => {
     let page: ConsolePage = "roster";
-    let mode: "list" | "detail" = "list";
+    let mode: "list" | "detail" | "picker" = "list";
     let selectedRoster = 0;
     let selectedBoard = 0;
     let listOffset = 0;
@@ -341,6 +382,109 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
 
     const style = createPiThemeStyle(theme);
 
+    // ── Teammate-model picker (type-to-filter) ──────────────────
+    let picker: SearchPicker<{ provider: string; id: string; name?: string }> | undefined;
+    let pickerOffset = 0;
+
+    /** Pinned pseudo-entry so clearing never collides with typed search text. */
+    const CLEAR_TEAM_MODEL_ENTRY: { provider: string; id: string; name?: string } = {
+      provider: "",
+      id: "__clear-team-default__",
+      name: "Use Pi default",
+    };
+    const isClearEntry = (model: { id: string } | undefined): boolean =>
+      model?.id === CLEAR_TEAM_MODEL_ENTRY.id;
+
+    const openModelPicker = (): void => {
+      const models = sortModels([...(ctx.modelRegistry?.getAvailable() ?? [])]);
+      if (models.length === 0) return;
+      picker = createSearchPicker([CLEAR_TEAM_MODEL_ENTRY, ...models], {
+        filter: fuzzyFilter,
+        getText: (model) => (isClearEntry(model) ? "clear team default use pi default" : modelSearchText(model)),
+      });
+      pickerOffset = 0;
+      mode = "picker";
+      requestRender();
+    };
+
+    const closeModelPicker = (): void => {
+      picker = undefined;
+      mode = "list";
+      requestRender();
+    };
+
+    const PICKER_CHROME_LINES = 7;
+    const pickerViewport = (): number => Math.max(1, tui.terminal.rows - PICKER_CHROME_LINES);
+
+    const renderPicker = (width: number): string[] => {
+      if (!picker) return [];
+      const results = picker.results();
+      const selected = picker.selectedIndex();
+      if (selected < pickerOffset) pickerOffset = selected;
+      else if (selected >= pickerOffset + pickerViewport()) pickerOffset = selected - pickerViewport() + 1;
+      pickerOffset = clampConsoleScroll(pickerOffset, results.length, pickerViewport());
+      const leaderRef = currentLeaderModelRef();
+      const teamDefault = getTeamDefaultModel();
+      const rows = results.slice(pickerOffset, pickerOffset + pickerViewport()).map((model, index) => {
+        const absolute = pickerOffset + index;
+        const marker = absolute === selected ? style.accent("❯ ") : "  ";
+        if (isClearEntry(model)) {
+          return truncateToWidth(`${marker}${theme.fg("warning", "✕ clear team default")} ${style.dim("· use Pi default")}`, Math.max(10, width - 1));
+        }
+        const tags = [
+          modelLabel(model) === leaderRef ? style.dim("(leader)") : "",
+          modelLabel(model) === teamDefault ? style.success("(default)") : "",
+        ].filter(Boolean).join(" ");
+        const display = model.name && model.name !== model.id ? style.dim(model.name) : "";
+        return truncateToWidth(`${marker}${theme.fg("customMessageText", modelLabel(model))} ${display} ${tags}`.replace(/\s+$/g, ""), Math.max(10, width - 1));
+      });
+      return renderPiPanel({
+        width,
+        style,
+        fit: truncateToWidth,
+        title: `teammate model  ${teamDefault ?? "auto (Pi default)"}`,
+        body: [
+          "",
+          `${style.accent("❯ ")}${picker.query()}▏  ${style.dim(`${results.length} models`)}`,
+          ...rows,
+          "",
+        ],
+        footer: "type to filter · ↑↓ select · enter set/clear · esc cancel",
+      });
+    };
+
+    function handlePickerInput(data: string): void {
+      if (!picker) {
+        mode = "list";
+        return;
+      }
+      const action = mapPickerKey(data);
+      if (!action) return;
+      if (action.kind === "cancel") {
+        closeModelPicker();
+        return;
+      }
+      if (action.kind === "confirm") {
+        const picked = picker.selected();
+        // An empty filtered list must not clobber the current default.
+        if (!picked) return;
+        if (isClearEntry(picked)) {
+          setTeamDefaultModel(undefined);
+          notifyPi(ctx.ui, "Teammate model cleared — Pi picks its default", "info");
+        } else {
+          setTeamDefaultModel(modelLabel(picked));
+          notifyPi(ctx.ui, `Teammate model set to ${modelLabel(picked)} for this session`, "info");
+        }
+        closeModelPicker();
+        return;
+      }
+      if (action.kind === "up") picker.up();
+      else if (action.kind === "down") picker.down();
+      else if (action.kind === "backspace") picker.backspace();
+      else picker.type(action.text);
+      requestRender();
+    }
+
     const currentRows = (): ConsoleRow[] =>
       page === "roster"
         // The console is the management surface: idle and stopped stay
@@ -364,7 +508,7 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
     const headerLine = (): string => {
       const alive = livingTeammates();
       const tasks = listTasks();
-      return `team  ${alive.length} alive · ${alive.filter(isWorking).length} working · ${getRoles().length} roles · board ${tasks.filter((task) => task.status === "pending").length}p/${tasks.filter((task) => task.status === "claimed").length}c/${tasks.filter((task) => task.status === "completed").length}d · ${page}`;
+      return `team  ${alive.length} alive · ${alive.filter(isWorking).length} working · ${getRoles().length} roles · board ${tasks.filter((task) => task.status === "pending").length}p/${tasks.filter((task) => task.status === "claimed").length}c/${tasks.filter((task) => task.status === "completed").length}d/${tasks.filter((task) => task.status === "superseded").length}s · model ${getTeamDefaultModel() ?? "auto"} · ${page}`;
     };
 
     interface ContentLine {
@@ -396,7 +540,8 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
       const name = theme.bold(theme.fg(colorFor(key), `@${key}`));
       const live = livingTeammates().filter((teammate) => teammate.agent === key && teammate.status !== "stopped").length;
       const liveTag = live > 0 ? style.success(`[${live} live]`) : style.dim("[0 live]");
-      return `${marker}${name} ${style.dim(def.scope)} ${style.dim(displaySource(def.source))} ${liveTag}`;
+      const provenance = def.scope === "session" ? "in-memory" : displaySource(def.source);
+      return `${marker}${name} ${style.dim(def.scope)} ${style.dim(provenance)} ${liveTag}`;
     };
 
     const taskRowText = (key: string, selected: boolean): string => {
@@ -406,9 +551,11 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
       const holder = task.claimedBy ? style.dim(` @${task.claimedBy}`) : "";
       const statusText = task.status === "completed"
         ? style.success("✓")
-        : task.status === "claimed"
-          ? theme.fg("warning", "◐ claimed")
-          : style.dim("○ pending");
+        : task.status === "superseded"
+          ? style.dim("⊘ superseded")
+          : task.status === "claimed"
+            ? theme.fg("warning", "◐ claimed")
+            : style.dim("○ pending");
       return `${marker}${label} ${statusText} ${theme.fg("customMessageText", task.subject)}${holder}`;
     };
 
@@ -431,7 +578,7 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
       }
       teammates.forEach((row, index) => lines.push({ text: teammateRowText(row.key, index === selection), select: index }));
       lines.push({ text: "", select: -1 });
-      lines.push({ text: style.dim("== agent roles (persistent definitions) =="), select: -1 });
+      lines.push({ text: style.dim("== agent roles (persistent and session definitions) =="), select: -1 });
       if (roles.length === 0) {
         lines.push({ text: style.dim("No agent definitions discovered."), select: -1 });
       }
@@ -442,7 +589,6 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
     const listViewport = (): number => Math.max(1, tui.terminal.rows - LIST_CHROME_LINES);
 
     const renderList = (width: number): string[] => {
-      const border = style.border("─".repeat(Math.max(1, width)));
       const content = buildContent();
       const viewport = listViewport();
       const selection = currentPageSelection(currentRows().length);
@@ -455,18 +601,16 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
       const footer = confirmName
         ? theme.fg("warning", `Shut down @${confirmName}? Its claimed task returns to the board · y yes / n no`)
         : style.dim(page === "roster"
-            ? "↑↓ select · enter open · tab board · x shutdown · esc/q close"
+            ? "↑↓ select · enter open · tab board · x shutdown · m model · esc/q close"
             : "↑↓ select · enter open · tab roster · esc/q close");
-      return [
-        border,
-        style.accent(truncateToWidth(headerLine(), width)),
-        "",
-        ...content.slice(listOffset, listOffset + viewport)
-          .map((line) => truncateToWidth(line.text, Math.max(10, width - 1))),
-        "",
-        truncateToWidth(footer, Math.max(10, width - 1)),
-        border,
-      ];
+      return renderPiPanel({
+        width,
+        style,
+        fit: truncateToWidth,
+        title: headerLine(),
+        body: content.slice(listOffset, listOffset + viewport).map((line) => line.text),
+        footer,
+      });
     };
 
     const currentPageSelection = (rowCount: number): number => {
@@ -478,20 +622,16 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
       detailKind === "task" ? `[${detailKey}]` : detailKind === "role" ? `@${detailKey} · role` : `@${detailKey}`;
 
     const renderDetail = (width: number): string[] => {
-      const border = style.border("─".repeat(Math.max(1, width)));
       const source = detailSource();
       const detail = windowLines(source, width);
-      const footer = style.dim(`  ${detail.range} · ↑↓ scroll · pgup/pgdn page · home/end jump · esc back · q close`);
-      const lines = [
-        border,
-        style.accent(truncateToWidth(`agent-teams  ${detailTitle()}`, width)),
-        "",
-        ...detail.lines.map((line) => `  ${line}`),
-        "",
-        footer,
-        border,
-      ];
-      return lines.map((line) => truncateToWidth(line, Math.max(10, width - 1)));
+      return renderPiPanel({
+        width,
+        style,
+        fit: truncateToWidth,
+        title: `agent-teams  ${detailTitle()}`,
+        body: detail.lines,
+        footer: `${detail.range} · ↑↓ scroll · pgup/pgdn page · home/end jump · esc back · q close`,
+      });
     };
 
     const detailSource = (): string[] => {
@@ -501,8 +641,12 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
     };
 
     return {
-      render: (width) => (mode === "list" ? renderList(width) : renderDetail(width)),
+      render: (width) => (mode === "picker" ? renderPicker(width) : mode === "detail" ? renderDetail(width) : renderList(width)),
       handleInput: (data: string) => {
+        if (mode === "picker") {
+          handlePickerInput(data);
+          return;
+        }
         if (mode !== "list") {
           handleDetailInput(data);
           return;
@@ -546,6 +690,14 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
           }
           return;
         }
+        if (data === "m" || data === "M") {
+          if (ctx.modelRegistry && ctx.modelRegistry.getAvailable().length > 0) {
+            openModelPicker();
+          } else {
+            notifyPi(ctx.ui, "No models are available in the model registry.", "warning");
+          }
+          return;
+        }
         const wheel = wheelDelta(data);
         if (wheel !== undefined) {
           scrollList(wheel * 3);
@@ -610,8 +762,8 @@ export function openTeamConsole(ctx: { ui: ExtensionUIContext }): Promise<void>
 
 async function shutdownFromConsole(ctx: { ui: ExtensionUIContext }, name: string): Promise<void> {
   const result = await shutdownTeammate(name);
-  if (result.ok) ctx.ui.notify(result.body, "info");
-  else ctx.ui.notify(result.error, "error");
+  if (result.ok) notifyPi(ctx.ui, result.body, "info");
+  else notifyPi(ctx.ui, result.error, "error");
 }
 
 function clampIndex(index: number, rowCount: number): number {