Every tool call, log line, DB read, RAG chunk, and file your agent injects into a prompt is mostly boilerplate. Headroom strips the noise and keeps the signal โ losslessly, locally, and without touching accuracy.
100 logs. One FATAL error buried at position 67. Both runs found it. Baseline 10,144 tokens โ Headroom 1,260 tokens โ 87% fewer, identical answer.
python examples/needle_in_haystack_test.py
Works with Anthropic, OpenAI, Google, Bedrock, Vertex, Azure, OpenRouter, and 100+ models via LiteLLM.
Wrap your coding agent โ one command:
pip install "headroom-ai[all]"
headroom wrap claude # Claude Code
headroom wrap codex # Codex
headroom wrap cursor # Cursor
headroom wrap aider # Aider
headroom wrap copilot # GitHub Copilot CLIDrop it into your own code โ Python or TypeScript:
from headroom import compress
result = compress(messages, model="claude-sonnet-4-5")
response = client.messages.create(model="claude-sonnet-4-5", messages=result.messages)
print(f"Saved {result.tokens_saved} tokens ({result.compression_ratio:.0%})")import { compress } from 'headroom-ai';
const result = await compress(messages, { model: 'gpt-4o' });Or run it as a proxy โ zero code changes, any language:
headroom proxy --port 8787
ANTHROPIC_BASE_URL=http://localhost:8787 your-app
OPENAI_BASE_URL=http://localhost:8787/v1 your-app- Accuracy-preserving. GSM8K 0.870 โ 0.870 (ยฑ0.000). TruthfulQA +0.030. SQuAD v2 and BFCL both 97% accuracy after compression. Validated on public OSS benchmarks you can rerun yourself.
- Runs on your machine. No cloud API, no data egress. Compression latency is milliseconds โ faster end-to-end for Sonnet / Opus / GPT-4 class models than a hosted service round-trip.
- Kompress-base on HuggingFace. Our open-source text compressor, fine-tuned on real agentic traces โ tool outputs, logs, RAG chunks, code. Install with
pip install "headroom-ai[ml]". - Cross-agent memory and learning. Claude Code saves a fact, Codex reads it back.
headroom learnmines failed sessions and writes corrections straight toCLAUDE.md/AGENTS.md/GEMINI.mdโ reliability compounds over time. - Reversible (CCR). Compression is not deletion. The model can always call
headroom_retrieveto pull the original bytes. Nothing is thrown away.
Bundles the RTK binary for shell-output rewriting โ full attribution below.
Your agent / app
(Claude Code, Cursor, Codex, LangChain, Agno, Strands, your own codeโฆ)
โ prompts ยท tool outputs ยท logs ยท RAG results ยท files
โผ
โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ
โ Headroom (runs locally โ your data stays here) โ
โ โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ โ
โ CacheAligner โ ContentRouter โ CCR โ
โ โโ SmartCrusher (JSON) โ
โ โโ CodeCompressor (AST) โ
โ โโ Kompress-base (text, HF) โ
โ โ
โ Cross-agent memory ยท headroom learn ยท MCP โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ
โ compressed prompt + retrieval tool
โผ
LLM provider (Anthropic ยท OpenAI ยท Bedrock ยท โฆ)
โ Architecture ยท CCR reversible compression ยท Kompress-base model card
Savings on real agent workloads:
| Workload | Before | After | Savings |
|---|---|---|---|
| Code search (100 results) | 17,765 | 1,408 | 92% |
| SRE incident debugging | 65,694 | 5,118 | 92% |
| GitHub issue triage | 54,174 | 14,761 | 73% |
| Codebase exploration | 78,502 | 41,254 | 47% |
Accuracy preserved on standard benchmarks:
| Benchmark | Category | N | Baseline | Headroom | Delta |
|---|---|---|---|---|---|
| GSM8K | Math | 100 | 0.870 | 0.870 | ยฑ0.000 |
| TruthfulQA | Factual | 100 | 0.530 | 0.560 | +0.030 |
| SQuAD v2 | QA | 100 | โ | 97% | 19% compression |
| BFCL | Tools | 100 | โ | 97% | 32% compression |
Reproduce:
python -m headroom.evals suite --tier 1Community, live:
โ Full benchmarks & methodology
| Agent | One-command wrap | Notes |
|---|---|---|
| Claude Code | headroom wrap claude |
--memory for cross-agent memory, --code-graph for codebase intel |
| Codex | headroom wrap codex --memory |
Shares the same memory store as Claude |
| Cursor | headroom wrap cursor |
Prints Cursor config โ paste once, done |
| Aider | headroom wrap aider |
Starts proxy, launches Aider |
| Copilot CLI | headroom wrap copilot |
Starts proxy, launches Copilot |
| OpenClaw | headroom wrap openclaw |
Installs Headroom as ContextEngine plugin |
MCP-native too โ headroom mcp install exposes headroom_compress, headroom_retrieve, and headroom_stats to any MCP client.
Drop Headroom into any stack
| Your setup | Hook in with |
|---|---|
| Any Python app | compress(messages, model=โฆ) |
| Any TypeScript app | await compress(messages, { model }) |
| Anthropic / OpenAI SDK | withHeadroom(new Anthropic()) ยท withHeadroom(new OpenAI()) |
| Vercel AI SDK | wrapLanguageModel({ model, middleware: headroomMiddleware() }) |
| LiteLLM | litellm.callbacks = [HeadroomCallback()] |
| LangChain | HeadroomChatModel(your_llm) |
| Agno | HeadroomAgnoModel(your_model) |
| Strands | Strands guide |
| ASGI apps | app.add_middleware(CompressionMiddleware) |
| Multi-agent | SharedContext().put / .get |
| MCP clients | headroom mcp install |
What's inside
- SmartCrusher โ universal JSON: arrays of dicts, nested objects, mixed types.
- CodeCompressor โ AST-aware for Python, JS, Go, Rust, Java, C++.
- Kompress-base โ our HuggingFace model, trained on agentic traces.
- Image compression โ 40โ90% reduction via trained ML router.
- CacheAligner โ stabilizes prefixes so Anthropic/OpenAI KV caches actually hit.
- IntelligentContext โ score-based context fitting with learned importance.
- CCR โ reversible compression; LLM retrieves originals on demand.
- Cross-agent memory โ shared store, agent provenance, auto-dedup.
- SharedContext โ compressed context passing across multi-agent workflows.
headroom learnโ plugin-based failure mining for Claude, Codex, Gemini.
pip install "headroom-ai[all]" # Python, everything
npm install headroom-ai # TypeScript / Node
docker pull ghcr.io/chopratejas/headroom:latestGranular extras: [proxy], [mcp], [ml] (Kompress-base), [agno], [langchain], [evals]. Requires Python 3.10+.
โ Installation guide โ Docker tags, persistent service, PowerShell, devcontainers.
| Start here | Go deeper |
|---|---|
| Quickstart | Architecture |
| Proxy | How compression works |
| MCP tools | CCR โ reversible compression |
| Memory | Cache optimization |
| Failure learning | Benchmarks |
| Configuration | Limitations |
Headroom runs locally, covers every content type (not just CLI or text), works with every major framework, and is reversible.
| Scope | Deploy | Local | Reversible | |
|---|---|---|---|---|
| Headroom | All context โ tools, RAG, logs, files, history | Proxy ยท library ยท middleware ยท MCP | Yes | Yes |
| RTK | CLI command outputs | CLI wrapper | Yes | No |
| Compresr, Token Co. | Text sent to their API | Hosted API call | No | No |
| OpenAI Compaction | Conversation history | Provider-native | No | No |
Attribution. Headroom ships with the excellent RTK binary for shell-output rewriting โ
git showโgit show --short, noisylsโ scoped, chatty installers โ summarized. Huge thanks to the RTK team; their tool is a first-class part of our stack, and Headroom compresses everything downstream of it.
git clone https://github.com/chopratejas/headroom.git && cd headroom
pip install -e ".[dev]" && pytestDevcontainers in .devcontainer/ (default + memory-stack with Qdrant & Neo4j). See CONTRIBUTING.md.
- Live leaderboard โ 60B+ tokens saved and counting.
- Discord โ questions, feedback, war stories.
- Kompress-base on HuggingFace โ the model behind our text compression.
Apache 2.0 โ see LICENSE.


