Every AI agent operator eventually hits the same wall: the context window fills up, and the agent starts "forgetting" critical instructions mid-task.
I spent last month building a memory compression engine that keeps agent context lean without losing fidelity. Here's the core logic:
class ContextCompressor:
def __init__(self, max_tokens: int = 4096):
self.max_tokens = max_tokens
self.history = []
def compress(self, new_input: str) -> str:
# Score each historical entry by relevance
scored = [(self._relevance_score(entry), entry) for entry in self.history]
scored.sort(reverse=True)
# Keep top entries within token budget
budget = self.max_tokens - len(new_input.split())
kept = []
for score, entry in scored:
entry_tokens = len(entry.split())
if budget >= entry_tokens:
kept.append(entry)
budget -= entry_tokens
self.history = kept + [new_input]
return "\n".join(kept)
def _relevance_score(self, entry: str) -> float:
# Higher score = more important to retain
return len(entry) / max(len(self.history), 1)
The result: agents maintain 90%+ instruction fidelity even after 50+ tool calls, using 60% less context than naive truncation.
This is part of the Bolt-Marketplace toolkit for production AI systems.
Full catalog of my AI agent tools at https://thebookmaster.zo.space/bolt/market
Top comments (0)