From 38ba14ea9f9fbb32d5bfb9b103be12a7b47f4d94 Mon Sep 17 00:00:00 2001 From: 1alyx Date: Thu, 29 Jan 2026 23:07:12 -0500 Subject: [PATCH] fix: trim leading newlines from streaming/block-flush text output Co-authored-by: Zach Canepa --- src/agents/pi-embedded-helpers/errors.ts | 4 ++-- src/agents/pi-embedded-subscribe.ts | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/agents/pi-embedded-helpers/errors.ts b/src/agents/pi-embedded-helpers/errors.ts index 849c4293e..3b2f47f56 100644 --- a/src/agents/pi-embedded-helpers/errors.ts +++ b/src/agents/pi-embedded-helpers/errors.ts @@ -81,7 +81,7 @@ function collapseConsecutiveDuplicateBlocks(text: string): string { const trimmed = text.trim(); if (!trimmed) return text; const blocks = trimmed.split(/\n{2,}/); - if (blocks.length < 2) return text; + if (blocks.length < 2) return trimmed; const normalizeBlock = (value: string) => value.trim().replace(/\s+/g, " "); const result: string[] = []; @@ -344,7 +344,7 @@ export function sanitizeUserFacingText(text: string): string { return formatRawAssistantErrorForUi(trimmed); } - return collapseConsecutiveDuplicateBlocks(stripped); + return collapseConsecutiveDuplicateBlocks(stripped).trim(); } export function isRateLimitAssistantError(msg: AssistantMessage | undefined): boolean { diff --git a/src/agents/pi-embedded-subscribe.ts b/src/agents/pi-embedded-subscribe.ts index a4a4b906a..978f60d3a 100644 --- a/src/agents/pi-embedded-subscribe.ts +++ b/src/agents/pi-embedded-subscribe.ts @@ -359,7 +359,7 @@ export function subscribeEmbeddedPiSession(params: SubscribeEmbeddedPiSessionPar const emitBlockChunk = (text: string) => { if (state.suppressBlockChunks) return; // Strip and blocks across chunk boundaries to avoid leaking reasoning. - const chunk = stripBlockTags(text, state.blockState).trimEnd(); + const chunk = stripBlockTags(text, state.blockState).trim(); if (!chunk) return; if (chunk === state.lastBlockReplyText) return;