lsd-pi 1.2.4 → 1.3.2

package/dist/resources/extensions/subagent/index.js

@@ -32,11 +32,167 @@ import { loadEffectivePreferences } from "../shared/preferences.js";
 import { CmuxClient } from "../cmux/index.js";
 import { BackgroundJobManager } from "./background-job-manager.js";
 import { runSubagentInBackground } from "./background-runner.js";
+import { showAgentSwitcher } from "./agent-switcher-component.js";
+import { buildAgentSwitchTargets, } from "./agent-switcher-model.js";
 const MAX_PARALLEL_TASKS = 8;
 const MAX_CONCURRENCY = 4;
 const COLLAPSED_ITEM_COUNT = 10;
 const DEFAULT_AWAIT_SUBAGENT_TIMEOUT_SECONDS = 120;
 const liveSubagentProcesses = new Set();
+const agentSessionLinksById = new Map();
+const agentSessionIdsByParent = new Map();
+const parentSessionByChild = new Map();
+const liveRuntimeBySessionFile = new Map();
+let agentSessionLinkCounter = 0;
+function listSessionFiles(sessionDir) {
+    if (!fs.existsSync(sessionDir))
+        return [];
+    try {
+        return fs
+            .readdirSync(sessionDir)
+            .filter((name) => name.endsWith(".jsonl"))
+            .map((name) => path.join(sessionDir, name));
+    }
+    catch {
+        return [];
+    }
+}
+function detectNewSubagentSessionFile(sessionDir, before, startedAt) {
+    const after = listSessionFiles(sessionDir);
+    const created = after.filter((file) => !before.has(file));
+    const candidates = created.length > 0 ? created : after;
+    const ranked = candidates
+        .map((file) => {
+        let mtime = 0;
+        try {
+            mtime = fs.statSync(file).mtimeMs;
+        }
+        catch {
+            mtime = 0;
+        }
+        return { file, mtime };
+    })
+        .filter((entry) => entry.mtime >= startedAt - 5000)
+        .sort((a, b) => b.mtime - a.mtime);
+    return ranked[0]?.file;
+}
+function registerAgentSessionLink(link) {
+    const now = Date.now();
+    const id = `agent-${++agentSessionLinkCounter}`;
+    const full = { ...link, id, createdAt: now, updatedAt: now };
+    agentSessionLinksById.set(id, full);
+    const list = agentSessionIdsByParent.get(link.parentSessionFile) ?? [];
+    list.push(id);
+    agentSessionIdsByParent.set(link.parentSessionFile, list);
+    parentSessionByChild.set(link.subagentSessionFile, link.parentSessionFile);
+    return full;
+}
+function updateAgentSessionLinkState(subagentSessionFile, state) {
+    for (const link of agentSessionLinksById.values()) {
+        if (link.subagentSessionFile === subagentSessionFile) {
+            link.state = state;
+            link.updatedAt = Date.now();
+            return;
+        }
+    }
+}
+function upsertAgentSessionLink(agentName, task, parentSessionFile, subagentSessionFile, state) {
+    const existingParent = parentSessionByChild.get(subagentSessionFile);
+    if (!existingParent) {
+        registerAgentSessionLink({
+            agentName,
+            task,
+            parentSessionFile,
+            subagentSessionFile,
+            state,
+        });
+        return;
+    }
+    updateAgentSessionLinkState(subagentSessionFile, state);
+}
+function getAgentSessionLinksForParent(parentSessionFile) {
+    const ids = agentSessionIdsByParent.get(parentSessionFile) ?? [];
+    return ids
+        .map((id) => agentSessionLinksById.get(id))
+        .filter((entry) => Boolean(entry))
+        .sort((a, b) => b.updatedAt - a.updatedAt);
+}
+function readSessionHeader(sessionFile) {
+    try {
+        const content = fs.readFileSync(sessionFile, "utf-8");
+        const firstLine = content.split("\n").find((line) => line.trim().length > 0);
+        if (!firstLine)
+            return null;
+        const parsed = JSON.parse(firstLine);
+        if (!parsed || parsed.type !== "session")
+            return null;
+        return {
+            parentSession: typeof parsed.parentSession === "string" ? parsed.parentSession : undefined,
+            subagentName: typeof parsed.subagentName === "string" ? parsed.subagentName : undefined,
+            subagentTask: typeof parsed.subagentTask === "string" ? parsed.subagentTask : undefined,
+            subagentSystemPrompt: typeof parsed.subagentSystemPrompt === "string" ? parsed.subagentSystemPrompt : undefined,
+            subagentTools: Array.isArray(parsed.subagentTools)
+                ? parsed.subagentTools.filter((tool) => typeof tool === "string")
+                : undefined,
+        };
+    }
+    catch {
+        return null;
+    }
+}
+function backfillAgentSessionLinksForParent(parentSessionFile, sessionDir) {
+    for (const sessionFile of listSessionFiles(sessionDir)) {
+        if (sessionFile === parentSessionFile)
+            continue;
+        const header = readSessionHeader(sessionFile);
+        if (header?.parentSession !== parentSessionFile)
+            continue;
+        const existingParent = parentSessionByChild.get(sessionFile);
+        if (!existingParent) {
+            registerAgentSessionLink({
+                agentName: header.subagentName ?? "subagent",
+                task: header.subagentTask ?? "Recovered from persisted session lineage",
+                parentSessionFile,
+                subagentSessionFile: sessionFile,
+                state: "completed",
+            });
+        }
+    }
+    return getAgentSessionLinksForParent(parentSessionFile);
+}
+function formatSwitchTargetSummary(target) {
+    const current = target.isCurrent ? " (current)" : "";
+    if (target.kind === "parent") {
+        return `● parent — main session${current}`;
+    }
+    const icon = target.state === "running" ? "▶" : target.state === "failed" ? "✗" : "✓";
+    return `${icon} ${target.agentName} — ${target.taskPreview}${current}`;
+}
+function buildSwitchTargetsForParent(parentSessionFile, currentSessionFile, currentCwd, trackedLinks, runningJobs) {
+    return buildAgentSwitchTargets({
+        currentSessionFile,
+        rootParentSessionFile: parentSessionFile,
+        currentCwd,
+        trackedLinks: trackedLinks.map((link) => ({
+            id: link.id,
+            agentName: link.agentName,
+            task: link.task,
+            parentSessionFile: link.parentSessionFile,
+            subagentSessionFile: link.subagentSessionFile,
+            updatedAt: link.updatedAt,
+            state: link.state,
+        })),
+        runningJobs: runningJobs.map((job) => ({
+            id: job.id,
+            agentName: job.agentName,
+            task: job.task,
+            startedAt: job.startedAt,
+            parentSessionFile: job.parentSessionFile,
+            sessionFile: job.sessionFile,
+            cwd: job.cwd,
+        })),
+    });
+}
 const AwaitSubagentParams = Type.Object({
     jobs: Type.Optional(Type.Array(Type.String(), {
         description: "Subagent job IDs to wait for. Omit to wait for the next running background subagent.",
@@ -340,7 +496,7 @@ function resolveSubagentCliPath(defaultCwd) {
     }
     return null;
 }
-function processSubagentEventLine(line, currentResult, emitUpdate, proc) {
+function processSubagentEventLine(line, currentResult, emitUpdate, proc, onSessionInfo, onEventType, onParsedEvent) {
     if (!line.trim())
         return false;
     let event;
@@ -350,6 +506,29 @@ function processSubagentEventLine(line, currentResult, emitUpdate, proc) {
     catch {
         return false;
     }
+    const eventType = typeof event.type === "string" ? event.type : "unknown";
+    onEventType?.(eventType);
+    onParsedEvent?.(event);
+    if (event.type === "subagent_session_info") {
+        let changed = false;
+        if (typeof event.sessionFile === "string" && event.sessionFile) {
+            if (currentResult.sessionFile !== event.sessionFile)
+                changed = true;
+            currentResult.sessionFile = event.sessionFile;
+        }
+        if (typeof event.parentSessionFile === "string" && event.parentSessionFile) {
+            if (currentResult.parentSessionFile !== event.parentSessionFile)
+                changed = true;
+            currentResult.parentSessionFile = event.parentSessionFile;
+        }
+        if (changed) {
+            onSessionInfo?.({
+                sessionFile: currentResult.sessionFile,
+                parentSessionFile: currentResult.parentSessionFile,
+            });
+        }
+        return false;
+    }
     if (proc && isSubagentPermissionRequest(event)) {
         void handleSubagentPermissionRequest(event, proc);
         return false;
@@ -394,7 +573,7 @@ async function waitForFile(filePath, signal, timeoutMs = 30 * 60 * 1000) {
     }
     return false;
 }
-async function runSingleAgent(defaultCwd, agents, agentName, task, cwd, step, modelOverride, parentModel, signal, onUpdate, makeDetails, foregroundHooks) {
+async function runSingleAgent(defaultCwd, agents, agentName, task, cwd, step, modelOverride, parentModel, signal, onUpdate, makeDetails, parentSessionFile, attachableSession, onSessionInfo, onSubagentEvent, foregroundHooks) {
     const agent = agents.find((a) => a.name === agentName);
     if (!agent) {
         const available = agents.map((a) => `"${a.name}"`).join(", ") || "none";
@@ -425,6 +604,7 @@ async function runSingleAgent(defaultCwd, agents, agentName, task, cwd, step, mo
         usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, contextTokens: 0, turns: 0 },
         model: inferredModel,
         step,
+        parentSessionFile,
     };
     const emitUpdate = () => {
         if (onUpdate) {
@@ -462,17 +642,27 @@ async function runSingleAgent(defaultCwd, agents, agentName, task, cwd, step, mo
             tmpPromptDir = tmp.dir;
             tmpPromptPath = tmp.filePath;
         }
-        const args = buildSubagentProcessArgs(agent, task, tmpPromptPath, inferredModel);
+        const effectiveCwd = cwd ?? defaultCwd;
+        const subagentSessionDir = parentSessionFile ? path.dirname(parentSessionFile) : undefined;
+        const sessionFilesBefore = attachableSession && subagentSessionDir
+            ? new Set(listSessionFiles(subagentSessionDir))
+            : undefined;
+        const launchStartedAt = Date.now();
+        const args = buildSubagentProcessArgs(agent, task, tmpPromptPath, inferredModel, {
+            noSession: !attachableSession,
+            parentSessionFile: parentSessionFile,
+            mode: attachableSession ? "rpc" : "json",
+        });
         const exitCode = await new Promise((resolve) => {
             const bundledPaths = getBundledExtensionPathsFromEnv();
             const extensionArgs = bundledPaths.flatMap((p) => ["--extension", p]);
-            const cliPath = resolveSubagentCliPath(cwd ?? defaultCwd);
+            const cliPath = resolveSubagentCliPath(effectiveCwd);
             if (!cliPath) {
                 currentResult.stderr += "Unable to resolve LSD/GSD CLI path for subagent launch.";
                 resolve(1);
                 return;
             }
-            const proc = spawn(process.execPath, [cliPath, ...extensionArgs, ...args], { cwd: cwd ?? defaultCwd, shell: false, stdio: ["pipe", "pipe", "pipe"] });
+            const proc = spawn(process.execPath, [cliPath, ...extensionArgs, ...args], { cwd: effectiveCwd, shell: false, stdio: ["pipe", "pipe", "pipe"] });
             // Keep stdin open so approval/classifier responses can be proxied back
             // into the child process. Closing it here can leave the subagent stuck
             // after its first turn when it requests permission for a tool call.
@@ -481,6 +671,9 @@ async function runSingleAgent(defaultCwd, agents, agentName, task, cwd, step, mo
             let completionSeen = false;
             let resolved = false;
             let foregroundReleased = false;
+            let isBusy = false;
+            let commandSeq = 0;
+            const pendingCommandResponses = new Map();
             const procAbortController = new AbortController();
             let resolveBackgroundResult;
             let rejectBackgroundResult;
@@ -488,6 +681,15 @@ async function runSingleAgent(defaultCwd, agents, agentName, task, cwd, step, mo
                 resolveBackgroundResult = resolveBg;
                 rejectBackgroundResult = rejectBg;
             });
+            const sendRpcCommand = async (command) => {
+                const id = `sa_cmd_${++commandSeq}`;
+                if (!proc.stdin)
+                    throw new Error("Subagent RPC stdin is not available.");
+                return new Promise((resolveCmd, rejectCmd) => {
+                    pendingCommandResponses.set(id, { resolve: resolveCmd, reject: rejectCmd });
+                    proc.stdin.write(JSON.stringify({ id, ...command }) + "\n");
+                });
+            };
             const finishForeground = (code) => {
                 if (resolved)
                     return;
@@ -511,16 +713,60 @@ async function runSingleAgent(defaultCwd, agents, agentName, task, cwd, step, mo
                 agentName,
                 task,
                 cwd: cwd ?? defaultCwd,
+                parentSessionFile,
                 abortController: procAbortController,
                 resultPromise: backgroundResultPromise,
                 adoptToBackground,
+                sendPrompt: attachableSession
+                    ? async (text, images) => {
+                        await sendRpcCommand({ type: "prompt", message: text, images });
+                    }
+                    : undefined,
+                sendSteer: attachableSession
+                    ? async (text, images) => {
+                        await sendRpcCommand({ type: "steer", message: text, images });
+                    }
+                    : undefined,
+                sendFollowUp: attachableSession
+                    ? async (text, images) => {
+                        await sendRpcCommand({ type: "follow_up", message: text, images });
+                    }
+                    : undefined,
+                isBusy: attachableSession ? () => isBusy : undefined,
             });
             proc.stdout.on("data", (data) => {
                 buffer += data.toString();
                 const lines = buffer.split("\n");
                 buffer = lines.pop() || "";
                 for (const line of lines) {
-                    if (processSubagentEventLine(line, currentResult, emitUpdate, proc)) {
+                    const trimmed = line.trim();
+                    if (!trimmed)
+                        continue;
+                    if (attachableSession) {
+                        try {
+                            const parsed = JSON.parse(trimmed);
+                            if (parsed?.type === "response" && typeof parsed.id === "string" && pendingCommandResponses.has(parsed.id)) {
+                                const pending = pendingCommandResponses.get(parsed.id);
+                                pendingCommandResponses.delete(parsed.id);
+                                if (parsed.success === false) {
+                                    pending.reject(new Error(typeof parsed.error === "string" ? parsed.error : "Subagent RPC command failed."));
+                                }
+                                else {
+                                    pending.resolve(parsed.data);
+                                }
+                                continue;
+                            }
+                        }
+                        catch {
+                            // Fall through to generic event processing.
+                        }
+                    }
+                    if (processSubagentEventLine(trimmed, currentResult, emitUpdate, proc, onSessionInfo, (eventType) => {
+                        if (eventType === "agent_start")
+                            isBusy = true;
+                        if (eventType === "agent_end")
+                            isBusy = false;
+                    }, (event) => onSubagentEvent?.(event, currentResult))) {
                         completionSeen = true;
                         try {
                             proc.kill("SIGTERM");
@@ -537,27 +783,65 @@ async function runSingleAgent(defaultCwd, agents, agentName, task, cwd, step, mo
             proc.on("close", (code) => {
                 liveSubagentProcesses.delete(proc);
                 if (buffer.trim()) {
-                    const completedOnFlush = processSubagentEventLine(buffer, currentResult, emitUpdate, proc);
+                    const completedOnFlush = processSubagentEventLine(buffer, currentResult, emitUpdate, proc, onSessionInfo, (eventType) => {
+                        if (eventType === "agent_start")
+                            isBusy = true;
+                        if (eventType === "agent_end")
+                            isBusy = false;
+                    }, (event) => onSubagentEvent?.(event, currentResult));
                     completionSeen = completionSeen || completedOnFlush;
                 }
+                isBusy = false;
+                for (const pending of pendingCommandResponses.values()) {
+                    pending.reject(new Error("Subagent process closed before command response."));
+                }
+                pendingCommandResponses.clear();
                 const finalExitCode = completionSeen && (code === null || code === 143 || code === 15) ? 0 : (code ?? 0);
                 currentResult.exitCode = finalExitCode;
+                if (attachableSession && sessionFilesBefore && subagentSessionDir && !currentResult.sessionFile) {
+                    const detected = detectNewSubagentSessionFile(subagentSessionDir, sessionFilesBefore, launchStartedAt);
+                    if (detected)
+                        currentResult.sessionFile = detected;
+                }
                 resolveBackgroundResult?.({
                     summary: getFinalOutput(currentResult.messages),
                     stderr: currentResult.stderr,
                     exitCode: finalExitCode,
                     model: currentResult.model,
+                    sessionFile: currentResult.sessionFile,
+                    parentSessionFile: currentResult.parentSessionFile,
                 });
                 foregroundHooks?.onFinish?.();
                 finishForeground(finalExitCode);
             });
             proc.on("error", (error) => {
                 liveSubagentProcesses.delete(proc);
+                isBusy = false;
+                for (const pending of pendingCommandResponses.values()) {
+                    pending.reject(error instanceof Error ? error : new Error(String(error)));
+                }
+                pendingCommandResponses.clear();
                 rejectBackgroundResult?.(error);
                 foregroundHooks?.onFinish?.();
                 finishForeground(1);
             });
+            if (attachableSession) {
+                void sendRpcCommand({ type: "prompt", message: task }).catch((error) => {
+                    currentResult.stderr += error instanceof Error ? error.message : String(error);
+                    try {
+                        proc.kill("SIGTERM");
+                    }
+                    catch {
+                        /* ignore */
+                    }
+                });
+            }
             const killProc = () => {
+                // If the process has been adopted to the background (e.g. via Ctrl+B or
+                // /agent attach_live), foregroundReleased is true and the process should
+                // survive the parent session's abort — don't kill it.
+                if (foregroundReleased)
+                    return;
                 wasAborted = true;
                 procAbortController.abort();
                 proc.kill("SIGTERM");
@@ -586,6 +870,12 @@ async function runSingleAgent(defaultCwd, agents, agentName, task, cwd, step, mo
             }
         });
         currentResult.exitCode = exitCode;
+        if (attachableSession && sessionFilesBefore && subagentSessionDir) {
+            const detected = detectNewSubagentSessionFile(subagentSessionDir, sessionFilesBefore, launchStartedAt);
+            if (detected) {
+                currentResult.sessionFile = detected;
+            }
+        }
         if (wasAborted)
             throw new Error("Subagent was aborted");
         return currentResult;
@@ -646,14 +936,62 @@ export default function (pi) {
     const foregroundSubagentStatusKey = "foreground-subagent";
     const foregroundSubagentHint = "Ctrl+B: move foreground subagent to background";
     let activeForegroundSubagent = null;
+    let activeSessionFileForUi;
+    const liveStreamBufferBySession = new Map();
+    function flushLiveStream(sessionFile) {
+        const buffered = liveStreamBufferBySession.get(sessionFile);
+        if (!buffered || !buffered.trim())
+            return;
+        liveStreamBufferBySession.set(sessionFile, "");
+        pi.sendMessage({
+            customType: "live_subagent_stream",
+            content: buffered,
+            display: true,
+        }, { deliverAs: "followUp" });
+    }
+    function pushLiveStreamDelta(sessionFile, delta) {
+        const prev = liveStreamBufferBySession.get(sessionFile) ?? "";
+        const next = prev + delta;
+        liveStreamBufferBySession.set(sessionFile, next);
+        if (next.length >= 120 || next.includes("\n")) {
+            flushLiveStream(sessionFile);
+        }
+    }
+    function getCurrentSessionSubagentMetadata(sessionFile) {
+        if (!sessionFile)
+            return null;
+        return readSessionHeader(sessionFile);
+    }
+    function applyCurrentSessionSubagentTools(ctx) {
+        const metadata = getCurrentSessionSubagentMetadata(ctx.sessionManager.getSessionFile());
+        if (metadata?.subagentTools && metadata.subagentTools.length > 0) {
+            ctx.setActiveTools(metadata.subagentTools);
+        }
+    }
     function getBgManager() {
         if (!bgManager)
             throw new Error("BackgroundJobManager not initialized.");
         return bgManager;
     }
-    pi.on("session_start", async () => {
+    pi.on("session_start", async (_event, ctx) => {
+        activeSessionFileForUi = ctx.sessionManager.getSessionFile();
         bgManager = new BackgroundJobManager({
             onJobComplete: (job) => {
+                if (job.sessionFile && job.parentSessionFile) {
+                    const existingParent = parentSessionByChild.get(job.sessionFile);
+                    if (!existingParent) {
+                        registerAgentSessionLink({
+                            agentName: job.agentName,
+                            task: job.task,
+                            parentSessionFile: job.parentSessionFile,
+                            subagentSessionFile: job.sessionFile,
+                            state: job.status === "failed" ? "failed" : "completed",
+                        });
+                    }
+                    else {
+                        updateAgentSessionLinkState(job.sessionFile, job.status === "failed" ? "failed" : "completed");
+                    }
+                }
                 if (job.awaited)
                     return;
                 const statusEmoji = job.status === "completed" ? "✓" : job.status === "cancelled" ? "✗ cancelled" : "✗ failed";
@@ -676,22 +1014,80 @@ export default function (pi) {
                 }, { deliverAs: "followUp" });
             },
         });
+        applyCurrentSessionSubagentTools(ctx);
     });
-    pi.on("session_before_switch", async () => {
-        activeForegroundSubagent = null;
-        if (bgManager) {
-            for (const job of bgManager.getRunningJobs()) {
-                bgManager.cancel(job.id);
+    pi.on("session_switch", async (_event, ctx) => {
+        activeSessionFileForUi = ctx.sessionManager.getSessionFile();
+        applyCurrentSessionSubagentTools(ctx);
+    });
+    pi.on("before_agent_start", async (event, ctx) => {
+        const metadata = getCurrentSessionSubagentMetadata(ctx.sessionManager.getSessionFile());
+        if (!metadata?.subagentSystemPrompt)
+            return;
+        const subagentName = metadata.subagentName ?? "subagent";
+        const taskNote = metadata.subagentTask
+            ? `Original delegated task: ${metadata.subagentTask}`
+            : "Continue operating as the delegated subagent for this session.";
+        const antiRecursion = [
+            `You are already the ${subagentName} subagent for this session.`,
+            "Do not spawn or delegate to another subagent with the same name as yourself.",
+            `If the user asks you to continue ${subagentName} work, do that work directly in this session.`,
+            taskNote,
+            "IMPORTANT: There is NO human available to answer questions in this session. Do NOT call ask_user_questions. Make all decisions autonomously based on the task and context.",
+        ].join("\n");
+        return {
+            systemPrompt: `${event.systemPrompt}\n\n${antiRecursion}\n\n${metadata.subagentSystemPrompt}`,
+        };
+    });
+    pi.on("input", async (event, ctx) => {
+        const sessionFile = ctx.sessionManager.getSessionFile();
+        if (!sessionFile)
+            return;
+        const runtime = liveRuntimeBySessionFile.get(sessionFile);
+        if (!runtime)
+            return;
+        const text = event.text?.trim();
+        if (!text)
+            return { action: "handled" };
+        const isSlashCommand = text.startsWith("/");
+        if (isSlashCommand)
+            return;
+        try {
+            if (runtime.isBusy()) {
+                await runtime.sendSteer(text, event.images);
+                ctx.ui.notify(`Sent steer to running subagent ${runtime.agentName}.`, "info");
             }
+            else {
+                await runtime.sendPrompt(text, event.images);
+                ctx.ui.notify(`Sent prompt to live subagent ${runtime.agentName}.`, "info");
+            }
+            return { action: "handled" };
+        }
+        catch (error) {
+            ctx.ui.notify(`Failed to send input to live subagent ${runtime.agentName}: ${error instanceof Error ? error.message : String(error)}`, "error");
+            return { action: "handled" };
         }
     });
+    pi.on("session_before_switch", async () => {
+        if (activeSessionFileForUi)
+            flushLiveStream(activeSessionFileForUi);
+        activeForegroundSubagent = null;
+    });
     pi.on("session_shutdown", async () => {
+        if (activeSessionFileForUi)
+            flushLiveStream(activeSessionFileForUi);
+        activeSessionFileForUi = undefined;
         activeForegroundSubagent = null;
         await stopLiveSubagents();
         if (bgManager) {
             bgManager.shutdown();
             bgManager = null;
         }
+        agentSessionLinksById.clear();
+        agentSessionIdsByParent.clear();
+        parentSessionByChild.clear();
+        liveRuntimeBySessionFile.clear();
+        liveStreamBufferBySession.clear();
     });
     // /subagents command
     pi.registerCommand("subagents", {
@@ -795,6 +1191,191 @@ export default function (pi) {
             ctx.ui.notify(`Available agents (${discovery.agents.length}):\n${lines.join("\n")}`, "info");
         },
     });
+    // /agent command - switch to the parent or a tracked subagent session
+    pi.registerCommand("agent", {
+        description: "Switch focus to parent/subagent sessions (/agent picker, /agent <id|index|name>, /agent parent)",
+        handler: async (args, ctx) => {
+            const currentSessionFile = ctx.sessionManager.getSessionFile();
+            if (!currentSessionFile) {
+                ctx.ui.notify("Current session is in-memory only; /agent requires a persisted session file.", "warning");
+                return;
+            }
+            const arg = args.trim();
+            const parentSessionFile = parentSessionByChild.get(currentSessionFile);
+            const currentParent = parentSessionFile ?? currentSessionFile;
+            const currentSessionDir = path.dirname(currentParent);
+            let tracked = getAgentSessionLinksForParent(currentParent).filter((entry) => fs.existsSync(entry.subagentSessionFile));
+            if (tracked.length === 0) {
+                tracked = backfillAgentSessionLinksForParent(currentParent, currentSessionDir)
+                    .filter((entry) => fs.existsSync(entry.subagentSessionFile));
+            }
+            const runningJobs = bgManager?.getRunningJobs() ?? [];
+            const switchTargets = buildSwitchTargetsForParent(currentParent, currentSessionFile, ctx.cwd, tracked, runningJobs);
+            const applySwitchTarget = async (target) => {
+                if (target.selectionAction === "blocked") {
+                    ctx.ui.notify(target.blockedReason ?? "That target cannot be selected yet.", "warning");
+                    return;
+                }
+                if (target.selectionAction === "attach_live") {
+                    if (!fs.existsSync(target.sessionFile)) {
+                        ctx.ui.notify(`Live subagent session file is missing: ${target.sessionFile}`, "error");
+                        return;
+                    }
+                    const liveRuntime = liveRuntimeBySessionFile.get(target.sessionFile);
+                    if (!liveRuntime) {
+                        ctx.ui.notify("Live runtime is no longer available for this subagent. It may have completed.", "warning");
+                        return;
+                    }
+                    // Adopt the foreground subagent to background before switching sessions.
+                    // switchSession calls abort() which would fire the tool signal and SIGTERM
+                    // the running subagent process. Adopting to background detaches the process
+                    // from the foreground abort chain so it survives the session switch.
+                    const foreground = activeForegroundSubagent;
+                    if (foreground && !foreground.claimed && bgManager) {
+                        foreground.claimed = true;
+                        try {
+                            const jobId = bgManager.adoptRunning(foreground.agentName, foreground.task, foreground.cwd, foreground.abortController, foreground.resultPromise, {
+                                parentSessionFile: foreground.parentSessionFile ?? ctx.sessionManager.getSessionFile(),
+                            });
+                            const released = foreground.adoptToBackground(jobId);
+                            if (!released) {
+                                foreground.claimed = false;
+                                bgManager.cancel(jobId);
+                            }
+                            else {
+                                activeForegroundSubagent = null;
+                                ctx.ui.setStatus(foregroundSubagentStatusKey, undefined);
+                            }
+                        }
+                        catch {
+                            foreground.claimed = false;
+                        }
+                    }
+                    const switched = await ctx.switchSession(target.sessionFile);
+                    if (switched.cancelled) {
+                        ctx.ui.notify("Session switch was cancelled.", "warning");
+                        return;
+                    }
+                    ctx.ui.notify(`Attached to running subagent ${target.agentName}. Prompts in this session are routed live (busy => steer, idle => prompt). Use /agent parent to return.`, "info");
+                    return;
+                }
+                if (target.kind === "parent") {
+                    if (!fs.existsSync(target.sessionFile)) {
+                        ctx.ui.notify(`Parent session file not found: ${target.sessionFile}`, "error");
+                        return;
+                    }
+                    const switched = await ctx.switchSession(target.sessionFile);
+                    if (switched.cancelled) {
+                        ctx.ui.notify("Session switch was cancelled.", "warning");
+                        return;
+                    }
+                    ctx.ui.notify("Switched to parent session.", "info");
+                    return;
+                }
+                if (!fs.existsSync(target.sessionFile)) {
+                    ctx.ui.notify(`Subagent session file is missing: ${target.sessionFile}`, "error");
+                    return;
+                }
+                const switched = await ctx.switchSession(target.sessionFile);
+                if (switched.cancelled) {
+                    ctx.ui.notify("Session switch was cancelled.", "warning");
+                    return;
+                }
+                updateAgentSessionLinkState(target.sessionFile, target.state === "failed" ? "failed" : "completed");
+                ctx.ui.notify(`Switched to subagent ${target.agentName}. This resumes the saved subagent session; use /agent parent to return.`, "info");
+            };
+            if (!arg) {
+                const subagentTargets = switchTargets.filter((target) => target.kind === "subagent");
+                if (ctx.hasUI) {
+                    if (subagentTargets.length === 0 && !parentSessionFile) {
+                        ctx.ui.notify("No tracked subagent sessions for this parent session yet. Run a single-mode subagent first (foreground or background).", "info");
+                        return;
+                    }
+                    const selected = await showAgentSwitcher(ctx, switchTargets);
+                    if (!selected)
+                        return;
+                    await applySwitchTarget(selected);
+                    return;
+                }
+                if (subagentTargets.length === 0 && !parentSessionFile) {
+                    ctx.ui.notify("No tracked subagent sessions for this parent session yet. Run a single-mode subagent first (foreground or background).", "info");
+                    return;
+                }
+                const lines = ["Agent switch targets:"];
+                switchTargets.forEach((target, index) => {
+                    lines.push(`${index + 1}. ${formatSwitchTargetSummary(target)}`);
+                });
+                lines.push("", "Use `/agent <index|id|name>` for explicit targeting, or `/agent parent`.");
+                ctx.ui.notify(lines.join("\n"), "info");
+                return;
+            }
+            if (arg === "parent" || arg === "main") {
+                if (!parentSessionFile) {
+                    ctx.ui.notify("You are already in the parent/main session.", "info");
+                    return;
+                }
+                if (!fs.existsSync(parentSessionFile)) {
+                    ctx.ui.notify(`Parent session file not found: ${parentSessionFile}`, "error");
+                    return;
+                }
+                const switched = await ctx.switchSession(parentSessionFile);
+                if (switched.cancelled) {
+                    ctx.ui.notify("Session switch was cancelled.", "warning");
+                    return;
+                }
+                ctx.ui.notify("Switched to parent session.", "info");
+                return;
+            }
+            let target;
+            if (/^\d+$/.test(arg)) {
+                const index = Number.parseInt(arg, 10) - 1;
+                target = tracked[index];
+            }
+            if (!target) {
+                target = tracked.find((entry) => entry.id === arg);
+            }
+            if (!target) {
+                target = tracked.find((entry) => entry.agentName === arg);
+            }
+            if (!target) {
+                target = tracked.find((entry) => path.basename(entry.subagentSessionFile) === arg);
+            }
+            if (!target) {
+                const runningTarget = switchTargets.find((entry) => entry.id === arg && entry.kind === "subagent");
+                if (runningTarget?.state === "running") {
+                    ctx.ui.notify(runningTarget.blockedReason ?? "Selected subagent is still running. Live attach is not implemented yet.", "warning");
+                    return;
+                }
+                ctx.ui.notify(`Unknown subagent target: ${arg}. Run /agent to list available targets.`, "warning");
+                return;
+            }
+            if (!fs.existsSync(target.subagentSessionFile)) {
+                ctx.ui.notify(`Subagent session file is missing: ${target.subagentSessionFile}`, "error");
+                return;
+            }
+            if (target.state === "running") {
+                const liveRuntime = liveRuntimeBySessionFile.get(target.subagentSessionFile);
+                if (!liveRuntime) {
+                    ctx.ui.notify("Live runtime is no longer available for this subagent. It may have completed.", "warning");
+                    return;
+                }
+                const switched = await ctx.switchSession(target.subagentSessionFile);
+                if (switched.cancelled) {
+                    ctx.ui.notify("Session switch was cancelled.", "warning");
+                    return;
+                }
+                ctx.ui.notify(`Attached to running subagent ${target.agentName}. Prompts in this session are routed live (busy => steer, idle => prompt). Use /agent parent to return.`, "info");
+                return;
+            }
+            const switched = await ctx.switchSession(target.subagentSessionFile);
+            if (switched.cancelled) {
+                ctx.ui.notify("Session switch was cancelled.", "warning");
+                return;
+            }
+            updateAgentSessionLinkState(target.subagentSessionFile, target.state === "failed" ? "failed" : "completed");
+            ctx.ui.notify(`Switched to subagent ${target.agentName}. This resumes the saved subagent session; use /agent parent to return.`, "info");
+        },
+    });
     pi.registerShortcut(Key.ctrl("b"), {
         description: shortcutDesc("Move foreground subagent to background", "/subagents list"),
         handler: async (ctx) => {
@@ -809,7 +1390,9 @@ export default function (pi) {
             running.claimed = true;
             let jobId;
             try {
-                jobId = manager.adoptRunning(running.agentName, running.task, running.cwd, running.abortController, running.resultPromise);
+                jobId = manager.adoptRunning(running.agentName, running.task, running.cwd, running.abortController, running.resultPromise, {
+                    parentSessionFile: running.parentSessionFile ?? ctx.sessionManager.getSessionFile(),
+                });
             }
             catch (error) {
                 running.claimed = false;
@@ -873,7 +1456,7 @@ export default function (pi) {
             "For broad review or audit requests, use scout only as a prep step; the parent model or a reviewer should make the final judgments.",
             "Skip scout when the user already named the exact file/function to inspect or the task is obviously narrow.",
             "Use parallel mode when tasks are independent and don't need each other's output.",
-            "Use background: true when the user wants to keep chatting while a long-running agent works in parallel.",
+            "Default to foreground (background: false) for single-mode subagents. Only set background: true when the user explicitly asks to run it in the background or to keep chatting while it runs.",
             "If the user wants to wait for a background subagent result, use await_subagent.",
         ],
         parameters: SubagentParams,
@@ -884,6 +1467,7 @@ export default function (pi) {
             const confirmProjectAgents = params.confirmProjectAgents ?? false;
             const cmuxClient = CmuxClient.fromPreferences(loadEffectivePreferences()?.preferences);
             const cmuxSplitsEnabled = cmuxClient.getConfig().splits;
+            const invokingSessionFile = ctx.sessionManager.getSessionFile();
             // Resolve isolation mode
             const isolationMode = readIsolationMode();
             const useIsolation = Boolean(params.isolated) && isolationMode !== "none";
@@ -960,7 +1544,7 @@ export default function (pi) {
                             }
                         }
                         : undefined;
-                    const result = await runSingleAgent(ctx.cwd, agents, step.agent, taskWithContext, step.cwd, i + 1, step.model, ctx.model ? { provider: ctx.model.provider, id: ctx.model.id } : undefined, signal, chainUpdate, makeDetails("chain"));
+                    const result = await runSingleAgent(ctx.cwd, agents, step.agent, taskWithContext, step.cwd, i + 1, step.model, ctx.model ? { provider: ctx.model.provider, id: ctx.model.id } : undefined, signal, chainUpdate, makeDetails("chain"), invokingSessionFile, false, undefined, undefined);
                     results.push(result);
                     const isError = result.exitCode !== 0 || result.stopReason === "error" || result.stopReason === "aborted";
                     if (isError) {
@@ -1027,7 +1611,7 @@ export default function (pi) {
                             allResults[index] = partial.details.results[0];
                             emitParallelUpdate();
                         }
-                    }, makeDetails("parallel"));
+                    }, makeDetails("parallel"), invokingSessionFile, false, undefined);
                     let result = await runTask();
                     // Auto-retry failed tasks (likely API rate limit or transient error)
                     const isFailed = result.exitCode !== 0 || (result.messages.length === 0 && !signal?.aborted);
@@ -1084,14 +1668,61 @@ export default function (pi) {
                     const bgInferredModel = resolveSubagentModel({ name: agentForBg.name, model: bgResolvedModelCfg }, { overrideModel: params.model, parentModel: ctx.model ? { provider: ctx.model.provider, id: ctx.model.id } : undefined });
                     let jobId;
                     try {
-                        jobId = runSubagentInBackground(manager, agents, params.agent, params.task, params.cwd, params.model, { defaultCwd: ctx.cwd, model: ctx.model ? { provider: ctx.model.provider, id: ctx.model.id } : undefined }, async (bgSignal) => {
+                        jobId = runSubagentInBackground(manager, agents, params.agent, params.task, params.cwd, params.model, { defaultCwd: ctx.cwd, model: ctx.model ? { provider: ctx.model.provider, id: ctx.model.id } : undefined, parentSessionFile: invokingSessionFile }, async (bgSignal) => {
+                            let liveSessionFile;
+                            let liveRuntime;
                             const result = await runSingleAgent(ctx.cwd, agents, params.agent, params.task, params.cwd, undefined, params.model, ctx.model ? { provider: ctx.model.provider, id: ctx.model.id } : undefined, bgSignal, undefined, // no streaming updates for background jobs
-                            makeDetails("single"));
+                            makeDetails("single"), invokingSessionFile, true, (info) => {
+                                if (!invokingSessionFile || !info.sessionFile)
+                                    return;
+                                upsertAgentSessionLink(params.agent, params.task, invokingSessionFile, info.sessionFile, "running");
+                                liveSessionFile = info.sessionFile;
+                                if (liveRuntime) {
+                                    liveRuntime.sessionFile = info.sessionFile;
+                                    liveRuntime.parentSessionFile = info.parentSessionFile ?? invokingSessionFile;
+                                    liveRuntimeBySessionFile.set(info.sessionFile, liveRuntime);
+                                }
+                            }, (event, partial) => {
+                                const sessionFile = partial.sessionFile;
+                                if (!sessionFile || activeSessionFileForUi !== sessionFile)
+                                    return;
+                                if (event?.type === "message_update" && event.assistantMessageEvent?.type === "text_delta") {
+                                    const delta = String(event.assistantMessageEvent.delta ?? "");
+                                    if (delta)
+                                        pushLiveStreamDelta(sessionFile, delta);
+                                }
+                                if (event?.type === "message_end") {
+                                    flushLiveStream(sessionFile);
+                                }
+                            }, {
+                                onStart: (control) => {
+                                    if (!control.sendPrompt || !control.sendSteer || !control.sendFollowUp || !control.isBusy)
+                                        return;
+                                    liveRuntime = {
+                                        sessionFile: liveSessionFile,
+                                        parentSessionFile: invokingSessionFile,
+                                        agentName: params.agent,
+                                        isBusy: control.isBusy,
+                                        sendPrompt: control.sendPrompt,
+                                        sendSteer: control.sendSteer,
+                                        sendFollowUp: control.sendFollowUp,
+                                    };
+                                    if (liveSessionFile) {
+                                        liveRuntimeBySessionFile.set(liveSessionFile, liveRuntime);
+                                    }
+                                },
+                                onFinish: () => {
+                                    if (liveSessionFile)
+                                        liveRuntimeBySessionFile.delete(liveSessionFile);
+                                },
+                            });
                             return {
                                 exitCode: result.exitCode,
                                 finalOutput: getFinalOutput(result.messages),
                                 stderr: result.stderr,
                                 model: result.model,
+                                sessionFile: result.sessionFile,
+                                parentSessionFile: result.parentSessionFile,
                             };
                         });
                     }
@@ -1117,18 +1748,77 @@ export default function (pi) {
                         const taskId = crypto.randomUUID();
                         isolation = await createIsolation(effectiveCwd, taskId, isolationMode);
                     }
-                    const result = await runSingleAgent(ctx.cwd, agents, params.agent, params.task, isolation ? isolation.workDir : params.cwd, undefined, params.model, ctx.model ? { provider: ctx.model.provider, id: ctx.model.id } : undefined, signal, onUpdate, makeDetails("single"), !isolation
+                    let liveSessionFile;
+                    let liveRuntime;
+                    const result = await runSingleAgent(ctx.cwd, agents, params.agent, params.task, isolation ? isolation.workDir : params.cwd, undefined, params.model, ctx.model ? { provider: ctx.model.provider, id: ctx.model.id } : undefined, signal, onUpdate, makeDetails("single"), invokingSessionFile, !isolation, !isolation
+                        ? (info) => {
+                            if (!invokingSessionFile || !info.sessionFile)
+                                return;
+                            upsertAgentSessionLink(params.agent, params.task, invokingSessionFile, info.sessionFile, "running");
+                            liveSessionFile = info.sessionFile;
+                            if (liveRuntime) {
+                                liveRuntime.sessionFile = info.sessionFile;
+                                liveRuntime.parentSessionFile = info.parentSessionFile ?? invokingSessionFile;
+                                liveRuntimeBySessionFile.set(info.sessionFile, liveRuntime);
+                            }
+                        }
+                        : undefined, !isolation
+                        ? (event, partial) => {
+                            const sessionFile = partial.sessionFile;
+                            if (!sessionFile || activeSessionFileForUi !== sessionFile)
+                                return;
+                            if (event?.type === "message_update" && event.assistantMessageEvent?.type === "text_delta") {
+                                const delta = String(event.assistantMessageEvent.delta ?? "");
+                                if (delta)
+                                    pushLiveStreamDelta(sessionFile, delta);
+                            }
+                            if (event?.type === "message_end") {
+                                flushLiveStream(sessionFile);
+                            }
+                        }
+                        : undefined, !isolation
                         ? {
                             onStart: (control) => {
                                 activeForegroundSubagent = { ...control, claimed: false };
                                 ctx.ui.setStatus(foregroundSubagentStatusKey, foregroundSubagentHint);
+                                if (!control.sendPrompt || !control.sendSteer || !control.sendFollowUp || !control.isBusy)
+                                    return;
+                                liveRuntime = {
+                                    sessionFile: liveSessionFile,
+                                    parentSessionFile: invokingSessionFile,
+                                    agentName: params.agent,
+                                    isBusy: control.isBusy,
+                                    sendPrompt: control.sendPrompt,
+                                    sendSteer: control.sendSteer,
+                                    sendFollowUp: control.sendFollowUp,
+                                };
+                                if (liveSessionFile && liveRuntime) {
+                                    liveRuntimeBySessionFile.set(liveSessionFile, liveRuntime);
+                                }
                             },
                             onFinish: () => {
                                 activeForegroundSubagent = null;
                                 ctx.ui.setStatus(foregroundSubagentStatusKey, undefined);
+                                if (liveSessionFile)
+                                    liveRuntimeBySessionFile.delete(liveSessionFile);
                             },
                         }
                         : undefined);
+                    if (result.sessionFile && invokingSessionFile) {
+                        const existingParent = parentSessionByChild.get(result.sessionFile);
+                        if (!existingParent) {
+                            registerAgentSessionLink({
+                                agentName: result.agent,
+                                task: result.task,
+                                parentSessionFile: invokingSessionFile,
+                                subagentSessionFile: result.sessionFile,
+                                state: result.exitCode === 0 ? "completed" : "failed",
+                            });
+                        }
+                        else {
+                            updateAgentSessionLinkState(result.sessionFile, result.exitCode === 0 ? "completed" : "failed");
+                        }
+                    }
                     if (result.backgroundJobId) {
                         return {
                             content: [{ type: "text", text: `Moved ${result.agent} to background as **${result.backgroundJobId}**. Use \`await_subagent\`, \`/subagents wait ${result.backgroundJobId}\`, or \`/subagents output ${result.backgroundJobId}\`.` }],
@@ -1143,10 +1833,11 @@ export default function (pi) {
                         }
                     }
                     const isError = result.exitCode !== 0 || result.stopReason === "error" || result.stopReason === "aborted";
+                    const agentSwitchHint = result.sessionFile ? "\n\nTip: run `/agent` to switch focus to this subagent session." : "";
                     if (isError) {
                         const errorMsg = result.errorMessage || result.stderr || getFinalOutput(result.messages) || "(no output)";
                         return {
-                            content: [{ type: "text", text: `Agent ${result.stopReason || "failed"}: ${errorMsg}` }],
+                            content: [{ type: "text", text: `Agent ${result.stopReason || "failed"}: ${errorMsg}${agentSwitchHint}` }],
                             details: makeDetails("single")([result]),
                             isError: true,
                         };
@@ -1155,6 +1846,8 @@ export default function (pi) {
                     if (mergeResult && !mergeResult.success) {
                         outputText += `\n\n⚠ Patch merge failed: ${mergeResult.error || "unknown error"}`;
                     }
+                    if (agentSwitchHint)
+                        outputText += agentSwitchHint;
                     return {
                         content: [{ type: "text", text: outputText }],
                         details: makeDetails("single")([result]),