SWE-Pruner Pro

SWE-Pruner Pro uses agent's internal representations for efficient context pruning in coding agents

SWE-Pruner Pro is a new context pruning method for coding agents that leverages the agent's own internal representations to decide which lines of tool output to keep or discard, rather than relying on a separate classifier. A lightweight head converts these internal representations into keep-or-prune labels per line, augmented with a length-aware embedding. Evaluated across two open-weight backbones and four multi-turn benchmarks, the method saves up to 39% of prompt and completion tokens while maintaining task quality, and on MiMo-V2-Flash additionally improves SWE-Bench Verified resolve rate by +3.8%.