github/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-09 11:47:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix (tui): preserve streamed text across tool boundary deltas

Browse Source
This commit is contained in:
Vignesh Natarajan
2026-02-14 18:40:54 -08:00
parent 725741486f
commit 7d7ab8a09a
1 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/tui/tui-stream-assembler.ts
View File

@@ -93,7 +93,12 @@ export class TuiStreamAssembler {
return state; return state;
} }
private updateRunState(state: RunStreamState, message: unknown, showThinking: boolean) { private updateRunState(
state: RunStreamState,
message: unknown,
showThinking: boolean,
opts?: { protectBoundaryDrops?: boolean },
) {
const thinkingText = extractThinkingFromMessage(message); const thinkingText = extractThinkingFromMessage(message);
const contentText = extractContentFromMessage(message); const contentText = extractContentFromMessage(message);
const { textBlocks, sawNonTextContentBlocks } = extractTextBlocksAndSignals(message); const { textBlocks, sawNonTextContentBlocks } = extractTextBlocksAndSignals(message);
@@ -102,8 +107,19 @@ export class TuiStreamAssembler {
state.thinkingText = thinkingText; state.thinkingText = thinkingText;
} }
if (contentText) { if (contentText) {
state.contentText = contentText; const nextContentBlocks = textBlocks.length > 0 ? textBlocks : [contentText];
state.contentBlocks = textBlocks.length > 0 ? textBlocks : [contentText]; const shouldPreserveBoundaryDroppedText =
opts?.protectBoundaryDrops === true &&
(state.sawNonTextContentBlocks || sawNonTextContentBlocks) &&
isDroppedBoundaryTextBlockSubset({
streamedTextBlocks: state.contentBlocks,
finalTextBlocks: nextContentBlocks,
});
if (!shouldPreserveBoundaryDroppedText) {
state.contentText = contentText;
state.contentBlocks = nextContentBlocks;
}
} }
if (sawNonTextContentBlocks) { if (sawNonTextContentBlocks) {
state.sawNonTextContentBlocks = true; state.sawNonTextContentBlocks = true;
@@ -121,7 +137,7 @@ export class TuiStreamAssembler {
ingestDelta(runId: string, message: unknown, showThinking: boolean): string | null { ingestDelta(runId: string, message: unknown, showThinking: boolean): string | null {
const state = this.getOrCreateRun(runId); const state = this.getOrCreateRun(runId);
const previousDisplayText = state.displayText; const previousDisplayText = state.displayText;
this.updateRunState(state, message, showThinking); this.updateRunState(state, message, showThinking, { protectBoundaryDrops: true });
if (!state.displayText || state.displayText === previousDisplayText) { if (!state.displayText || state.displayText === previousDisplayText) {
return null; return null;