github/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-10 03:12:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(agents): reuse embedded block flush helper

Browse Source
This commit is contained in:
Peter Steinberger
2026-02-18 18:54:15 +00:00
parent fa5902f210
commit e1419f3a02
2 changed files with 2 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/agents/pi-embedded-subscribe.handlers.lifecycle.ts
View File

@@ -67,15 +67,7 @@ export function handleAgentEnd(ctx: EmbeddedPiSubscribeContext) {
}); });
} }
if (ctx.params.onBlockReply) { ctx.flushBlockReplyBuffer();
if (ctx.blockChunker?.hasBuffered()) {
ctx.blockChunker.drain({ force: true, emit: ctx.emitBlockChunk });
ctx.blockChunker.reset();
} else if (ctx.state.blockBuffer.length > 0) {
ctx.emitBlockChunk(ctx.state.blockBuffer);
ctx.state.blockBuffer = "";
}
}
ctx.state.blockState.thinking = false; ctx.state.blockState.thinking = false;
ctx.state.blockState.final = false; ctx.state.blockState.final = false;

8
src/agents/pi-embedded-subscribe.handlers.messages.ts
View File

@@ -201,13 +201,7 @@ export function handleMessageUpdate(
} }
if (evtType === "text_end" && ctx.state.blockReplyBreak === "text_end") { if (evtType === "text_end" && ctx.state.blockReplyBreak === "text_end") {
if (ctx.blockChunker?.hasBuffered()) { ctx.flushBlockReplyBuffer();
ctx.blockChunker.drain({ force: true, emit: ctx.emitBlockChunk });
ctx.blockChunker.reset();
} else if (ctx.state.blockBuffer.length > 0) {
ctx.emitBlockChunk(ctx.state.blockBuffer);
ctx.state.blockBuffer = "";
}
} }
} }