cat-shark/pi-commandcode-provider
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: flush remaining thinking blocks after stream ends

Browse Source
This commit is contained in:
Patrick Wozniak
2025-05-05 10:45:00 +02:00
parent 38deb03c6c
commit 29f4565a53
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
index.ts
+11
View File
@@ -373,10 +373,21 @@ function streamCommandCode(
} }
} }
// End any lingering text block
if (textBlock) { if (textBlock) {
stream.push({ type: "text_end", contentIndex: currentTextIdx, content: textBlock.text, partial: output }); stream.push({ type: "text_end", contentIndex: currentTextIdx, content: textBlock.text, partial: output });
} }
// Emit remaining thinking (may arrive after finish without reasoning-end)
if (thinkingBlock.length > 0) {
const thinkingText = thinkingBlock.join("");
output.content.push({ type: "thinking", thinking: thinkingText });
const idx = output.content.length - 1;
stream.push({ type: "thinking_start", contentIndex: idx, partial: output });
stream.push({ type: "thinking_delta", contentIndex: idx, delta: thinkingText, partial: output });
stream.push({ type: "thinking_end", contentIndex: idx, content: thinkingText, partial: output });
}
stream.push({ type: "done", reason: output.stopReason as "stop" | "length" | "toolUse", message: output }); stream.push({ type: "done", reason: output.stopReason as "stop" | "length" | "toolUse", message: output });
stream.end(); stream.end();
} catch (error: any) { } catch (error: any) {