@dreb/coding-agent 2.70.0 → 2.71.0

package/dist/modes/rpc/rpc-mode.js

@@ -19,17 +19,17 @@ import { DailyCostTracker, dailyCostSourcesForSession, } from "../../core/daily-
 import { acquireDreamLock, buildDreamPrompt, cleanupDreamTmpDirs, parseDreamCommand, performDreamBackup, pruneOldBackups, resolveDreamContext, validateArchivePath, validateMemoryLinks, } from "../../core/dream.js";
 import { getGitBranch } from "../../core/git-branch.js";
 import { parseModelPattern, resolveModelScopePatterns } from "../../core/model-resolver.js";
-import { takeOverStdout, writeRawStdout } from "../../core/output-guard.js";
+import { isFatalExitPending, takeOverStdout, writeRawStdout } from "../../core/output-guard.js";
 import { SessionManager } from "../../core/session-manager.js";
 import { BUILTIN_SLASH_COMMANDS, parseBuiltinSlashCommand } from "../../core/slash-commands.js";
 import { SubagentSessionCostTracker } from "../../core/subagent-session-cost-tracker.js";
 import { TabTitleGenerator } from "../../core/tab-title.js";
 import { validateThinkingLevelForModel } from "../../core/thinking.js";
 import { resolveToCwd } from "../../core/tools/path-utils.js";
-import { discoverAgentTypes, getBackgroundAgentPendingSteering, getBackgroundAgents, rehydrateBackgroundAgentsFromDisk, steerBackgroundAgent, } from "../../core/tools/subagent.js";
+import { discoverAgentTypes, getBackgroundAgentPendingSteering, getBackgroundAgents, rehydrateBackgroundAgentsFromDisk, setSubagentRpcFullMessageUpdates, steerBackgroundAgent, } from "../../core/tools/subagent.js";
 import { theme } from "../interactive/theme/theme.js";
 import { attachJsonlLineReader, serializeJsonLine } from "./jsonl.js";
-import { createDashboardRpcEventProjector } from "./rpc-event-projection.js";
+import { createDashboardRpcEventProjector, projectRpcMessageUpdates } from "./rpc-event-projection.js";
 /**
  * Map a core {@link SessionInfo} to the RPC DTO, converting Date fields to ISO strings.
  * Shared by the `list_sessions` and `list_all_sessions` handlers so their shapes cannot drift.
@@ -1419,7 +1419,7 @@ export function cancelPendingRpcExtensionRequests(pendingExtensionRequests) {
         pending.resolve({ type: "extension_ui_response", id, cancelled: true });
     }
 }
-export async function runRpcMode(session, modelFallbackMessage) {
+export async function runRpcMode(session, modelFallbackMessage, options = {}) {
     takeOverStdout();
     const output = (obj) => {
         writeRawStdout(serializeJsonLine(obj));
@@ -1524,17 +1524,25 @@ export async function runRpcMode(session, modelFallbackMessage) {
             },
         })
         : undefined;
-    // Dashboard-launched runtimes (--ui dashboard) get events projected before
-    // serialization: message_update's cumulative `message` and
-    // `assistantMessageEvent.partial` fields are quadratic in response length on
-    // the JSONL pipe, and no dashboard consumer reads them (deltas, message_end,
-    // and get_dashboard_snapshot responses carry the authoritative data). Inline
-    // image blocks are deduped over the process lifetime so each unique image
-    // crosses stdout at most once — later occurrences become image_references
-    // the dashboard resolves from its image cache (issue 495). Generic RPC
-    // consumers keep the full protocol unchanged. Only the event stream is
+    // Events are projected before serialization. message_update's cumulative
+    // `message` and `assistantMessageEvent.partial` fields are quadratic in
+    // response length on the JSONL pipe (issues 448, 535), so every RPC runtime
+    // strips them by default; deltas, message_end, and command responses carry
+    // the authoritative data. `--rpc-full-message-updates` restores the legacy
+    // full frames for generic clients. Dashboard runtimes (--ui dashboard) are
+    // always projected and additionally dedupe inline images over the process
+    // lifetime — later occurrences become image_references the dashboard
+    // resolves from its image cache (issue 495). Only the event stream is
     // projected — command responses (output() calls below) always stay complete.
-    const dashboardEventProjector = session.uiType === "dashboard" ? createDashboardRpcEventProjector() : undefined;
+    const eventProjector = session.uiType === "dashboard"
+        ? createDashboardRpcEventProjector()
+        : options.fullMessageUpdates
+            ? undefined
+            : projectRpcMessageUpdates;
+    // Forward the opt-in to RPC-controlled subagent children: their relayed
+    // background_agent_event payloads are projected inside the child process and
+    // cannot be restored here, so the child must emit full frames itself.
+    setSubagentRpcFullMessageUpdates(eventProjector === undefined);
     // Output all agent events as JSON
     session.subscribe((event) => {
         if (event.type === "background_agent_end") {
@@ -1553,7 +1561,7 @@ export async function runRpcMode(session, modelFallbackMessage) {
                 tabTitleGenerator.onMessageEnd(event.message);
             }
         }
-        output(dashboardEventProjector ? dashboardEventProjector(event) : event);
+        output(eventProjector ? eventProjector(event) : event);
     });
     // Handle a single command
     const handleCommand = async (command) => {
@@ -1993,6 +2001,12 @@ export async function runRpcMode(session, modelFallbackMessage) {
         subagentSessionCostTracker?.dispose();
         detachInput();
         process.stdin.pause();
+        if (isFatalExitPending()) {
+            // A stdout write failure is already exiting the process nonzero (a dying
+            // consumer often closes stdin and stdout together). Exiting 0 here would
+            // truncate its diagnostic and report success; let the fatal path finish.
+            return new Promise(() => { });
+        }
         process.exit(0);
     }
     async function checkShutdownRequested() {