getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-17 01:04:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(plan-mode): buffer text_delta to handle split [DONE:id] patterns

Browse source 
The [DONE:id] pattern may be split across multiple streaming chunks.
Now accumulates text in a buffer and scans for complete patterns.
This commit is contained in:
Helmut Januschka 2026-01-03 16:28:53 +01:00 • committed by Mario Zechner
parent d1eea3ac4e
commit e27a2c226c
1 changed files with 16 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

17
packages/coding-agent/examples/hooks/plan-mode.ts
View file

@ -291,11 +291,18 @@ export default function planModeHook(pi: HookAPI) {
} }
}); });
// Buffer for accumulating text to handle [DONE:id] split across chunks
let textBuffer = "";
// Watch for [DONE:id] tags in streaming text // Watch for [DONE:id] tags in streaming text
pi.on("text_delta", async (event, ctx) => { pi.on("text_delta", async (event, ctx) => {
if (!executionMode || todoItems.length === 0) return; if (!executionMode || todoItems.length === 0) return;
const doneIds = findDoneTags(event.text); // Accumulate text in buffer
textBuffer += event.text;
// Look for complete [DONE:id] patterns
const doneIds = findDoneTags(textBuffer);
if (doneIds.length === 0) return; if (doneIds.length === 0) return;
let changed = false; let changed = false;
@ -307,6 +314,11 @@ export default function planModeHook(pi: HookAPI) {
} }
} }
// Clear processed patterns from buffer (keep last 20 chars for partial matches)
if (textBuffer.length > 50) {
textBuffer = textBuffer.slice(-20);
}
if (changed) { if (changed) {
updateStatus(ctx); updateStatus(ctx);
} }
@ -360,6 +372,9 @@ IMPORTANT: After completing each step, output [DONE:id] where id is the step's I
// After agent finishes in plan mode // After agent finishes in plan mode
pi.on("agent_end", async (event, ctx) => { pi.on("agent_end", async (event, ctx) => {
// Clear text buffer
textBuffer = "";
// Check for done tags in the final message too // Check for done tags in the final message too
if (executionMode && todoItems.length > 0) { if (executionMode && todoItems.length > 0) {
const messages = event.messages; const messages = event.messages;