pi-better-subagents 0.14.1 → 0.15.0

package/docs/usage.md

@@ -67,7 +67,7 @@ is present; without this extension it uses adaptive guidance.
   `subagent_spawn` starts a detached `pi -p` child and returns immediately,
   leaving the session free for the human. When the child finishes, it sends a
   lightweight trigger; the foreground calls `subagent_result` and presents the
-  result (as a `followUp`, never cutting into work in progress). The foreground
+  result (as a `followUp` by default; optional busy steering is described below). The foreground
   is nudged once, at completion — never on a wait/poll loop.
 - **Subagents are autonomous; communication is one-way (parent → child).** The
   parent front-loads everything the child needs into the spawn; the child runs to
@@ -247,7 +247,7 @@ given.
   its output is durable in a log file.
 - **Completions trigger one batched turn that fetches durable results.** Ordinary
   terminal callbacks stay in the harness queue while the foreground agent is
-  busy. Once Pi is idle, they accumulate for 100 ms and share one
+  busy by default. Once Pi is idle, they accumulate for 100 ms and share one
   `pi.sendMessage(..., { deliverAs: "followUp", triggerTurn: true })` notification
   with background-task completions from the same Pi host. Each row contains only
   source, id, label, terminal status, and the `subagent_result` lookup. Full
@@ -260,6 +260,23 @@ given.
   pending records retryable, and `/reload` recovers records explicitly marked
   pending. `callback:false` sends no model message; read the durable result later
   with `subagent_result`.
+- **Optional busy steering.** In `/harness-settings`, set **Completions while
+  busy** to **Steer active run** to receive ordinary completions during
+  foreground work. Changes autosave to the current session branch and apply
+  immediately; **Ctrl+S** saves the choice as the user default for future
+  sessions, including standalone callback packages. **Wait until idle** is the
+  default. The former `PI_BETTER_CALLBACK_WHILE_BUSY` environment variable is
+  no longer supported. Completions spanning multiple accumulation windows during
+  one tool call stay together until the final active tool ends, then share one
+  compact `steer` notification without triggering a new run. Parallel and nested
+  tools are tracked together by both extensions. While busy with no active tool,
+  the normal accumulation window applies. Manual compaction and branch
+  summarization hold callbacks until an agent run starts or Pi becomes idle.
+  Idle delivery remains
+  `{ deliverAs: "followUp", triggerTurn: true }`. The same byte budget, queued overflow,
+  dedupe, delivery receipts, retry, ownership checks, and `callback:false` rules
+  apply to both modes; a completion handed off by steer is not sent again at
+  idle. Urgent health/failure alerts are unchanged.
 - **The prompt guidelines forbid polling.** The foreground agent is told, in the
   tool guidelines, that spawning is done and it must not loop on `output`/`result`
   or sleep to wait.

package/index.ts

@@ -13,7 +13,7 @@
 
 import { execSync } from "node:child_process";
 import { writeFileSync, mkdirSync, renameSync, statSync } from "node:fs";
-import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import type { ExtensionAPI, ExtensionCommandContext, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import * as PiCodingAgent from "@earendil-works/pi-coding-agent";
 import * as PiTui from "@earendil-works/pi-tui";
 import { CustomEditor } from "@earendil-works/pi-coding-agent";
@@ -2283,6 +2283,27 @@ export default function (pi: ExtensionAPI) {
         },
         enrich: createLaunchEnricher(),
     });
+    const openSettings = async (ctx: ExtensionCommandContext): Promise<void> => {
+        if (ctx.mode === "tui" && ctx.hasUI && typeof ctx.ui.custom === "function") {
+            await ctx.ui.custom<void>((tui, theme, _kb, done) => createSubagentSettingsPage(theme, {
+                get: () => ({
+                    mode: activeDelegationMode(), maxConcurrent: activeConcurrencyCap(),
+                    modeSource: delegationOverride ? "session" : "config",
+                    capSource: capOverride !== undefined ? "session" : "config",
+                    defaultMode: normalizeDelegationMode(loadConfig().delegationMode),
+                    defaultCap: normalizeConcurrencyCap(loadConfig().maxConcurrent),
+                }),
+                changeMode, changeCap, save: saveSettings, reset: resetSettings,
+            }, () => { refreshBackgroundWorkNavigator(ctx); tui.requestRender(); }, () => done()));
+        } else {
+            ctx.ui.notify(`Delegation mode: ${activeDelegationMode()}. Concurrent subagents: ${activeConcurrencyCap()}. Use /subagents mode <mode> or /subagents cap <number>; /subagents save persists both.`, "info");
+        }
+    };
+    const contribution = { id: "subagents", label: "Subagents", command: "/subagents settings", open: openSettings };
+    const registerSettings = () => pi.events?.emit("harness-settings:register", contribution);
+    const unsubscribeSettingsRequest = pi.events?.on?.("harness-settings:request", registerSettings);
+    registerSettings();
+
     if (typeof pi.registerCommand === "function") {
         agentOperations.registerCommands(pi);
         pi.registerCommand("subagents", {
@@ -2291,20 +2312,7 @@ export default function (pi: ExtensionAPI) {
             async handler(args, ctx) {
                 const tokens = args.trim().split(/\s+/).filter(Boolean);
                 if (tokens.length === 0 || (tokens.length === 1 && tokens[0] === "settings")) {
-                    if (ctx.mode === "tui" && ctx.hasUI && typeof ctx.ui.custom === "function") {
-                        await ctx.ui.custom<void>((tui, theme, _kb, done) => createSubagentSettingsPage(theme, {
-                            get: () => ({
-                                mode: activeDelegationMode(), maxConcurrent: activeConcurrencyCap(),
-                                modeSource: delegationOverride ? "session" : "config",
-                                capSource: capOverride !== undefined ? "session" : "config",
-                                defaultMode: normalizeDelegationMode(loadConfig().delegationMode),
-                                defaultCap: normalizeConcurrencyCap(loadConfig().maxConcurrent),
-                            }),
-                            changeMode, changeCap, save: saveSettings, reset: resetSettings,
-                        }, () => { refreshBackgroundWorkNavigator(ctx); tui.requestRender(); }, () => done()));
-                    } else {
-                        ctx.ui.notify(`Delegation mode: ${activeDelegationMode()}. Concurrent subagents: ${activeConcurrencyCap()}. Use /subagents mode <mode> or /subagents cap <number>; /subagents save persists both.`, "info");
-                    }
+                    await openSettings(ctx);
                     return;
                 }
                 const requested = tokens[1];
@@ -2346,18 +2354,23 @@ export default function (pi: ExtensionAPI) {
     // ---- live-status lifecycle -----------------------------------------
     pi.on("agent_start", async (_event, ctx) => {
         setCallbackBatchContext(pi, ctx);
+        getCallbackBatcher(pi).setForegroundRunning(true);
         uiCtx = ctx;
         mainAgentStartedAt = Date.now();
         mainAgentTools.clear();
         refreshBackgroundWorkNavigator(ctx);
     });
 
+    pi.on("agent_end", () => { getCallbackBatcher(pi).setForegroundRunning(false); });
+
     pi.on("tool_execution_start", async (event, ctx) => {
+        getCallbackBatcher(pi).toolStarted(event.toolCallId);
         mainAgentTools.set(event.toolCallId, event.toolName);
         refreshBackgroundWorkNavigator(ctx);
     });
 
     pi.on("tool_execution_end", async (event, ctx) => {
+        await getCallbackBatcher(pi).toolEnded(event.toolCallId);
         mainAgentTools.delete(event.toolCallId);
         refreshBackgroundWorkNavigator(ctx);
     });
@@ -2378,6 +2391,7 @@ export default function (pi: ExtensionAPI) {
     });
 
     pi.on("agent_settled", async (_event, ctx) => {
+        getCallbackBatcher(pi).setForegroundRunning(false);
         setCallbackBatchContext(pi, ctx);
         mainAgentStartedAt = undefined;
         mainAgentTools.clear();
@@ -2385,6 +2399,7 @@ export default function (pi: ExtensionAPI) {
     });
 
     pi.on("session_tree", async (_event, ctx) => {
+        setCallbackBatchContext(pi, ctx);
         agentSessionSettings.restore(ctx);
         try { noteCatalogHost(catalogHostFrom(ctx)); } catch { /* catalog inspection stays undecided */ }
         restoreDelegationMode(ctx);
@@ -2446,6 +2461,7 @@ export default function (pi: ExtensionAPI) {
 
     // Tear down the timer and clear the widget when the session ends.
     pi.on("session_shutdown", async (_event, ctx) => {
+        if (typeof unsubscribeSettingsRequest === "function") unsubscribeSettingsRequest();
         // The foreground Pi session owns this work. Stop current-session
         // running/orphaned children before dropping the session origin, so they
         // cannot become live process groups with no coordinator.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-better-subagents",
-  "version": "0.14.1",
+  "version": "0.15.0",
   "description": "Pi extension for detached, sandboxed subagent runs that keep the foreground session free.",
   "license": "MIT",
   "type": "module",

package/shared-callback-batcher.ts

@@ -1,6 +1,126 @@
 // Generated from packages/callback-batcher/index.ts. Do not edit directly.
+import { closeSync, mkdirSync, openSync, readFileSync, renameSync, rmSync, writeFileSync } from "node:fs";
+import { randomUUID } from "node:crypto";
+import { join } from "node:path";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
+
 export type CallbackSource = "subagent" | "background-task";
 export type CallbackDetailTool = "subagent_result" | "bg_task_status";
+export type CallbackDeliveryMode = "hold" | "steer";
+
+export interface CallbackSettingsContext {
+  sessionManager?: { getBranch?(): readonly unknown[]; getSessionId?(): string };
+  ui?: { notify(message: string, type: "error"): void };
+}
+
+export interface CallbackSettingsSeams {
+  agentDir?: () => string;
+}
+
+export interface CallbackSettings {
+  mode: CallbackDeliveryMode;
+  source: "session" | "default";
+}
+
+export const CALLBACK_SETTINGS_ENTRY = "pi-better-callback-settings";
+const CALLBACK_PREFERENCES_FILE = "pi-better-callback-preferences.json";
+
+function isDeliveryMode(mode: unknown): mode is CallbackDeliveryMode {
+  return mode === "hold" || mode === "steer";
+}
+
+function validCallbackPreferences(value: unknown): value is { version: 1; mode: CallbackDeliveryMode } {
+  if (!value || typeof value !== "object" || Array.isArray(value)) return false;
+  const data = value as Record<string, unknown>;
+  return Object.keys(data).length === 2 && Object.hasOwn(data, "version") && Object.hasOwn(data, "mode")
+    && data.version === 1 && isDeliveryMode(data.mode);
+}
+
+function callbackPreferencesPath(seams: CallbackSettingsSeams): string {
+  return join((seams.agentDir ?? getAgentDir)(), "extensions", CALLBACK_PREFERENCES_FILE);
+}
+
+/** Branch history is authoritative; abandoned branches and invalid entries are ignored. */
+export function getCallbackSettings(
+  ctx: CallbackSettingsContext,
+  seams: CallbackSettingsSeams = {},
+): CallbackSettings {
+  const branch = ctx.sessionManager?.getBranch?.() ?? [];
+  for (let i = branch.length - 1; i >= 0; i--) {
+    const entry = branch[i] as { type?: unknown; customType?: unknown; data?: unknown } | null;
+    if (entry?.type === "custom" && entry.customType === CALLBACK_SETTINGS_ENTRY && validCallbackPreferences(entry.data)) {
+      return { mode: entry.data.mode, source: "session" };
+    }
+  }
+  return getSessionDefault(ctx, seams);
+}
+
+function getSessionDefault(ctx: CallbackSettingsContext, seams: CallbackSettingsSeams): CallbackSettings {
+  const sessionId = ctx.sessionManager?.getSessionId?.();
+  const defaults = globalState().sessionDefaults ??= new WeakMap<object, { sessionId?: string; mode: CallbackDeliveryMode }>();
+  const initial = ctx.sessionManager ? defaults.get(ctx.sessionManager) : undefined;
+  if (initial && initial.sessionId === sessionId) return { mode: initial.mode, source: "default" };
+  const snapshot = (mode: CallbackDeliveryMode): CallbackSettings => {
+    if (ctx.sessionManager) defaults.set(ctx.sessionManager, { sessionId, mode });
+    return { mode, source: "default" };
+  };
+  const path = callbackPreferencesPath(seams);
+  let text: string;
+  try {
+    text = readFileSync(path, "utf8");
+  } catch (error) {
+    if ((error as NodeJS.ErrnoException).code === "ENOENT") return snapshot("hold");
+    throw new Error(`Cannot read callback default at ${path}: ${String(error)}`, { cause: error });
+  }
+  try {
+    const data: unknown = JSON.parse(text);
+    if (!validCallbackPreferences(data)) throw new Error('Expected { version: 1, mode: "hold" | "steer" }');
+    return snapshot(data.mode);
+  } catch (error) {
+    throw new Error(`Invalid callback default at ${path}: ${String(error)}`, { cause: error });
+  }
+}
+
+/** Saves only the future-session default; never changes the active branch or batcher. */
+export function saveCallbackDefault(mode: CallbackDeliveryMode, seams: CallbackSettingsSeams = {}): void {
+  if (!isDeliveryMode(mode)) throw new Error("Invalid callback delivery mode. Nothing saved.");
+  const dir = join((seams.agentDir ?? getAgentDir)(), "extensions");
+  const path = join(dir, CALLBACK_PREFERENCES_FILE);
+  const tmp = join(dir, `.${CALLBACK_PREFERENCES_FILE}.${randomUUID()}.tmp`);
+  let created = false;
+  try {
+    mkdirSync(dir, { recursive: true });
+    const fd = openSync(tmp, "wx", 0o600);
+    created = true;
+    try { writeFileSync(fd, `${JSON.stringify({ version: 1, mode })}\n`); }
+    finally { closeSync(fd); }
+    renameSync(tmp, path);
+  } catch (error) {
+    throw new Error(`Cannot save callback default at ${path}: ${String(error)}`, { cause: error });
+  } finally {
+    if (created) rmSync(tmp, { force: true });
+  }
+}
+
+/** Persist first so a failed append cannot become a transient delivery-mode change. */
+export function changeCallbackSetting(
+  host: CallbackBatchHost & { appendEntry(customType: string, data: unknown): void },
+  ctx: CallbackSettingsContext & { isIdle(): boolean },
+  mode: CallbackDeliveryMode,
+): void {
+  if (!isDeliveryMode(mode)) throw new Error("Invalid callback delivery mode. Nothing changed.");
+  const data = { version: 1, mode };
+  try { host.appendEntry(CALLBACK_SETTINGS_ENTRY, data); }
+  catch (error) {
+    // Pi advances its in-memory branch before disk persistence. Invalidate
+    // our payload so that failed entry cannot become a restored setting.
+    data.version = 0;
+    throw error;
+  }
+  const batcher = bindCallbackBatcher(host, ctx);
+  batcher.setWhileBusy(mode);
+  batcher.setAvailability(() => ctx.isIdle());
+}
 
 export interface CallbackBatchHost {
   /** Older Pi shares this object; newer Pi supplies per-extension wrappers. */
@@ -70,6 +190,8 @@ export interface CallbackBatcherOptions {
   retryMs?: number;
   /** Ordinary callbacks stay here, not in Pi's follow-up queue, while unavailable. */
   isAvailable?: () => boolean;
+  /** Internal initial mode. Session/default settings are restored with context. */
+  whileBusy?: CallbackDeliveryMode;
   /** UTF-8 byte cap for one sendMessage payload. Defaults to 2 KiB. */
   maxBytes?: number;
 }
@@ -81,6 +203,14 @@ export interface CallbackBatcher {
   setHost(host: CallbackBatchHost): void;
   /** Update the session predicate and schedule an asynchronous drain when available. */
   setAvailability(isAvailable: () => boolean): void;
+  /** Change delivery immediately without clearing pending callbacks or handoff receipts. */
+  setWhileBusy(mode: CallbackDeliveryMode): void;
+  /** Non-idle context operations are not foreground agent runs. */
+  setForegroundRunning(running: boolean): void;
+  /** Shared IDs make duplicate notifications from both extensions idempotent. */
+  toolStarted(toolCallId: string): void;
+  /** Await the final active tool's handoff before Pi checks its steering queue. */
+  toolEnded(toolCallId: string): Promise<boolean>;
   deliverUrgent(event: UrgentCallbackEvent): boolean | Promise<boolean>;
   cancel(): void;
   pendingCount(): number;
@@ -104,6 +234,7 @@ interface PendingEvent {
 interface SharedCallbackBatcherState {
   byHost: WeakMap<object, CallbackBatcher>;
   bySession?: WeakMap<object, CallbackBatcher>;
+  sessionDefaults?: WeakMap<object, { sessionId?: string; mode: CallbackDeliveryMode }>;
 }
 
 const GLOBAL_STATE_KEY = Symbol.for("@1aboveio/pi-better-harness/callback-batcher");
@@ -131,7 +262,6 @@ const RETRIEVAL_FOOTER =
 
 export const CALLBACK_BATCH_WINDOW_ENV = "PI_BETTER_CALLBACK_BATCH_MS";
 export const DEFAULT_CALLBACK_BATCH_WINDOW_MS = DEFAULT_WINDOW_MS;
-
 export function resolveCallbackBatchWindowMs(
   value: unknown = process.env[CALLBACK_BATCH_WINDOW_ENV],
 ): number {
@@ -436,26 +566,31 @@ export function createCallbackBatcher(
 ): CallbackBatcher {
   const windowMs = options.windowMs ?? resolveCallbackBatchWindowMs();
   const retryMs = Math.max(0, options.retryMs ?? DEFAULT_RETRY_MS);
+  let whileBusy = options.whileBusy ?? "hold";
   const maxBytes = callbackBatchBudget(options.maxBytes);
   const pending = new Map<string, PendingEvent>();
   const inFlight = new Set<string>();
   const urgentInFlight = new Set<string>();
   const handedOff = new Map<string, number>();
+  const activeTools = new Set<string>();
   let sequence = 0;
   let timer: ReturnType<typeof setTimeout> | undefined;
   let flushPromise: Promise<boolean> | undefined;
   let isAvailable = options.isAvailable ?? (() => true);
   let gated = options.isAvailable !== undefined;
   let awaitingRun = false;
+  let cancelled = false;
+  let foregroundRunning = false;
 
-  const available = (): boolean => {
+  const deliveryMode = (): "followUp" | "steer" | undefined => {
+    if (cancelled) return undefined;
     try {
       if (!isAvailable()) {
         awaitingRun = false;
-        return false;
+        return whileBusy === "steer" && foregroundRunning && activeTools.size === 0 ? "steer" : undefined;
       }
-      return !awaitingRun;
-    } catch { return false; }
+      return !awaitingRun ? "followUp" : undefined;
+    } catch { return undefined; }
   };
 
   const cancelTimer = (): void => {
@@ -466,7 +601,7 @@ export function createCallbackBatcher(
   const schedule = (delayMs: number): void => {
     if (timer || pending.size === 0) return;
     // Some idle transitions (manual compaction) have no agent_settled event.
-    const delay = available() ? delayMs : retryMs;
+    const delay = deliveryMode() ? delayMs : retryMs;
     timer = setTimeout(() => {
       timer = undefined;
       void api.flush();
@@ -485,7 +620,8 @@ export function createCallbackBatcher(
 
   const performFlush = async (): Promise<boolean> => {
     cancelTimer();
-    if (!available()) {
+    const mode = deliveryMode();
+    if (!mode) {
       schedule(retryMs);
       return false;
     }
@@ -539,14 +675,16 @@ export function createCallbackBatcher(
     try {
       // Pi may defer the requested run while settled handlers finish, leaving
       // isIdle() true. Permit no second handoff until a run starts.
-      awaitingRun = gated;
+      awaitingRun = mode === "followUp" && gated;
       await host.sendMessage(
         {
           customType: "background-completion-batch",
           content: packed.text,
           display: true,
         },
-        { deliverAs: "followUp", triggerTurn: true },
+        // Newer Pi requires triggerTurn for steering too; false only appends
+        // a custom message at the turn boundary without continuing the agent.
+        { deliverAs: mode, triggerTurn: true },
       );
     } catch {
       awaitingRun = false;
@@ -632,15 +770,42 @@ export function createCallbackBatcher(
     setAvailability(check) {
       gated = true;
       isAvailable = check;
-      available();
+      cancelled = false;
+      deliveryMode();
       cancelTimer();
       schedule(windowMs);
     },
+    setWhileBusy(mode) {
+      if (!isDeliveryMode(mode)) throw new Error("Invalid callback delivery mode. Nothing changed.");
+      whileBusy = mode;
+      cancelTimer();
+      schedule(windowMs);
+    },
+    setForegroundRunning(running) {
+      foregroundRunning = running;
+      cancelTimer();
+      schedule(windowMs);
+    },
+    toolStarted(toolCallId) {
+      activeTools.add(toolCallId);
+    },
+    async toolEnded(toolCallId) {
+      const ended = activeTools.delete(toolCallId);
+      if (activeTools.size > 0 || whileBusy !== "steer" || cancelled) return false;
+      if (!ended) return flushPromise ?? false;
+      // A timer may still be completing a deferred flush. Recheck after it so
+      // the tool boundary cannot lose the pending batch to that older attempt.
+      if (flushPromise) await flushPromise;
+      return pending.size > 0 ? api.flush() : true;
+    },
     deliverUrgent,
     cancel() {
       cancelTimer();
       pending.clear();
       awaitingRun = false;
+      activeTools.clear();
+      foregroundRunning = false;
+      cancelled = true;
     },
     pendingCount() {
       return pending.size;
@@ -671,11 +836,39 @@ export function cancelCallbackBatch(host: CallbackBatchHost): void {
   batcher?.cancel();
 }
 
-/** Called on session_start, agent_start and agent_settled by both extensions. */
+/** Restore current branch/default mode without resetting foreground tools or receipts. */
 export function setCallbackBatchContext(
   host: CallbackBatchHost,
-  ctx: { isIdle(): boolean; sessionManager?: object },
+  ctx: CallbackSettingsContext & { isIdle(): boolean },
+  seams: CallbackSettingsSeams = {},
 ): void {
+  const batcher = bindCallbackBatcher(host, ctx);
+  let mode: CallbackDeliveryMode = "hold";
+  try {
+    const settings = getCallbackSettings(ctx, seams);
+    if (settings.source === "session") {
+      // Capture the initial default even when an override currently hides it,
+      // so navigating to an unconfigured branch cannot inherit a later save.
+      try { getSessionDefault(ctx, seams); }
+      catch {
+        if (ctx.sessionManager) {
+          const defaults = globalState().sessionDefaults ??= new WeakMap();
+          defaults.set(ctx.sessionManager, { sessionId: ctx.sessionManager.getSessionId?.(), mode: "hold" });
+        }
+      }
+    }
+    mode = settings.mode;
+  } catch (error) {
+    try { ctx.ui?.notify(`Callback delivery restored to hold: ${String(error)}`, "error"); }
+    catch { /* A failing UI must not retain the prior session's delivery mode. */ }
+  }
+  batcher.setWhileBusy(mode);
+  // Schedule rather than await delivery inside agent_settled: its handlers are
+  // still settling the previous run, and sendMessage may start a new one.
+  batcher.setAvailability(() => ctx.isIdle());
+}
+
+function bindCallbackBatcher(host: CallbackBatchHost, ctx: CallbackSettingsContext): CallbackBatcher {
   const state = globalState();
   const bySession = state.bySession ??= new WeakMap<object, CallbackBatcher>();
   const shared = ctx.sessionManager ? bySession.get(ctx.sessionManager) : undefined;
@@ -683,9 +876,7 @@ export function setCallbackBatchContext(
   if (ctx.sessionManager) bySession.set(ctx.sessionManager, batcher);
   state.byHost.set(host.events ?? host, batcher);
   batcher.setHost(host);
-  // Schedule rather than await delivery inside agent_settled: its handlers are
-  // still settling the previous run, and sendMessage may start a new one.
-  batcher.setAvailability(() => ctx.isIdle());
+  return batcher;
 }
 
 function globalState(): SharedCallbackBatcherState {

package/shared-task-tools.ts

@@ -4,7 +4,7 @@
  *
  * - Guarded tools are harness adapters that go through the task's guarded file
  *   operations, so the file rules govern them. Today: apply_patch.
- * - Trusted tools are third-party tools admitted by name AND owning package.
+ * - Trusted tools are extension tools admitted by name AND owning package.
  *   They run in the child Pi process, outside the file rules; a human ticks each
  *   one in /sandbox. Network tools are refused while Network is Off.
  */
@@ -36,15 +36,17 @@ export function defaultSubagentTools(): SubagentToolSettings {
 
 /** Tools that reach the network; refused while the profile's Network access is Off. */
 export function isNetworkTool(name: string): boolean {
-    return ["web_fetch", "web_search", "firecrawl_scrape", "firecrawl_extract", "mcp", "mcpScript", "remote_bash"].includes(name) ||
+    return ["web_fetch", "web_search", "firecrawl_scrape", "firecrawl_extract", "mcp", "mcpScript", ...SSH_TOOL_NAMES].includes(name) ||
         name.startsWith("mcp__");
 }
 
 /** Names never offered as trusted: builtins and the guarded adapters. */
 export const RESERVED_TOOL_NAMES = Object.freeze(["read", "write", "edit", "bash", "grep", "find", "ls", "powershell", "apply_patch", "process_list"]);
 
-/** Packages that are this harness: their tools are never offered as trusted tools. */
+/** Harness tools stay excluded except for the explicitly supported SSH tools. */
 const HARNESS_PACKAGES = /(^|[/:])pi-better-[a-z-]+$/;
+const SSH_PACKAGE = /(^|[/:])pi-better-ssh$/;
+const SSH_TOOL_NAMES = ["remote_bash", "ssh_profile", "ssh_mux"];
 
 /** Strict decoding: a malformed tool list must not silently admit anything. */
 export function parseSubagentTools(value: unknown): SubagentToolSettings {
@@ -95,13 +97,16 @@ export function toolPackage(tool: ToolSource): DiscoveredTool | undefined {
     return { name: tool.name, package: pkg, root: base };
 }
 
-/** Candidate trusted tools in a running Pi, excluding core builtins, guarded names and this harness. */
+/** Candidate trusted tools, excluding core builtins, guarded names and non-SSH harness tools. */
 export function discoverTrustedTools(tools: readonly ToolSource[]): DiscoveredTool[] {
     const found: DiscoveredTool[] = [];
     for (const tool of tools) {
         if (RESERVED_TOOL_NAMES.includes(tool.name)) continue;
         const owner = toolPackage(tool);
-        if (!owner || [owner.package, owner.root].some((id) => HARNESS_PACKAGES.test(id.replace(/\/+$/, "")))) continue;
+        if (!owner) continue;
+        const harnessOwners = [owner.package, owner.root].map((id) => id.replace(/\/+$/, ""))
+            .filter((id) => HARNESS_PACKAGES.test(id));
+        if (harnessOwners.length && !(SSH_TOOL_NAMES.includes(tool.name) && harnessOwners.every((id) => SSH_PACKAGE.test(id)))) continue;
         if (!found.some((t) => t.name === owner.name && t.package === owner.package)) {
             found.push(tool.sourceInfo?.source === "builtin" ? { ...owner, source: "builtin" } : owner);
         }