@code-yeongyu/senpi 2026.8.30-2 → 2026.8.30-3

package/dist/modes/interactive/interactive-host-runtime.js

@@ -4,8 +4,9 @@ import { executeBashWithOperations } from "../../core/bash-executor.js";
 import { SessionManager } from "../../core/session-manager.js";
 import { SettingsManager } from "../../core/settings-manager.js";
 import { ensureHost } from "../rpc/host-ensure.js";
-import { RpcClient } from "../rpc/rpc-client.js";
+import { isTransportGoneError, RpcClient } from "../rpc/rpc-client.js";
 export const INTERACTIVE_HOST_FALLBACK_WARNING = "Warning: shared interactive host unavailable; continuing locally";
+export const INTERACTIVE_HOST_RECONNECTING_WARNING = "Warning: shared interactive host connection lost; reconnecting";
 /**
  * Replace only the transport-facing session operations. The object returned is
  * deliberately still an AgentSessionRuntime: InteractiveMode and extensions
@@ -17,7 +18,82 @@ export async function createInteractiveHostRuntime(localRuntime, options) {
     if (!sessionPath)
         return localRuntime;
     const startHost = options.ensureHost ?? ((hostOptions) => ensureHost(hostOptions));
-    const client = new RpcClient({ socketPath: options.socket });
+    let remoteSession;
+    let remoteRuntime;
+    let reconnecting;
+    let disconnectQueued = false;
+    let disposed = false;
+    let fallbackWarned = false;
+    let reconnectWarningShown = false;
+    const warnReconnect = (cause) => {
+        if (reconnectWarningShown)
+            return;
+        reconnectWarningShown = true;
+        options.onWarning?.({
+            type: "interactive_host_action_failed",
+            message: INTERACTIVE_HOST_RECONNECTING_WARNING,
+            cause,
+        });
+    };
+    const warnFallback = (cause) => {
+        if (fallbackWarned)
+            return;
+        fallbackWarned = true;
+        options.onWarning?.({ type: "interactive_host_fallback", message: INTERACTIVE_HOST_FALLBACK_WARNING, cause });
+    };
+    let scheduleReconnect = () => { };
+    const client = new RpcClient({
+        socketPath: options.socket,
+        onDisconnect: () => {
+            disconnectQueued = true;
+            scheduleReconnect();
+        },
+    });
+    scheduleReconnect = () => {
+        if (!remoteSession || reconnecting || disposed || remoteRuntime?.isFallback)
+            return;
+        remoteRuntime?.enterReconnecting();
+        reconnecting = (async () => {
+            let cause;
+            for (let attempt = 0; attempt < 3; attempt++) {
+                if (disposed)
+                    return;
+                disconnectQueued = false;
+                try {
+                    await startHost({ socket: options.socket, agentDir: options.agentDir });
+                    if (disposed)
+                        return;
+                    await client.start();
+                    if (disposed)
+                        return;
+                    await client.openSession({ sessionPath, cwd: localRuntime.cwd });
+                    if (disposed)
+                        return;
+                    await remoteSession.refresh();
+                    if (disposed)
+                        return;
+                    fallbackWarned = false;
+                    reconnectWarningShown = false;
+                    remoteRuntime?.enterConnected();
+                    return;
+                }
+                catch (error) {
+                    cause = error;
+                    await client.stop().catch(() => { });
+                    if (disposed)
+                        return;
+                }
+            }
+            if (!disposed) {
+                await remoteRuntime?.enterFallback();
+                warnFallback(cause);
+            }
+        })().finally(() => {
+            reconnecting = undefined;
+            if (disconnectQueued && !disposed && !remoteRuntime?.isFallback)
+                scheduleReconnect();
+        });
+    };
     try {
         await startHost({ socket: options.socket, agentDir: options.agentDir });
         await client.start();
@@ -28,7 +104,12 @@ export async function createInteractiveHostRuntime(localRuntime, options) {
             modelId: localRuntime.session.model?.id,
             thinkingLevel: localRuntime.session.thinkingLevel,
         });
-        const remoteSession = createRemoteSessionProxy(localRuntime.session, localRuntime.services.agentDir, client, opened.state, options.onWarning);
+        remoteSession = createRemoteSessionProxy(localRuntime.session, localRuntime.services.agentDir, client, opened.state, options.onWarning, (cause) => {
+            if (remoteRuntime?.isReconnecting)
+                warnReconnect(cause);
+            else
+                warnFallback(cause);
+        });
         if (opened.state.isBashRunning && !opened.attached) {
             // Only a newly opened session can have an execution orphaned by a
             // previous connection. An attach is an observer of the same live runtime;
@@ -37,15 +118,25 @@ export async function createInteractiveHostRuntime(localRuntime, options) {
         }
         if (opened.attached)
             await remoteSession.refresh();
-        return new RemoteInteractiveRuntime(localRuntime, remoteSession, client);
+        remoteRuntime = new RemoteInteractiveRuntime(localRuntime, remoteSession, client, {
+            onTransportGone: (cause) => {
+                if (remoteRuntime?.isReconnecting)
+                    warnReconnect(cause);
+                else
+                    warnFallback(cause);
+            },
+            get reconnecting() {
+                return reconnecting;
+            },
+            dispose: () => {
+                disposed = true;
+            },
+        });
+        return remoteRuntime;
     }
     catch (cause) {
         await client.stop().catch(() => { });
-        options.onWarning?.({
-            type: "interactive_host_fallback",
-            message: `${INTERACTIVE_HOST_FALLBACK_WARNING}: ${cause instanceof Error ? cause.message : String(cause)}`,
-            cause,
-        });
+        warnFallback(cause);
         return localRuntime;
     }
 }
@@ -53,21 +144,68 @@ export class RemoteInteractiveRuntime {
     #local;
     #remoteSession;
     #client;
+    #lifecycle;
+    #onTransportGone;
     #rebindSession;
     #beforeSessionInvalidate;
-    constructor(local, remoteSession, client) {
+    #state = "connected";
+    #fallbackHandoff;
+    get isReconnecting() {
+        return this.#state === "reconnecting";
+    }
+    get isFallback() {
+        return this.#state === "fallback";
+    }
+    constructor(local, remoteSession, client, lifecycle) {
         this.#local = local;
         this.#remoteSession = remoteSession;
         this.#client = client;
+        this.#lifecycle = lifecycle;
+        this.#onTransportGone = lifecycle?.onTransportGone ?? (() => { });
+    }
+    async #call(call) {
+        try {
+            return await call();
+        }
+        catch (error) {
+            if (isTransportGoneError(error)) {
+                this.#onTransportGone(error);
+                return { cancelled: true };
+            }
+            throw error;
+        }
+    }
+    enterConnected() {
+        if (this.#state !== "disposed" && this.#state !== "fallback")
+            this.#state = "connected";
+    }
+    enterReconnecting() {
+        if (this.#state !== "disposed" && this.#state !== "fallback")
+            this.#state = "reconnecting";
+    }
+    async enterFallback() {
+        if (this.#state === "disposed" || this.#state === "fallback")
+            return;
+        this.#state = "fallback";
+        this.#beforeSessionInvalidate?.();
+        const rebind = this.#rebindSession?.();
+        if (rebind) {
+            this.#fallbackHandoff = rebind.catch((error) => {
+                this.#onTransportGone(error);
+            });
+            await this.#fallbackHandoff;
+        }
     }
     get session() {
-        return this.#remoteSession.session;
+        return this.#state === "fallback" || this.#state === "disposed"
+            ? this.#local.session
+            : this.#remoteSession.session;
     }
     get services() {
         return this.#local.services;
     }
     get cwd() {
-        return this.#remoteSession.session.sessionManager.getCwd();
+        return this.session.sessionManager.getCwd();
     }
     get diagnostics() {
         return this.#local.diagnostics;
@@ -80,14 +218,27 @@ export class RemoteInteractiveRuntime {
     }
     setBeforeSessionInvalidate(callback) {
         this.#beforeSessionInvalidate = callback;
+        this.#local.setBeforeSessionInvalidate(callback);
     }
     setRebindSession(callback) {
         this.#rebindSession = callback;
+        this.#local.setRebindSession(callback);
+        if (callback && this.#state === "fallback" && !this.#fallbackHandoff) {
+            this.#fallbackHandoff = callback().catch((error) => {
+                this.#onTransportGone(error);
+            });
+        }
     }
     setHostUiHandler(callback) {
         this.#remoteSession.setHostUiHandler(callback);
     }
     async dispose() {
+        if (this.#state === "disposed")
+            return;
+        this.#state = "disposed";
+        this.#lifecycle?.dispose();
+        await this.#lifecycle?.reconnecting?.catch(() => { });
+        await this.#fallbackHandoff?.catch(() => { });
         const errors = [];
         for (const cleanup of [
             () => this.#remoteSession.abortLocalBash(),
@@ -106,12 +257,15 @@ export class RemoteInteractiveRuntime {
             throw errors[0];
     }
     async newSession(options) {
+        if (this.#state === "fallback")
+            return this.#local.newSession(options);
         // Suppresses for the ENTIRE command, not just the refresh tail: a
         // multi-session host emits session_replaced while completing it, so the echo
         // can arrive before the refresh/rebind sequence below starts and race it -
         // newSession transports setup entries between two refreshes.
-        return this.#remoteSession.aroundLocalReplacement(async () => {
-            const result = await this.#client.newSession(options?.parentSession);
+        return this.#remoteSession
+            .aroundLocalReplacement(async () => {
+            const result = await this.#call(() => this.#client.newSession(options?.parentSession));
             if (!result.cancelled) {
                 this.#beforeSessionInvalidate?.();
                 this.#remoteSession.abortLocalBash();
@@ -128,13 +282,21 @@ export class RemoteInteractiveRuntime {
                     await options.withSession(this.#remoteSession.createReplacedSessionContext());
             }
             return result;
+        })
+            .catch((error) => {
+            if (isTransportGoneError(error))
+                return { cancelled: true };
+            throw error;
         });
     }
     async switchSession(sessionPath, options) {
+        if (this.#state === "fallback")
+            return this.#local.switchSession(sessionPath, options);
         // See newSession: the suppression must cover the command itself, since the
         // host emits session_replaced while completing it.
-        return this.#remoteSession.aroundLocalReplacement(async () => {
-            const result = await this.#client.switchSession(sessionPath, options?.cwdOverride === undefined ? undefined : { cwdOverride: options.cwdOverride });
+        return this.#remoteSession
+            .aroundLocalReplacement(async () => {
+            const result = await this.#call(() => this.#client.switchSession(sessionPath, options?.cwdOverride === undefined ? undefined : { cwdOverride: options.cwdOverride }));
             if (!result.cancelled) {
                 this.#beforeSessionInvalidate?.();
                 this.#remoteSession.abortLocalBash();
@@ -147,12 +309,20 @@ export class RemoteInteractiveRuntime {
                     await options.withSession(this.#remoteSession.createReplacedSessionContext());
             }
             return result;
+        })
+            .catch((error) => {
+            if (isTransportGoneError(error))
+                return { cancelled: true };
+            throw error;
         });
     }
     async fork(entryId, options) {
+        if (this.#state === "fallback")
+            return this.#local.fork(entryId, options);
         // See newSession: the suppression must cover the command itself.
-        return this.#remoteSession.aroundLocalReplacement(async () => {
-            const result = await this.#client.fork(entryId, options);
+        return this.#remoteSession
+            .aroundLocalReplacement(async () => {
+            const result = await this.#call(() => this.#client.fork(entryId, options));
             if (!result.cancelled) {
                 this.#beforeSessionInvalidate?.();
                 this.#remoteSession.abortLocalBash();
@@ -161,18 +331,31 @@ export class RemoteInteractiveRuntime {
                     await options.withSession(this.#remoteSession.createReplacedSessionContext());
             }
             return { cancelled: result.cancelled, selectedText: result.text };
+        })
+            .catch((error) => {
+            if (isTransportGoneError(error))
+                return { cancelled: true };
+            throw error;
         });
     }
     async importFromJsonl(inputPath, cwdOverride) {
+        if (this.#state === "fallback")
+            return this.#local.importFromJsonl(inputPath, cwdOverride);
         // See newSession: the suppression must cover the command itself.
-        return this.#remoteSession.aroundLocalReplacement(async () => {
-            const result = await this.#client.importJsonl(inputPath, cwdOverride);
+        return this.#remoteSession
+            .aroundLocalReplacement(async () => {
+            const result = await this.#call(() => this.#client.importJsonl(inputPath, cwdOverride));
             if (!result.cancelled) {
                 this.#beforeSessionInvalidate?.();
                 this.#remoteSession.abortLocalBash();
                 await this.#refreshAndRebind();
             }
             return result;
+        })
+            .catch((error) => {
+            if (isTransportGoneError(error))
+                return { cancelled: true };
+            throw error;
         });
     }
     async #refreshAndRebind() {
@@ -181,16 +364,31 @@ export class RemoteInteractiveRuntime {
     }
 }
 /** Exported for direct coverage of the wire-event handling below. */
-export function createRemoteSessionProxy(local, agentDir, client, initialState, onWarning) {
+export function createRemoteSessionProxy(local, agentDir, client, initialState, onWarning, onTransportGone) {
     // Fire-and-forget setters keep the sync AgentSession signature, but their RPC
     // failures must not vanish: the matching *_changed wire event confirms success,
     // and a rejection here is the only signal of failure.
     const reportActionFailure = (action) => (error) => {
-        onWarning?.({
-            type: "interactive_host_action_failed",
-            message: `Warning: shared interactive host ${action} failed: ${error instanceof Error ? error.message : String(error)}`,
-            cause: error,
-        });
+        const transportGone = isTransportGoneError(error);
+        if (transportGone)
+            onTransportGone?.(error);
+        else
+            onWarning?.({
+                type: "interactive_host_action_failed",
+                message: `Warning: shared interactive host ${action} failed: ${error instanceof Error ? error.message : String(error)}`,
+                cause: error,
+            });
+    };
+    const transportCall = async (action, call, cancelledValue) => {
+        try {
+            return await call();
+        }
+        catch (error) {
+            if (!isTransportGoneError(error))
+                throw error;
+            reportActionFailure(action)(error);
+            return cancelledValue;
+        }
     };
     let state = { ...initialState };
     const waitForBashCallbacks = async (promises, allowAbandonment) => {
@@ -262,7 +460,7 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
     const remoteSessionManager = new Proxy({}, {
         get(_target, property, _receiver) {
             if (property === "appendLabelChange") {
-                return (entryId, label) => void client.setLabel(entryId, label);
+                return (entryId, label) => void client.setLabel(entryId, label).catch(reportActionFailure("appendLabelChange"));
             }
             if (property === "getCwd")
                 return () => state.cwd;
@@ -283,7 +481,9 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
         if (wireEvent.type === "extension_ui_request") {
             const request = wireEvent;
             if (hostUiHandler)
-                void Promise.resolve(hostUiHandler(request)).then((response) => response && client.sendExtensionUIResponse(response));
+                void Promise.resolve(hostUiHandler(request))
+                    .then((response) => response && client.sendExtensionUIResponse(response))
+                    .catch(reportActionFailure("host UI response"));
             else
                 pendingUiRequests.push(request);
             return;
@@ -441,7 +641,13 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
             listener(event);
     });
     const performRefresh = async () => {
-        const nextState = await client.getState();
+        // Captured before the first await: entries that arrive via `entry_appended`
+        // while this refresh is in flight are newer than the snapshot it reconciles
+        // against, and must survive the rebuild below.
+        const idsAtRefreshStart = new Set(sessionManager.getEntries().map((entry) => entry.id));
+        const nextState = await transportCall("refresh", () => client.getState());
+        if (!nextState)
+            return;
         state = { ...stateFromRpc(nextState) };
         nextQueuedInputOrder = Math.max(0, ...nextState.ordered.map((item) => item.enqueueOrder));
         let messages;
@@ -460,14 +666,40 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
                 // deferred creating the file for a setup-only session.
                 sessionManager = SessionManager.open(nextState.sessionFile, undefined, nextState.cwd);
             }
-            if (nextState.entries?.length && !sessionManager.getEntries().length) {
-                for (const entry of nextState.entries)
-                    sessionManager.appendEntry(entry);
+            // The host ships its complete entry list while the session file is still
+            // deferred (no assistant message yet, so nothing has been written), which
+            // makes that list the only authoritative view of the session. Reconcile the
+            // mirror to it exactly: `entry_appended` notifications that cross a
+            // concurrent refresh land on whichever manager is current, so the mirror
+            // can hold an arbitrary partial set rather than a prefix, and a snapshot
+            // taken mid-bind can simply be short. Backfilling only an empty mirror
+            // wedged it at whatever partial set it happened to hold.
+            const authoritative = nextState.entries;
+            if (authoritative?.length) {
+                const mirror = sessionManager.getEntries();
+                const matches = mirror.length === authoritative.length &&
+                    mirror.every((entry, index) => entry.id === authoritative[index]?.id);
+                if (!matches) {
+                    const rebuilt = SessionManager.open(nextState.sessionFile, undefined, nextState.cwd);
+                    for (const entry of authoritative)
+                        rebuilt.appendEntry(entry);
+                    // A notification that crossed this refresh refers to an entry the
+                    // snapshot predates, so it is newer than the snapshot; dropping it
+                    // would strand the entry until an unrelated refresh happened to
+                    // run. Keep it, in arrival order, after the authoritative list.
+                    const authoritativeIds = new Set(authoritative.map((entry) => entry.id));
+                    for (const entry of mirror) {
+                        if (!idsAtRefreshStart.has(entry.id) && !authoritativeIds.has(entry.id)) {
+                            rebuilt.appendEntry(entry);
+                        }
+                    }
+                    sessionManager = rebuilt;
+                }
             }
             messages = sessionManager.buildSessionContext().messages;
         }
         else {
-            messages = await client.getMessages();
+            messages = await transportCall("refresh", () => client.getMessages(), []);
         }
         local.agent.state.messages.splice(0, local.agent.state.messages.length, ...structuredClone(messages));
         streamingAssistant = undefined;
@@ -485,23 +717,32 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
     const session = new Proxy(local, {
         get(target, property, receiver) {
             if (property === "prompt")
-                return (message, options) => client.prompt(message, {
-                    ...(options?.images ? { images: options.images } : {}),
-                    ...(options?.streamingBehavior ? { streamingBehavior: options.streamingBehavior } : {}),
-                    ...(options?.thinkingLevel ? { thinkingLevel: options.thinkingLevel } : {}),
-                    ...(options?.sessionTitlePrompt !== undefined
-                        ? { sessionTitlePrompt: options.sessionTitlePrompt }
-                        : {}),
-                    ...(options?.expandPromptTemplates !== undefined
-                        ? { expandPromptTemplates: options.expandPromptTemplates }
-                        : {}),
-                    ...(options?.promptDisposition ? { promptDisposition: options.promptDisposition } : {}),
-                    ...(options?.preflightResult ? { preflightResult: options.preflightResult } : {}),
-                });
+                return async (message, options) => {
+                    try {
+                        await client.prompt(message, {
+                            ...(options?.images ? { images: options.images } : {}),
+                            ...(options?.streamingBehavior ? { streamingBehavior: options.streamingBehavior } : {}),
+                            ...(options?.thinkingLevel ? { thinkingLevel: options.thinkingLevel } : {}),
+                            ...(options?.sessionTitlePrompt !== undefined
+                                ? { sessionTitlePrompt: options.sessionTitlePrompt }
+                                : {}),
+                            ...(options?.expandPromptTemplates !== undefined
+                                ? { expandPromptTemplates: options.expandPromptTemplates }
+                                : {}),
+                            ...(options?.promptDisposition ? { promptDisposition: options.promptDisposition } : {}),
+                            ...(options?.preflightResult ? { preflightResult: options.preflightResult } : {}),
+                        });
+                    }
+                    catch (error) {
+                        if (!isTransportGoneError(error))
+                            throw error;
+                        reportActionFailure("prompt")(error);
+                    }
+                };
             if (property === "abort")
                 return () => client.abort();
             if (property === "abortCompaction")
-                return () => void client.abortCompaction();
+                return () => void client.abortCompaction().catch(reportActionFailure("abortCompaction"));
             if (property === "steer")
                 return (message, images, recovery) => {
                     const enqueueOrder = recovery?.enqueueOrder ?? Math.max(0, ...state.ordered.map((item) => item.enqueueOrder)) + 1;
@@ -511,7 +752,7 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
                         ordered: [...state.ordered, { text: message, mode: "steer", enqueueOrder }],
                         pendingMessageCount: state.pendingMessageCount + 1,
                     };
-                    return client.steer(message, images, { ...recovery, enqueueOrder });
+                    return transportCall("steer", () => client.steer(message, images, { ...recovery, enqueueOrder }));
                 };
             if (property === "followUp")
                 return (message, images, recovery) => {
@@ -522,7 +763,7 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
                         ordered: [...state.ordered, { text: message, mode: "followUp", enqueueOrder }],
                         pendingMessageCount: state.pendingMessageCount + 1,
                     };
-                    return client.followUp(message, images, { ...recovery, enqueueOrder });
+                    return transportCall("followUp", () => client.followUp(message, images, { ...recovery, enqueueOrder }));
                 };
             if (property === "waitForIdle")
                 return () => client.waitForIdle();
@@ -530,8 +771,8 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
                 return () => target.getLastAssistantText();
             if (property === "setModel" || property === "setSessionModel")
                 return async (model) => {
-                    const next = await client.setModel(model.provider, model.id);
-                    return { systemPromptName: next.systemPromptName, model: next };
+                    const next = await transportCall("setModel", () => client.setModel(model.provider, model.id), undefined);
+                    return next ? { systemPromptName: next.systemPromptName, model: next } : undefined;
                 };
             if (property === "setSessionThinkingLevel")
                 return (level) => void client
@@ -556,19 +797,19 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
                     void client.setScopedModels(models).catch(reportActionFailure("setScopedModels"));
                 };
             if (property === "cycleModel")
-                return (direction) => client.cycleModel(direction);
+                return (direction) => transportCall("cycleModel", () => client.cycleModel(direction), undefined);
             if (property === "setThinkingLevel")
                 return (level) => void client.setThinkingLevel(level).catch(reportActionFailure("setThinkingLevel"));
             if (property === "cycleThinkingLevel")
-                return () => client.cycleThinkingLevel().then((result) => result?.level);
+                return () => transportCall("cycleThinkingLevel", () => client.cycleThinkingLevel(), undefined).then((result) => result?.level);
             if (property === "getAvailableThinkingLevels")
-                return () => client.getAvailableThinkingLevels();
+                return () => transportCall("getAvailableThinkingLevels", () => client.getAvailableThinkingLevels(), []);
             if (property === "setSteeringMode")
                 return (mode) => void client.setSteeringMode(mode).catch(reportActionFailure("setSteeringMode"));
             if (property === "setFollowUpMode")
                 return (mode) => void client.setFollowUpMode(mode).catch(reportActionFailure("setFollowUpMode"));
             if (property === "compact")
-                return (instructions) => client.compact(instructions);
+                return (instructions) => transportCall("compact", () => client.compact(instructions));
             if (property === "setAutoCompactionEnabled")
                 return (enabled) => void client.setAutoCompaction(enabled).catch(reportActionFailure("setAutoCompaction"));
             if (property === "executeBash")
@@ -584,7 +825,7 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
                         try {
                             const result = await executeBashWithOperations(resolvedCommand, state.cwd, options.operations, { onChunk, signal: abortController.signal });
                             if (state.sessionId === sessionAtStart) {
-                                await client.recordBashResult(command, result, options.excludeFromContext);
+                                await transportCall("recordBashResult", () => client.recordBashResult(command, result, options.excludeFromContext));
                             }
                             return result;
                         }
@@ -603,11 +844,11 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
                     };
                     bashExecutions.set(executionId, execution);
                     try {
-                        const result = await client.bash(command, {
+                        const result = await transportCall("bash", () => client.bash(command, {
                             excludeFromContext: options?.excludeFromContext,
                             executionId,
                             operations: options?.operations,
-                        });
+                        }), { output: "", exitCode: undefined, cancelled: true, truncated: false });
                         const callbacksSettled = await waitForBashCallbacks(execution.promises, false);
                         if (!callbacksSettled) {
                             if (result.fullOutputPath)
@@ -635,20 +876,22 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
             if (property === "getContextUsage")
                 return () => state.contextUsage;
             if (property === "getSessionStats")
-                return () => client.getSessionStats();
+                return () => transportCall("getSessionStats", () => client.getSessionStats(), local.getSessionStats());
             if (property === "exportToHtml")
-                return (outputPath, options) => client.exportHtml(outputPath, options?.themeName).then((result) => result.path);
+                return (outputPath, options) => transportCall("exportToHtml", () => client.exportHtml(outputPath, options?.themeName)).then((result) => result?.path);
             if (property === "setSessionName")
                 return (name) => client.setSessionName(name).catch(reportActionFailure("setSessionName"));
             if (property === "navigateTree")
                 return async (targetId, options) => {
-                    const result = await client.navigateTree(targetId, options);
+                    const result = await transportCall("navigateTree", () => client.navigateTree(targetId, options), {
+                        cancelled: true,
+                    });
                     if (!result.cancelled)
                         await refresh();
                     return result;
                 };
             if (property === "getUserMessagesForForking")
-                return () => client.getForkMessages();
+                return () => transportCall("getUserMessagesForForking", () => client.getForkMessages(), []);
             if (property === "subscribe")
                 return (listener) => {
                     listeners.add(listener);
@@ -694,11 +937,11 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
             if (property === "isBashRunning")
                 return state.isBashRunning;
             if (property === "reload")
-                return (_options) => client.reload();
+                return (_options) => transportCall("reload", () => client.reload(), { cancelled: true });
             if (property === "checkReloadVeto")
-                return () => client.checkReloadVeto();
+                return () => transportCall("checkReloadVeto", () => client.checkReloadVeto(), { cancelled: true });
             if (property === "exportToJsonl")
-                return (outputPath) => client.exportJsonl(outputPath).then((result) => result.path);
+                return (outputPath) => transportCall("exportToJsonl", () => client.exportJsonl(outputPath)).then((result) => result?.path);
             // The footer and other renderers read session.state.*; surface the
             // host-authoritative fields there too, not only via the direct getters.
             if (property === "state") {
@@ -750,7 +993,9 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
             hostUiHandler = callback;
             if (callback)
                 for (const request of pendingUiRequests.splice(0))
-                    void Promise.resolve(callback(request)).then((response) => response && client.sendExtensionUIResponse(response));
+                    void Promise.resolve(callback(request))
+                        .then((response) => response && client.sendExtensionUIResponse(response))
+                        .catch(reportActionFailure("host UI response"));
         },
         refresh,
         aroundLocalReplacement: async (body) => {
@@ -767,26 +1012,26 @@ export function createRemoteSessionProxy(local, agentDir, client, initialState,
             const context = local.createReplacedSessionContext();
             Object.defineProperty(context, "cwd", { value: state.cwd });
             Object.defineProperty(context, "sessionManager", { value: remoteSessionManager });
-            context.sendMessage = (message, options) => client.sendCustomMessage(message, {
+            context.sendMessage = (message, options) => transportCall("sendMessage", () => client.sendCustomMessage(message, {
                 triggerTurn: options?.triggerTurn,
                 deliverAs: options?.deliverAs,
-            });
+            }));
             context.sendUserMessage = (content, options) => {
                 if (typeof content === "string")
-                    return client.prompt(content, {
+                    return transportCall("sendUserMessage", () => client.prompt(content, {
                         streamingBehavior: options?.deliverAs,
                         expandPromptTemplates: options?.expandPromptTemplates,
-                    });
+                    }));
                 const text = content
                     .filter((part) => part.type === "text")
                     .map((part) => part.text)
                     .join("\n");
                 const images = content.filter((part) => part.type === "image");
-                return client.prompt(text, {
+                return transportCall("sendUserMessage", () => client.prompt(text, {
                     images,
                     streamingBehavior: options?.deliverAs,
                     expandPromptTemplates: options?.expandPromptTemplates,
-                });
+                }));
             };
             return context;
         },