dont prune pending or running tool inputs

jorgenwh · jorgenwh · commit 063a7438f1a6 · 2025-12-18T23:13:51.000+01:00
diff --git a/lib/messages/prune.ts b/lib/messages/prune.ts
@@ -151,6 +151,11 @@ const pruneToolInputs = (
             if (part.tool !== 'write' && part.tool !== 'edit') {
                 continue
             }
+            // Don't prune yet if tool is still pending or running
+            if (part.state.status === "pending" || part.state.status === "running") {
+                continue
+            }
+
             if (part.state.input?.content !== undefined) {
                 part.state.input.content = PRUNED_TOOL_INPUT_REPLACEMENT
             }

Original file line number	Diff line number	Diff line change
`@@ -151,6 +151,11 @@ const pruneToolInputs = (`
`151`	`151`	`if (part.tool !== 'write' && part.tool !== 'edit') {`
`152`	`152`	`continue`
`153`	`153`	`}`
	`154`	`+ // Don't prune yet if tool is still pending or running`
	`155`	`+ if (part.state.status === "pending" \|\| part.state.status === "running") {`
	`156`	`+ continue`
	`157`	`+ }`
	`158`	`+`
`154`	`159`	`if (part.state.input?.content !== undefined) {`
`155`	`160`	`part.state.input.content = PRUNED_TOOL_INPUT_REPLACEMENT`
`156`	`161`	`}`