wassname/pi-dynamic-context-pruning
1
0
Fork 0
mirror of https://github.com/wassname/pi-dynamic-context-pruning.git synced 2026-06-27 18:05:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixing some errors and issues

Browse Source
This commit is contained in:
Greg Harvell
2026-03-27 19:38:35 -04:00
parent e4b4d13ac7
commit e62d04dfe8
3 changed files with 309 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pruner.ts
+24 -1
View File
@@ -278,12 +278,35 @@ function injectMessageIds(messages: any[], state: DcpState): void {
} else if (Array.isArray(msg.content)) {
msg.content = [...msg.content, { type: "text", text: idTag }];
}
} else if (role === "assistant" || role === "toolResult" || role === "bashExecution") {
} else if (role === "toolResult" || role === "bashExecution") {
if (Array.isArray(msg.content)) {
msg.content = [...msg.content, { type: "text", text: idTag }];
} else if (typeof msg.content === "string") {
msg.content = msg.content + idTag;
}
} else if (role === "assistant") {
if (Array.isArray(msg.content)) {
// Insert the ID tag before any tool_use (toolCall) blocks.
// Anthropic requires: thinking → text → tool_use.
// Appending after tool_use blocks violates that constraint.
const firstToolCallIdx = msg.content.findIndex(
(b: any) => b.type === "toolCall",
);
const idBlock = { type: "text", text: idTag };
if (firstToolCallIdx === -1) {
// No tool_use blocks — append as usual
msg.content = [...msg.content, idBlock];
} else {
// Insert immediately before the first tool_use block
msg.content = [
...msg.content.slice(0, firstToolCallIdx),
idBlock,
...msg.content.slice(firstToolCallIdx),
];
}
} else if (typeof msg.content === "string") {
msg.content = msg.content + idTag;
}
}
if (msg.timestamp !== undefined) {