@animalabs/context-manager
v0.18.0
Published
Context management layer for LLM-based agents using Chronicle and Membrane
Readme
context-manager
@animalabs/context-manager keeps an LLM agent's conversation history and compiles it into the messages for each model call, within a token budget. Every message is stored in a Chronicle store, and a strategy decides what each compile sends. The autobiographical strategy folds older history into summaries in the agent's own words, written by the model you name, so a long-running agent's messages stay in the store while each compile fits its budget.
It's part of the Connectome ecosystem, with chronicle, membrane, agent-framework and connectome-host. agent-framework opens one ContextManager for each agent on its shared Chronicle store, under the namespace agents/<name>. It compiles each request from that manager and calls tick() to run compression. Compiled messages are membrane's NormalizedMessages, and the strategy writes summaries through membrane.
Install
npm install @animalabs/context-managerIt needs Node 22.12 or later, as its package.json declares, and depends on @animalabs/chronicle and @animalabs/membrane. To compress with the autobiographical strategy you'll construct a Membrane yourself, so install @animalabs/membrane alongside it.
Quick start
import { ContextManager, PassthroughStrategy } from '@animalabs/context-manager';
// Opens the Chronicle store at this path, creating it if it doesn't exist.
const manager = await ContextManager.open({
path: './agent-store',
strategy: new PassthroughStrategy(),
});
manager.addMessage('User', [{ type: 'text', text: 'Hello!' }]);
manager.addMessage('Claude', [{ type: 'text', text: 'Hi. What are we working on?' }]);
const { messages, systemInjections } = await manager.compile({
maxTokens: 100_000, // the whole window, the response included
reserveForResponse: 4_000,
});
// `messages` go into a membrane request. Injections you pass to compile() as
// its second argument with position 'system' come back in `systemInjections`,
// as content blocks for your system prompt.
manager.close();To use a store your application already holds, pass it as store (a Chronicle JsStore) instead of path. Your application then closes it. Several managers can share one store. namespace gives each its own context log and strategy state, summaries included, and isolate: true with a namespace gives it its own messages too.
Strategies
| Strategy | What a compile sends |
|---|---|
| PassthroughStrategy (the default) | Every message as stored. When they don't all fit, the most recent ones that do. |
| WindowedPassthroughStrategy | Every message since an anchor, without compression. The anchor moves forward in coarse jumps when the window outgrows the budget, or wherever setAnchor puts it. Between moves the window only grows at its end, which keeps the compiled prefix stable for the provider's prompt cache, apart from images it strips past its live-image limits. |
| AutobiographicalStrategy | Recent messages as they are, and older history as summaries the agent's model writes, merged into higher levels as they accumulate. |
| KnowledgeStrategy | An AutobiographicalStrategy for knowledge-extraction work. It chunks at phase transitions, compresses each phase with its own prompts, and keeps unresolved leads across levels. |
The autobiographical strategy
import { Membrane, AnthropicAdapter } from '@animalabs/membrane';
import { ContextManager, AutobiographicalStrategy } from '@animalabs/context-manager';
const manager = await ContextManager.open({
path: './agent-store',
// The strategy writes summaries through this membrane. The adapter reads
// ANTHROPIC_API_KEY unless you pass { apiKey }.
membrane: new Membrane(new AnthropicAdapter()),
strategy: new AutobiographicalStrategy({
// Required: the model whose voice the summaries are in.
compressionModel: 'claude-sonnet-4-5',
}),
});
manager.addMessage('User', [{ type: 'text', text: 'Hello!' }]);
// ...more messages as the conversation goes...
await manager.tick(); // compresses the next waiting chunk or merge, if any
const { messages } = await manager.compile({ maxTokens: 200_000, reserveForResponse: 8_000 });
manager.close();How it works, with each option's default in parentheses:
- The newest
recentWindowTokens(30,000) stay raw. Older messages are cut into chunks of abouttargetChunkTokens(3,000), at message boundaries. Give compile a budget with room for the raw window and the summaries. A compile that can't fit is refused withOverBudgetError. - Each
tick()does at most one piece of compression work: it compresses the next waiting chunk, or else runs the next waiting merge. So call it regularly, for instance after each turn. For a chunk, the strategy askscompressionModel, through membrane, what it recalls from that part of the conversation. The answer is a level-1 summary, which takes the chunk's place in compiled context. SetautoTickOnNewMessage: true(false) to start a tick in the background on each new message instead of callingtick()yourself. compressionModelhas no default. Without it, the strategy refuses to write summaries, because a substitute model writing in the agent's voice would corrupt the record. Without a membrane,tick()only logs a warning.- A summary renders where its messages were, as a short exchange: a
Context Managerturn with a header such as[Recall L1-3], answered with the summary in a turn fromsummaryParticipant(Claude). With adaptive resolution (below), theContext Managerturn holdssummaryContextLabel("What do you remember from earlier?") instead. - Summaries merge into one a level up once
mergeThreshold(6) of them sit unmerged at a level. Withhierarchical(true) set to false, they stay at one level. headWindowTokens(0) keeps the start of the conversation raw as well, for instructions that should stay verbatim.- With
adaptiveResolution: true(false), each compile chooses how deeply to fold each part of history to fit the budget.foldingStrategynames the solver:flat-profile(the default),oldest-first,kv-stableorkv-unified.kv-unifiedneeds an explicitkvUnifiedpolicy.
With a strategy that supports them, as this one does, the manager also offers:
pinRangeandmarkDocumentkeep messages at their place in the conversation, raw by default. A pin that setslevelormaxLevelholds its messages at that level, or no deeper than it, only withadaptiveResolutionand thekv-stableorkv-unifiedsolver. Under the other solvers, and without adaptive resolution, the pin keeps them raw.listPinsandunpinmanage the pins.holdCompressionkeeps messages out of compression untilreleaseCompression, or until an optionaltimeoutMspasses, for content you'll still edit. Holds live in memory, so a reopened manager starts with none, and a hold that's never released stalls compression of everything after it.searchSummaries,getSummaryandgetSummariesInRangeread the summaries, for memory tools an agent can call.setToolDefinitionsandsetSystemPromptgive the summarizer's calls the agent's own tools and system prompt. A chunk whose messages hold tool calls or results waits to be compressed until tool definitions are set.getRenderStatsreports how much of the last compile was raw and how much was summaries, with counts for levels 1 to 3.
Every option is documented on AutobiographicalConfig in src/types/strategy.ts.
Branches
Chronicle keeps the store's history, so a conversation can branch. branchAt(messageId) creates a branch as of a message, and switchBranch(name) moves to it. fork() creates a branch at the current head and switches to it. The strategy's state, summaries included, follows the branch.
Inspecting and repairing stores
Among other tools, scripts/ holds ones for inspecting and repairing stores. For example, inspect-store.ts writes out a store's messages, summary tree and resolutions, audit-topology.ts checks that each summary covers a contiguous run of messages, and repair-topology.ts plans a repair for one that doesn't, which it applies with --apply. Opening a store takes its lock and can write to it, so run them on a stopped agent's store or on a copy. scripts/README.md shows how to compile a script and describes several of them.
Design notes
docs/ holds the design behind the strategies. These are good places to start:
adaptive-resolution-design.md: adaptive resolution and its solvers.kv-stable-context-control.md: thekv-stablesolver.unified-solve-design.md: thekv-unifiedsolver.compression-recall-curve-fallback-spec.md: what compression does when the model refuses a request.memory-tools-design.md: memory tools for agents.
Development
npm ci
npm test # builds, then runs the compiled tests in dist/
npm run typecheckCONTRIBUTING.md covers how changes land, what a pull request should contain, changelog fragments and releases. CHANGELOG.md lists what each release changed.
License
MIT, as package.json declares.
