context-doctor 0.19.0 → 0.20.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/README.md CHANGED
@@ -1,15 +1,30 @@
1
1
  # context-doctor 🩺
2
2
 
3
- [![CI](https://github.com/KushalP1/context-doctor/actions/workflows/ci.yml/badge.svg)](https://github.com/KushalP1/context-doctor/actions) [![npm](https://img.shields.io/npm/v/context-doctor)](https://www.npmjs.com/package/context-doctor)
3
+ [![CI](https://github.com/KushalP1/context-doctor/actions/workflows/ci.yml/badge.svg)](https://github.com/KushalP1/context-doctor/actions) [![npm](https://img.shields.io/npm/v/context-doctor)](https://www.npmjs.com/package/context-doctor) [![npm downloads](https://img.shields.io/npm/dm/context-doctor)](https://www.npmjs.com/package/context-doctor) [![license: MIT](https://img.shields.io/badge/license-MIT-blue)](./LICENSE) ![macOS | Linux | Windows](https://img.shields.io/badge/platform-macOS%20%7C%20Linux%20%7C%20Windows-lightgrey)
4
4
 
5
- **See what's eating your LLM context window — and fix it.**
5
+ **Keep every AI session's context lean, automatically, without ever making it more expensive.**
6
6
 
7
- Every long-running LLM conversation slowly fills up with junk: duplicated documents, 10k-token tool outputs nobody reads again, base64 blobs, stale history. You pay for those tokens on **every single call**, and model quality drops as the window fills.
7
+ Long agent sessions fill up with tool output nobody reads again: file dumps, shell logs, search results, screenshots. You pay for all of it on every request, the model gets slower, and it drifts as the window fills. `context-doctor` measures that, and with **autopilot** it removes it from every Claude Code (and GPT API) request on your machine, only at moments when doing so costs nothing extra.
8
8
 
9
- `context-doctor` is a zero-config profiler + optimizer for LLM contexts. It works with **Claude, GPT, Gemini** message formats, and plugs into **Claude Desktop, ChatGPT (developer mode), Cursor, Claude Code** — any MCP-capable app — or runs standalone from the terminal.
9
+ - **9.8% less input cost, no session worse.** Replaying 130 days of the author's real Claude Code use (43 sessions) through the shipped code: **3.8 billion input tokens not sent, $4,694 saved at API list price, about $1,080 a month**, and not one session more expensive. Up to 37.7% on a long session. [What it saves →](#what-it-saves)
10
+ - **Counts Claude correctly.** Current Claude models pack 2.75 characters per token, not the 4 most tools assume; estimates built on 4 undercount Claude by about 40%. The ratios here were measured from the API's own counts, and `context-doctor accuracy` re-checks them on yours. [How →](#why-token-counts-are--and-where-they-are-exact)
11
+ - **Works where you work:** Claude Code, Cursor, Codex, Claude Desktop, any Anthropic or OpenAI API app, VS Code, CI. macOS, Linux and Windows, Node 20+.
12
+ - **Local and keyless.** No account, no telemetry, no API key. Your own login passes through untouched. MIT.
10
13
 
11
14
  Built and maintained by [gAI Ventures](https://gai.ventures).
12
15
 
16
+ ## Quick start
17
+
18
+ ```bash
19
+ npm install -g context-doctor
20
+ context-doctor install # hooks, MCP server and skill in every AI app it finds
21
+ context-doctor autopilot on # every new Claude Code session keeps its context lean
22
+ ```
23
+
24
+ Then start a new Claude Code session and work as usual. `context-doctor autopilot status` shows what it did; `context-doctor doctor` checks the whole setup. Everything is reversible: `context-doctor autopilot off`, `context-doctor uninstall`.
25
+
26
+ Just want a look first? `npx context-doctor session` profiles your latest Claude Code or Codex session in place, no install.
27
+
13
28
  ```
14
29
  Where the tokens go
15
30
  ────────────────────────────────────────────────────────
@@ -26,35 +41,66 @@ Findings (4)
26
41
  → Truncate or summarize large tool outputs before they enter history.
27
42
  ```
28
43
 
29
- ## Quick start (30 seconds)
44
+ ## What it saves
30
45
 
31
- **One command sets up everything** — detects Claude Desktop, Claude Code, and Cursor on your machine, wires in the MCP server, installs the Agent Skill, and registers the Claude Code every-prompt hook:
46
+ Measured, not modelled: every Claude Code session on the author's machine from 18 May to 25 September 2026 (43 sessions, 130 days, mostly Opus 5 and Fable 5 with the 1M window) was replayed request by request through the shipped autopilot code, with the real timestamps, and priced the way the prompt cache bills it (cached reads 0.1x, writes 1.25x). Run it on your own history with `node scripts/replay-autopilot.mjs`.
32
47
 
33
- ```bash
34
- npx context-doctor install
35
- ```
48
+ | | Without autopilot | With autopilot | Saved |
49
+ |---|---|---|---|
50
+ | Input tokens sent | 41.4 billion | 37.6 billion | **3.8 billion (9.2%)** |
51
+ | Input cost at API list price | $48,962 | $44,268 | **$4,694 (9.8%)** |
52
+ | Per 30 days | | | **~875 million tokens, ~$1,080** |
53
+ | Sessions made more expensive | | | **0 of 43** |
36
54
 
37
- `install` configures every app it detects and does not stop at the first problem: a corrupt Claude Desktop config still gets you Claude Code and Cursor. It does not pretend either. Any target that failed is named with a ✗ line, the summary reads "Done with N problem(s)" instead of "Done.", and the **exit code is 1**, so dotfiles and onboarding scripts can react. A broken config file is never overwritten; fix it and re-run.
55
+ How it spreads: the median session saves 1.9%, the best 37.7%. Short sessions barely change, because they rarely pile up 20k tokens of stale tool output before they end. Long sessions are where the money is: on this machine 94% of input cost came from requests above 200k tokens, and those are the requests autopilot makes smaller. Savings scale with how long your sessions run and how much they read, so a lighter user saves proportionally less, and never pays more.
38
56
 
39
- That single command is also all it takes to **set up context-doctor on anyone else's machine**. Prefer a global install, or want the unreleased `main`? Both work (Node 20+):
57
+ On a Claude subscription you do not pay list price; the same tokens come out of your usage limit instead. Anthropic does not publish how limits weight cached tokens, so read the dollar column as the size of the effect, not as your bill. The token column holds either way, and every request that is 9% smaller is also faster to first token and further from auto-compaction.
40
58
 
41
- ```bash
42
- npm install -g context-doctor && context-doctor install
43
- ```
59
+ What is not counted here: the proxy's full optimizer for your own API apps, the hook's guidance to the model, and fixes you make from `session` findings. Those save more on top, but they depend on what the model or you do with the advice, so they are not in this table.
60
+
61
+ ## What happens on each platform
62
+
63
+ | Where you work | Automatic, every request | What you get on top |
64
+ |---|---|---|
65
+ | **Claude Code** (terminal, VS Code, JetBrains, desktop app's Code tab) | **Autopilot** clears stale tool output (cold cache only, never more expensive). **Hook** on every prompt warns the model with the real context size and its largest waste | Status bar context meter, `/context-doctor` skill, `session`, `watch`, `report`, dashboard |
66
+ | **Cursor** (agent) | Cursor runs Claude Code's hooks, so the same every-prompt check fires inside Cursor | MCP tools, editor status bar extension, `cursor` profiler. With your own OpenAI key, autopilot too via a tokened tunnel ([how](#putting-the-proxy-on-a-public-url-cursor-with-your-own-openai-key-remote-apps)) |
67
+ | **Codex** (ChatGPT app's Codex tab, IDE extension, CLI) | Every-prompt hook with the API's own token counts | MCP tools, skill, `session` reads Codex rollouts. On an API key, autopilot too (`OPENAI_BASE_URL`) |
68
+ | **Your own apps on the Anthropic or OpenAI API** | Autopilot on `/v1/messages`, `/v1/chat/completions` and `/v1/responses` (`ANTHROPIC_BASE_URL` / `OPENAI_BASE_URL`), or the full optimizing proxy | Exact usage and cache hit rates in `/stats`, prompt-cache placement advice |
69
+ | **Claude Desktop chat** | Standing context rules in every chat; one cheap `profile_context` call the model makes past ~30 turns or on any cost question | One-click `.mcpb` install, `context_checkup` prompt |
70
+ | **claude.ai, ChatGPT, the phone apps** | Your account's standing preferences (`context-doctor instructions --copy`) | Profile an exported chat with `analyze` |
71
+ | **CI** | `analyze --fail-over-budget` fails a build whose prompts outgrow a budget | `.contextdoctorrc` budgets and presets |
72
+
73
+ Not claimed, because no process on your machine sends those requests: trimming inside Claude Desktop chat, claude.ai, ChatGPT, Cursor's own subscription models, or Codex signed in with ChatGPT. Those get the rules and the measurements above, not autopilot.
74
+
75
+ ## What's new
76
+
77
+ - **0.20 Autopilot**: stale tool output cleared from every Claude Code request, only when the prompt cache is cold, so it cannot cost more (measured: 9.8% less input cost, ~$1,080 a month on the author's usage, no session worse); runs as a login service on macOS, Linux and Windows; now also for GPT via OpenAI's Chat Completions and Responses APIs.
78
+ - **0.19 Measured Claude tokenizer**: estimates were 40% low for Claude; fixed from the API's own counts, with a per-model check in `accuracy`.
79
+ - **0.18** `proxy --token` for putting the proxy on a public URL safely. **0.17** Claude Desktop: a `profile_context` the model can afford to call from chat, `.mcpb` bundle, standing preferences for web and mobile. **0.16** Codex. **0.15** Cursor.
80
+
81
+ Full history with the measurements behind each change: [ROADMAP.md](./ROADMAP.md).
44
82
 
45
- Restart your apps, then just ask Claude: *"what's eating my context?"* (`npx context-doctor uninstall` reverses it.)
83
+ ## Setup details
84
+
85
+ `install` configures every app it detects and does not stop at the first problem: a corrupt Claude Desktop config still gets you Claude Code and Cursor. It does not pretend either. Any target that failed is named with a ✗ line, the summary reads "Done with N problem(s)" instead of "Done.", and the **exit code is 1**, so dotfiles and onboarding scripts can react. A broken config file is never overwritten; fix it and re-run.
86
+
87
+ `npx context-doctor install` works too, but autopilot needs the global install: a background service cannot point into npx's cache, which npm deletes at will.
46
88
 
47
89
  **No API keys, ever.** Everything is deterministic local code; when an LLM is needed (summarizing pruned history), the model already running in your app does it. The proxy forwards *your app's* credentials untouched — context-doctor itself holds nothing.
48
90
 
49
91
  ## What `install` actually does — and what happens in every session after
50
92
 
51
- One run of `npx context-doctor install` writes five things (each config edit makes a `.backup` first; `uninstall` reverses all of it):
93
+ One run of `context-doctor install` writes these (each config edit makes a `.backup` first; `uninstall` reverses all of it):
52
94
 
53
95
  1. **Claude Desktop config** (`claude_desktop_config.json`) — registers the MCP server
54
96
  2. **Claude Code config** (`~/.claude.json`) — registers the MCP server
55
97
  3. **Cursor config** (`~/.cursor/mcp.json`) — registers the MCP server
56
98
  4. **Agent Skill** → `~/.claude/skills/context-doctor/` — context-hygiene playbook for Claude Code
57
- 5. **Every-prompt hook** → `~/.claude/settings.json` — the per-query context check for Claude Code
99
+ 5. **Every-prompt hook** → `~/.claude/settings.json` — the per-query context check for Claude Code (Cursor runs it too)
100
+ 6. **Codex**, when present: MCP server in `~/.codex/config.toml`, the hook in `~/.codex/hooks.json`, the skill in `~/.codex/skills/`
101
+ 7. With `--statusline`: live context size, cache share and cost in Claude Code's status bar
102
+
103
+ `context-doctor autopilot on` is separate and opt-in: it adds the background proxy service and one line (`env.ANTHROPIC_BASE_URL`) to `~/.claude/settings.json`, after the proxy has answered a health check.
58
104
 
59
105
  **In every chat afterward (Claude Desktop, Cursor):** when the conversation starts, the app launches the MCP server, which hands the model standing instructions that stay in force for the whole chat:
60
106
 
@@ -72,11 +118,12 @@ One run of `npx context-doctor install` writes five things (each config edit mak
72
118
 
73
119
  ## Do you need MCP? Only sometimes — all the ways to use context-doctor
74
120
 
75
- MCP is just one of six delivery mechanisms. It's only required when you want the AI **inside a chat app** to run the tools itself. Everything else works without it:
121
+ MCP is just one of seven delivery mechanisms. It's only required when you want the AI **inside a chat app** to run the tools itself. Everything else works without it:
76
122
 
77
123
  | How you use it | MCP needed? | What it requires |
78
124
  |---|---|---|
79
125
  | **CLI** — `analyze`, `optimize`, `session` on files/transcripts | ❌ No | Nothing but `npx` — works in any terminal, scripts, CI |
126
+ | **Autopilot** — every Claude Code session, and GPT API apps | ❌ No | `context-doctor autopilot on` (a login service + one line in `~/.claude/settings.json`) |
80
127
  | **Proxy** — always-on optimization of your API apps | ❌ No | `context-doctor proxy` + one env var in your app |
81
128
  | **Claude Code every-prompt hook** | ❌ No | Written by `install`; Claude Code invokes it directly |
82
129
  | **Agent Skill** — hygiene behavior in Claude Code / claude.ai | ❌ No | A markdown file; `install` places it (or upload to claude.ai) |
@@ -90,6 +137,7 @@ Practical upshot: a developer who only wants cheaper, faster API calls never tou
90
137
  | Command | What it does |
91
138
  |---|---|
92
139
  | `context-doctor install` / `uninstall` | Wire (or remove) everything: MCP for Claude Desktop/Code/Cursor/Codex, the Agent Skill, the every-prompt hook |
140
+ | `context-doctor autopilot on\|off\|pause\|resume\|status` | Every new Claude Code session goes through the local proxy, which clears stale tool output only when the prompt cache is cold: measured 9.8% less input cost, no session worse |
93
141
  | `context-doctor instructions [--copy]` | The ~180-token standing rules (~120 on GPT) for claude.ai / ChatGPT preferences, for web and phones where no server runs |
94
142
  | `context-doctor analyze <file>` | Profile a conversation: token breakdown, findings, cost + latency estimates. `--fail-over-budget` exits 1 on a breach, for CI |
95
143
  | `context-doctor optimize <file>` | Apply the safe fixes; add `--strategy trim-tool-calls` for big inline file writes, `--strategy prune-history` for consented lossy compaction |
@@ -112,14 +160,13 @@ Practical upshot: a developer who only wants cheaper, faster API calls never tou
112
160
 
113
161
  | Where you run LLMs | Mechanism | Guarantee |
114
162
  |---|---|---|
115
- | Your own apps/agents (API) | `context-doctor proxy` rewrites every request in flight | **Every call, automatic** |
163
+ | Claude Code sessions, with autopilot on | The login-service proxy clears stale tool output from every request, only when the cache is cold | **Every request; never more expensive (measured)** |
164
+ | Your own apps/agents (API) | `context-doctor proxy` rewrites every request in flight, or `proxy --autopilot` for the cache-safe mode | **Every call, automatic** |
116
165
  | Claude Code / Cowork sessions | `install` registers a **UserPromptSubmit hook**: every query measures the session; heavy sessions get injected hygiene guidance (silent when lean, rate-limited, never blocks a prompt) | **Every query checked** |
117
166
  | Claude Desktop chat / Cursor | **MCP server instructions** — standing hygiene directives injected into every conversation where the server is enabled, plus prescriptive tool triggers | **Every conversation carries the rules** |
118
- | claude.ai (web) / ChatGPT app | Upload `skills/context-doctor/SKILL.md` in the app's Skills settings for the same standing behavior | Manual one-time upload |
119
-
120
- Nothing runs in the background for the Claude apps — the hook, skill, MCP server, and its instructions are all delivered by the app itself at the right moment. The proxy is the only long-running piece, and only your API-calling apps need it.
167
+ | claude.ai (web) / ChatGPT / phone apps | `context-doctor instructions --copy`, pasted once into the account's preferences (or upload `skills/context-doctor/SKILL.md` as a skill) | Every chat on that account carries the rules |
121
168
 
122
- Optional belt-and-braces for any chat app: add one line to your profile preferences — *"Practice context hygiene: summarize large content instead of re-quoting it, and use context-doctor's tools when conversations get heavy."*
169
+ Without autopilot nothing runs in the background: the hook, skill, MCP server and its instructions are delivered by the apps themselves at the right moment. With autopilot, one small proxy runs as a login service (launchd / systemd user service / logon task) and is restarted by the service or by the next prompt's hook if it ever stops.
123
170
 
124
171
  Or use the CLI directly, no install needed:
125
172
 
@@ -150,6 +197,35 @@ npx context-doctor session --list # browse sessions
150
197
 
151
198
  Parses the transcripts Claude Code writes locally and answers "where did my tokens go today?" — it will happily tell you that one giant skill load is 67% of your context.
152
199
 
200
+ ## Autopilot: every Claude Code session keeps its own context lean
201
+
202
+ ```bash
203
+ context-doctor autopilot on # once; survives reboots
204
+ context-doctor autopilot status # what it has done
205
+ context-doctor autopilot pause # instant passthrough, nothing restarts
206
+ context-doctor autopilot off # remove it
207
+ ```
208
+
209
+ `autopilot on` runs the local proxy as a background service (launchd on macOS, a systemd user service on Linux, a logon task on Windows), waits until it answers, and only then points Claude Code at it through `env.ANTHROPIC_BASE_URL` in `~/.claude/settings.json`. Every Claude Code session started afterwards, in the terminal, an IDE, or the desktop app's Code tab, sends its requests through it. Your login (subscription or API key) passes through untouched.
210
+
211
+ **What it does to each request:** once old tool output adds up to 20k+ tokens (file reads, shell output, search and web results, screenshots inside them), it replaces that output with a one-line note, keeping the 3 most recent results. The tool call stays in the history, so the model can simply run it again if it needs the output. Your messages, its answers, answers you gave to its questions, subagent reports and MCP results are never touched.
212
+
213
+ **Why it cannot make a session more expensive.** A prompt cache matches a byte-identical prefix, and changing old history re-bills everything after the change at the write rate (1.25x instead of 0.1x). So autopilot changes history only when the cache is cold anyway: after an idle gap longer than the cache lifetime the request itself declares (1 hour for Claude Code on a subscription, 5 minutes otherwise), when the whole prompt is re-written regardless. Once cleared, an output stays cleared on every later request, so the prefix is identical between clearings and the cache keeps hitting.
214
+
215
+ **Measured before it shipped**, by replaying every Claude Code session on the author's machine (43 sessions, 130 days) request by request through the shipped code, priced as the cache bills it, with real timestamps: 3.8 billion input tokens not sent and $4,694 at list price, ~$1,080 a month ([details](#what-it-saves)). `scripts/replay-autopilot.mjs` does this on yours:
216
+
217
+ | Policy | Input cost saved | Worst session |
218
+ |---|---|---|
219
+ | **Autopilot: clear only when the cache is cold (default)** | **9.8%** (9.2% of raw input tokens; best session 37.7%) | **0.00%, no session worse** |
220
+ | Also clear on a warm cache when the saving "should" repay the rewrite | 9.9% | −0.13% (one session worse) |
221
+ | The proxy's general strategies (dedupe, trim, strip-base64) | 6.7% | −13% (one session worse) |
222
+
223
+ 94% of input cost on that machine came from requests above 200k tokens, which is where the clearing lands. Smaller requests also mean later auto-compaction and a faster first token.
224
+
225
+ **Never in the way:** anything it cannot parse is forwarded unchanged; only Anthropic `/v1/messages` requests are touched. It adds about 7 ms to a 2.9 MB (~1M token) request. If the proxy dies, the service restarts it within seconds, and the every-prompt hook checks it before each prompt and starts it if needed (measured: 0.6 s once, when it had to). `pause` turns it into a passthrough without restarting anything; `off` removes the setting before stopping the service, and sessions started while it was on need a restart.
226
+
227
+ **What it cannot reach** (no process on your machine sends those requests): Claude Desktop's chat tab and claude.ai, which use the standing instructions and the `profile_context` sketch instead; Cursor's own models (Cursor's servers call the model; with your own OpenAI key, see the tunnel section below); and Codex signed in with ChatGPT, where the every-prompt hook still reports context size.
228
+
153
229
  ## Always-on: optimize every request automatically
154
230
 
155
231
  Run the proxy and every Anthropic/OpenAI API call your apps make gets optimized in flight — no code changes:
@@ -0,0 +1,128 @@
1
+ /**
2
+ * Autopilot: clear stale tool output from agent requests in flight, without
3
+ * ever costing more than it saves.
4
+ *
5
+ * Why this and not the proxy's general strategies: replayed over 33 real
6
+ * Claude Code sessions with prompt-cache pricing, the general trim saved 6.7%
7
+ * overall but made one session 13% MORE expensive, because every change to
8
+ * history the cache already holds re-bills everything after it at the write
9
+ * rate. Most of the bill sits in long sessions (94% of input cost came from
10
+ * requests above 200k tokens), so the win is real, but only if each change
11
+ * pays for its cache rewrite.
12
+ *
13
+ * The rules, modelled on the microcompact logic Claude Code ships (and keeps
14
+ * switched off by a server flag), tightened for "never worse":
15
+ *
16
+ * 1. Only output of read-only, re-runnable tools is eligible (file reads,
17
+ * shell, search, web fetch). Answers the user gave, subagent reports and
18
+ * MCP results are never touched.
19
+ * 2. The most recent `keepRecent` tool results always stay.
20
+ * 3. Clearing happens in batches of at least `minClearTokens`, so each cache
21
+ * rewrite buys a lot.
22
+ * 4. A cleared result stays cleared on every later request (ids are
23
+ * remembered, across restarts), so the prefix is byte-stable between
24
+ * batches and the cache keeps hitting.
25
+ * 5. By default a batch is taken ONLY when the cache is cold anyway (idle
26
+ * longer than the request's own cache TTL), because the whole prompt is
27
+ * re-written on that request regardless, so clearing costs nothing and
28
+ * every later request is smaller. That makes it never-worse by
29
+ * construction, and measured: replaying every Claude Code session on the
30
+ * author's machine through this class, with real timestamps and cache
31
+ * pricing, it saved 9.8% of cache-weighted input cost (9.2% of raw input
32
+ * tokens, up to 37.7% in one long session) and made no session worse.
33
+ * Clearing on a warm cache when the saving "should" repay the rewrite
34
+ * (`paybackSafety` > 0) added 0.1% and made one session 0.13% worse: the
35
+ * future is not knowable, so it is off unless asked for.
36
+ *
37
+ * The model sees a one-line note in place of the output and can simply re-run
38
+ * the tool; the file, command or URL is still in the tool call.
39
+ */
40
+ /** Tools whose output can be regenerated by calling them again (read-only or re-runnable). */
41
+ export declare const CLEARABLE_TOOLS: Set<string>;
42
+ export interface AutoClearOptions {
43
+ /** Most recent tool results that always stay. Default 3. */
44
+ keepRecent?: number;
45
+ /** Smallest batch worth a cache rewrite. Default 20,000 tokens. */
46
+ minClearTokens?: number;
47
+ /**
48
+ * Warm-cache rule: clear only if the saving over the requests still to come
49
+ * repays the rewrite this many times over. Requests still to come are
50
+ * estimated as the number already made in this conversation. Default 0:
51
+ * warm clearing off, cold cache only (the never-worse setting).
52
+ */
53
+ paybackSafety?: number;
54
+ /** Where cleared ids persist so a restart does not un-clear (and re-bill) them. */
55
+ statePath?: string;
56
+ }
57
+ export interface AutoClearResult {
58
+ changed: boolean;
59
+ /** Results cleared by this request's new batch (0 when none was taken). */
60
+ newlyCleared: number;
61
+ /** Tokens removed from this request, old batches included. */
62
+ tokensRemoved: number;
63
+ cold: boolean;
64
+ reason: string;
65
+ /** Index of the first message rewritten on this request (the cache breaks from here), or -1. */
66
+ firstChanged: number;
67
+ }
68
+ interface Found {
69
+ /** Index of the message / input item that holds the result (cache breaks from here). */
70
+ unit: number;
71
+ id: string;
72
+ tokens: number;
73
+ content: unknown;
74
+ set: (note: string) => void;
75
+ }
76
+ /**
77
+ * One request, whatever its API: the units that make up history, tool names
78
+ * by call id, the tool results in order, and the cache lifetime to assume.
79
+ * Anthropic Messages (Claude Code, Anthropic SDKs), OpenAI Chat Completions
80
+ * (role "tool" messages) and OpenAI Responses (function_call_output items,
81
+ * as Codex with an API key sends).
82
+ */
83
+ interface View {
84
+ format: "anthropic" | "openai-chat" | "openai-responses";
85
+ units: Array<Record<string, unknown>>;
86
+ toolName: Map<string, string>;
87
+ results: Array<Omit<Found, "tokens">>;
88
+ firstUser: unknown;
89
+ firstToolId: string;
90
+ ttlMs: number;
91
+ }
92
+ export declare function clearedNote(tokens: number): string;
93
+ /**
94
+ * How long the provider may keep this prompt cached, taking the LONGER
95
+ * possibility when unsure: calling a warm cache cold would pay a rewrite,
96
+ * calling a cold one warm only delays a free clearing.
97
+ * - Anthropic: the request's own cache_control, "1h" or the 5-minute default.
98
+ * - OpenAI: caches live up to an hour (in-memory retention), or 24 hours when
99
+ * the request sets prompt_cache_retention "24h".
100
+ */
101
+ export declare function requestTtlMs(body: Record<string, unknown>): number;
102
+ export declare function viewOf(body: Record<string, unknown>): View | undefined;
103
+ export declare class AutoClearer {
104
+ private cleared;
105
+ /** Tool results never change once sent, so their size is computed once per id. */
106
+ private sizes;
107
+ private convs;
108
+ private opts;
109
+ private dirty;
110
+ constructor(options?: AutoClearOptions);
111
+ /**
112
+ * Rewrite `body` in place: an Anthropic Messages, OpenAI Chat Completions or
113
+ * OpenAI Responses request. Never throws; anything unrecognised is left alone.
114
+ */
115
+ apply(body: Record<string, unknown>, now?: number): AutoClearResult;
116
+ /**
117
+ * Warm cache: clearing rewrites everything after the first cleared result
118
+ * at the write rate (1.25x instead of 0.1x: 1.15x extra, once), and saves
119
+ * the cleared tokens at the read rate (0.1x) on every later request.
120
+ * Later requests are estimated as the number made so far (a conversation's
121
+ * age is the best predictor of how long it keeps going), and the saving must
122
+ * cover the rewrite `paybackSafety` times over.
123
+ */
124
+ private shouldClearWarm;
125
+ private load;
126
+ private save;
127
+ }
128
+ export {};
@@ -0,0 +1,295 @@
1
+ /**
2
+ * Autopilot: clear stale tool output from agent requests in flight, without
3
+ * ever costing more than it saves.
4
+ *
5
+ * Why this and not the proxy's general strategies: replayed over 33 real
6
+ * Claude Code sessions with prompt-cache pricing, the general trim saved 6.7%
7
+ * overall but made one session 13% MORE expensive, because every change to
8
+ * history the cache already holds re-bills everything after it at the write
9
+ * rate. Most of the bill sits in long sessions (94% of input cost came from
10
+ * requests above 200k tokens), so the win is real, but only if each change
11
+ * pays for its cache rewrite.
12
+ *
13
+ * The rules, modelled on the microcompact logic Claude Code ships (and keeps
14
+ * switched off by a server flag), tightened for "never worse":
15
+ *
16
+ * 1. Only output of read-only, re-runnable tools is eligible (file reads,
17
+ * shell, search, web fetch). Answers the user gave, subagent reports and
18
+ * MCP results are never touched.
19
+ * 2. The most recent `keepRecent` tool results always stay.
20
+ * 3. Clearing happens in batches of at least `minClearTokens`, so each cache
21
+ * rewrite buys a lot.
22
+ * 4. A cleared result stays cleared on every later request (ids are
23
+ * remembered, across restarts), so the prefix is byte-stable between
24
+ * batches and the cache keeps hitting.
25
+ * 5. By default a batch is taken ONLY when the cache is cold anyway (idle
26
+ * longer than the request's own cache TTL), because the whole prompt is
27
+ * re-written on that request regardless, so clearing costs nothing and
28
+ * every later request is smaller. That makes it never-worse by
29
+ * construction, and measured: replaying every Claude Code session on the
30
+ * author's machine through this class, with real timestamps and cache
31
+ * pricing, it saved 9.8% of cache-weighted input cost (9.2% of raw input
32
+ * tokens, up to 37.7% in one long session) and made no session worse.
33
+ * Clearing on a warm cache when the saving "should" repay the rewrite
34
+ * (`paybackSafety` > 0) added 0.1% and made one session 0.13% worse: the
35
+ * future is not knowable, so it is off unless asked for.
36
+ *
37
+ * The model sees a one-line note in place of the output and can simply re-run
38
+ * the tool; the file, command or URL is still in the tool call.
39
+ */
40
+ import { createHash } from "node:crypto";
41
+ import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
42
+ import { dirname } from "node:path";
43
+ import { estimateTokens } from "./tokens.js";
44
+ /** Tools whose output can be regenerated by calling them again (read-only or re-runnable). */
45
+ export const CLEARABLE_TOOLS = new Set([
46
+ // Claude Code
47
+ "Read", "Bash", "BashOutput", "Grep", "Glob", "LS", "WebFetch", "WebSearch", "NotebookRead",
48
+ // Codex and other OpenAI-API agents
49
+ "shell", "exec_command", "local_shell", "read_file", "list_dir", "grep_search", "file_search", "web_search",
50
+ // Cursor's agent
51
+ "Shell", "ReadFile", "rg", "run_terminal_cmd", "codebase_search",
52
+ ]);
53
+ /** Tokens billed for one image inside a tool result (a full-size screenshot is ~1.6k on Claude). */
54
+ const IMAGE_TOKENS = 1600;
55
+ const HOUR = 3_600_000;
56
+ export function clearedNote(tokens) {
57
+ return `[context-doctor: old tool output cleared to keep context lean (~${tokens} tokens). Run the tool again if you need it.]`;
58
+ }
59
+ function resultTokens(content, model) {
60
+ if (typeof content === "string")
61
+ return estimateTokens(content, model);
62
+ if (!Array.isArray(content))
63
+ return estimateTokens(JSON.stringify(content ?? ""), model);
64
+ let n = 0;
65
+ for (const part of content) {
66
+ if (part?.type === "image" || part?.type === "image_url" || part?.type === "input_image")
67
+ n += IMAGE_TOKENS;
68
+ else if (typeof part?.text === "string")
69
+ n += estimateTokens(part.text, model);
70
+ else
71
+ n += estimateTokens(JSON.stringify(part ?? ""), model);
72
+ }
73
+ return n;
74
+ }
75
+ /**
76
+ * How long the provider may keep this prompt cached, taking the LONGER
77
+ * possibility when unsure: calling a warm cache cold would pay a rewrite,
78
+ * calling a cold one warm only delays a free clearing.
79
+ * - Anthropic: the request's own cache_control, "1h" or the 5-minute default.
80
+ * - OpenAI: caches live up to an hour (in-memory retention), or 24 hours when
81
+ * the request sets prompt_cache_retention "24h".
82
+ */
83
+ export function requestTtlMs(body) {
84
+ const openai = Array.isArray(body.input) || (Array.isArray(body.messages) && isOpenAIChat(body.messages));
85
+ if (openai)
86
+ return body.prompt_cache_retention === "24h" ? 24 * HOUR : HOUR;
87
+ const json = JSON.stringify(body.system ?? "") + JSON.stringify(body.tools ?? "") + JSON.stringify(body.messages?.slice(-3) ?? "");
88
+ return /"ttl"\s*:\s*"1h"/.test(json) ? HOUR : 300_000;
89
+ }
90
+ function isOpenAIChat(messages) {
91
+ return messages.some((m) => m?.role === "tool" || (m?.role === "assistant" && Array.isArray(m.tool_calls)) || m?.role === "system" || m?.role === "developer");
92
+ }
93
+ export function viewOf(body) {
94
+ const toolName = new Map();
95
+ const results = [];
96
+ let firstToolId = "";
97
+ if (Array.isArray(body.input)) {
98
+ const units = body.input;
99
+ units.forEach((it, i) => {
100
+ const t = it?.type;
101
+ if ((t === "function_call" || t === "custom_tool_call" || t === "local_shell_call") && typeof it.call_id === "string") {
102
+ toolName.set(it.call_id, t === "local_shell_call" ? "shell" : String(it.name ?? ""));
103
+ firstToolId ||= it.call_id;
104
+ }
105
+ else if ((t === "function_call_output" || t === "custom_tool_call_output" || t === "local_shell_call_output") && typeof it.call_id === "string") {
106
+ results.push({ unit: i, id: it.call_id, content: it.output, set: (note) => { units[i] = { ...it, output: note }; } });
107
+ }
108
+ });
109
+ const firstUser = units.find((u) => u?.role === "user")?.content;
110
+ return { format: "openai-responses", units, toolName, results, firstUser, firstToolId, ttlMs: requestTtlMs(body) };
111
+ }
112
+ const messages = body.messages;
113
+ if (!Array.isArray(messages) || messages.length === 0)
114
+ return undefined;
115
+ const firstUser = messages.find((m) => m?.role === "user")?.content;
116
+ if (isOpenAIChat(messages)) {
117
+ messages.forEach((m, i) => {
118
+ if (m?.role === "assistant" && Array.isArray(m.tool_calls)) {
119
+ for (const c of m.tool_calls) {
120
+ if (typeof c?.id !== "string")
121
+ continue;
122
+ toolName.set(c.id, String(c.function?.name ?? c.name ?? ""));
123
+ firstToolId ||= c.id;
124
+ }
125
+ }
126
+ else if (m?.role === "tool" && typeof m.tool_call_id === "string") {
127
+ results.push({ unit: i, id: m.tool_call_id, content: m.content, set: (note) => { messages[i] = { ...m, content: note }; } });
128
+ }
129
+ });
130
+ return { format: "openai-chat", units: messages, toolName, results, firstUser, firstToolId, ttlMs: requestTtlMs(body) };
131
+ }
132
+ messages.forEach((m, i) => {
133
+ if (!Array.isArray(m?.content))
134
+ return;
135
+ const blocks = m.content;
136
+ blocks.forEach((b, j) => {
137
+ if (m.role === "assistant" && b?.type === "tool_use" && typeof b.id === "string") {
138
+ toolName.set(b.id, String(b.name ?? ""));
139
+ firstToolId ||= b.id;
140
+ }
141
+ else if (m.role === "user" && b?.type === "tool_result" && typeof b.tool_use_id === "string") {
142
+ results.push({ unit: i, id: b.tool_use_id, content: b.content, set: (note) => { blocks[j] = { ...b, content: note }; } });
143
+ }
144
+ });
145
+ });
146
+ return { format: "anthropic", units: messages, toolName, results, firstUser, firstToolId, ttlMs: requestTtlMs(body) };
147
+ }
148
+ /**
149
+ * A conversation's identity: its first user message plus its first tool call
150
+ * id. The message alone can repeat across sessions (the same project context
151
+ * opens every session in a folder); tool call ids are unique per session.
152
+ * The model is left out on purpose: switching models mid-session keeps the
153
+ * conversation (and the cleared set) but not the cache, which the TTL check
154
+ * cannot see, so a model switch is treated as warm, the safe direction.
155
+ */
156
+ function conversationKey(view) {
157
+ return createHash("sha1").update(JSON.stringify(view.firstUser ?? "").slice(0, 4000)).update(view.firstToolId).digest("hex");
158
+ }
159
+ export class AutoClearer {
160
+ cleared = new Map();
161
+ /** Tool results never change once sent, so their size is computed once per id. */
162
+ sizes = new Map();
163
+ convs = new Map();
164
+ opts;
165
+ dirty = false;
166
+ constructor(options = {}) {
167
+ this.opts = {
168
+ keepRecent: options.keepRecent ?? 3,
169
+ minClearTokens: options.minClearTokens ?? 20_000,
170
+ paybackSafety: options.paybackSafety ?? 0,
171
+ statePath: options.statePath,
172
+ };
173
+ this.load();
174
+ }
175
+ /**
176
+ * Rewrite `body` in place: an Anthropic Messages, OpenAI Chat Completions or
177
+ * OpenAI Responses request. Never throws; anything unrecognised is left alone.
178
+ */
179
+ apply(body, now = Date.now()) {
180
+ const none = (reason, cold = false) => ({ changed: false, newlyCleared: 0, tokensRemoved: 0, cold, reason, firstChanged: -1 });
181
+ try {
182
+ const view = viewOf(body);
183
+ if (!view)
184
+ return none("no messages");
185
+ const model = typeof body.model === "string" ? body.model : undefined;
186
+ const key = conversationKey(view);
187
+ const conv = this.convs.get(key);
188
+ const cold = !conv || now - conv.last > view.ttlMs;
189
+ this.convs.set(key, { last: now, requests: (conv?.requests ?? 0) + 1 });
190
+ if (this.convs.size > 2000)
191
+ this.convs.delete(this.convs.keys().next().value);
192
+ const results = view.results.map((r) => {
193
+ let tokens = this.sizes.get(r.id);
194
+ if (tokens === undefined) {
195
+ tokens = resultTokens(r.content, model);
196
+ this.sizes.set(r.id, tokens);
197
+ if (this.sizes.size > 100_000)
198
+ this.sizes.delete(this.sizes.keys().next().value);
199
+ }
200
+ return { ...r, tokens };
201
+ });
202
+ const eligible = results.slice(0, Math.max(0, results.length - this.opts.keepRecent))
203
+ .filter((r) => CLEARABLE_TOOLS.has(view.toolName.get(r.id) ?? ""));
204
+ const pending = eligible.filter((r) => !this.cleared.has(r.id) && r.tokens > 200);
205
+ const pendingTokens = pending.reduce((n, r) => n + r.tokens, 0);
206
+ let reason = "nothing to clear";
207
+ let taken = [];
208
+ if (pendingTokens >= this.opts.minClearTokens) {
209
+ if (cold) {
210
+ taken = pending;
211
+ reason = "cache cold: cleared for free";
212
+ }
213
+ else if (this.opts.paybackSafety > 0 && this.shouldClearWarm(view.units, pending, pendingTokens, model, (conv?.requests ?? 0) + 1)) {
214
+ taken = pending;
215
+ reason = "cache warm: saving repays the rewrite";
216
+ }
217
+ else {
218
+ reason = "cache warm: waiting, rewrite would not pay back yet";
219
+ }
220
+ }
221
+ else if (pendingTokens > 0) {
222
+ reason = `waiting for a ${Math.round(this.opts.minClearTokens / 1000)}k batch`;
223
+ }
224
+ for (const r of taken)
225
+ this.cleared.set(r.id, r.tokens);
226
+ if (taken.length)
227
+ this.dirty = true;
228
+ // Apply every clearing ever decided for this conversation, so the
229
+ // prefix is identical to what was sent last time.
230
+ let removed = 0;
231
+ let changed = false;
232
+ let firstChanged = -1;
233
+ for (const r of results) {
234
+ if (!this.cleared.has(r.id))
235
+ continue;
236
+ const note = clearedNote(this.cleared.get(r.id));
237
+ if (r.content === note)
238
+ continue;
239
+ removed += Math.max(0, r.tokens - estimateTokens(note, model));
240
+ r.set(note);
241
+ changed = true;
242
+ if (firstChanged === -1 || r.unit < firstChanged)
243
+ firstChanged = r.unit;
244
+ }
245
+ this.save();
246
+ return { changed, newlyCleared: taken.length, tokensRemoved: removed, cold, reason, firstChanged };
247
+ }
248
+ catch {
249
+ return none("error: request forwarded unchanged");
250
+ }
251
+ }
252
+ /**
253
+ * Warm cache: clearing rewrites everything after the first cleared result
254
+ * at the write rate (1.25x instead of 0.1x: 1.15x extra, once), and saves
255
+ * the cleared tokens at the read rate (0.1x) on every later request.
256
+ * Later requests are estimated as the number made so far (a conversation's
257
+ * age is the best predictor of how long it keeps going), and the saving must
258
+ * cover the rewrite `paybackSafety` times over.
259
+ */
260
+ shouldClearWarm(messages, pending, pendingTokens, model, requestsSoFar) {
261
+ const first = Math.min(...pending.map((r) => r.unit));
262
+ let suffix = -pendingTokens;
263
+ for (let i = first; i < messages.length; i++)
264
+ suffix += estimateTokens(JSON.stringify(messages[i]), model);
265
+ const rewriteCost = 1.15 * Math.max(0, suffix);
266
+ const saving = 0.1 * pendingTokens * requestsSoFar;
267
+ return saving >= this.opts.paybackSafety * rewriteCost;
268
+ }
269
+ load() {
270
+ if (!this.opts.statePath || !existsSync(this.opts.statePath))
271
+ return;
272
+ try {
273
+ const data = JSON.parse(readFileSync(this.opts.statePath, "utf8"));
274
+ for (const [id, tokens] of data.cleared ?? [])
275
+ this.cleared.set(id, tokens);
276
+ }
277
+ catch {
278
+ /* a corrupt state file only means results may be re-cleared once */
279
+ }
280
+ }
281
+ save() {
282
+ if (!this.dirty || !this.opts.statePath)
283
+ return;
284
+ this.dirty = false;
285
+ try {
286
+ // Newest 50k ids: far more than any live session holds.
287
+ const entries = [...this.cleared.entries()].slice(-50_000);
288
+ mkdirSync(dirname(this.opts.statePath), { recursive: true });
289
+ writeFileSync(this.opts.statePath, JSON.stringify({ cleared: entries }));
290
+ }
291
+ catch {
292
+ /* persistence is best effort */
293
+ }
294
+ }
295
+ }