Core context-budget metering and decision engine for LLM applications: normalized token accounting, effective-budget gauges, turns-to-exhaustion prediction, and compaction/handoff decisions.
Project description
tokenmaster
Core context-budget metering and decision engine for LLM applications.
tokenmaster answers four questions for any model, provider, and conversation:
- How much context has been used, and on what (messages, system prompt, tool schemas, reasoning tokens, cache reads)?
- How much usable budget remains, measured against calibrated effective capacity rather than the advertised window?
- How many turns remain until exhaustion at the current token velocity?
- Should this conversation be compacted or handed off now, and what would that decision cost?
Install
pip install tokenmaster
Quickstart
from tokenmaster import CostModelPolicy, Meter, TaskContext
meter = Meter.for_model("anthropic:claude-sonnet-4-6")
# after each model response, feed it the usage numbers
meter.record({
"input_tokens": 52_000,
"cache_read_tokens": 118_000,
"output_tokens": 1_800,
"reasoning_tokens": 3_200,
})
state = meter.state()
state.fill_effective # fraction of usable budget consumed
state.eta_turns # projected turns to exhaustion (needs 3 turns of data)
state.zone # green / caution / critical
state.provenance # where every number came from
# judgment, with the arithmetic attached
rec = meter.advise(TaskContext(expected_remaining_turns=12))
rec.action, rec.urgency # continue / compact / handoff
rec.rationale.comparison # the comparison that produced the verdict
# the cost model prices compact vs handoff vs continue, cache economics included
policy = CostModelPolicy.for_profile(meter.profile)
meter.advise(TaskContext(expected_remaining_turns=40), policy=policy)
What is in 0.1.0
- Normalized TurnUsage accounting, with the hidden consumers (reasoning tokens, cache reads and writes, system prompt and tool-schema overhead) as first-class categories and a provenance tag on every number.
- MeterState gauges: effective versus nominal budget, EWMA token velocity, turns-to-exhaustion with a conservative bound, zone classification.
- A bundled model registry (12 models with dated, cited pricing), alias and
dated-suffix resolution, user overrides, and
Meter.for_modelfor zero-configuration attachment. - A typed event stream (seven events with exact wire round-trips): the contract that visualizers such as ctxmaster, or your own, build on.
- Three advisor policies: a threshold baseline that reproduces current practice, a predictive policy that compares conservative ETA against the task horizon, and a cost model that prices continue, compact, and handoff, including the cache break-even horizon k* that compaction must clear before it saves money.
- A handoff fidelity protocol (probe question answering) that makes "was that continuation prompt any good" measurable, with every LLM touchpoint behind an adapter so the protocol runs fully offline.
- Conformance vectors under
spec/in the repository: the executable cross-language specification the JavaScript and Rust ports must match.
Not yet included (planned)
Provider adapters (Anthropic and OpenAI usage normalizers), tokenizer estimators, LLM-backed probe generators and judges, calibrated effective-capacity data (defaults equal the nominal window, and the provenance says so), async event delivery, and tiered long-context pricing in the registry. The npm and crates.io packages of the same name are reserved placeholders until the ports land.
Design
The core has zero hard dependencies and never touches the network. Every
quantity carries its provenance; every recommendation ships the arithmetic
that produced it; parameters that have not been measured yet are labeled
provisional. The full contract lives at docs/core-api.md in the
repository: https://github.com/jemsbhai/tokenmaster
The companion package ctxmaster provides the terminal gauge and other visual surfaces on top of the event stream.
License
MIT
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file tokenmaster-0.1.0.tar.gz.
File metadata
- Download URL: tokenmaster-0.1.0.tar.gz
- Upload date:
- Size: 30.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.2
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
f312a195e78da937fad6998aa8794e85703bbb219f6e6accbfe7604b6918ccab
|
|
| MD5 |
8ac3793374290edbf6289ab9a81a953f
|
|
| BLAKE2b-256 |
0a12239664f262aa6fbf936beeefb8acd12bbcacfd80f124ac22d678a8d26202
|
File details
Details for the file tokenmaster-0.1.0-py3-none-any.whl.
File metadata
- Download URL: tokenmaster-0.1.0-py3-none-any.whl
- Upload date:
- Size: 25.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.2
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
6a1630a909b1e5705c5688afaa45ae2323d03f190d3ad9da46b355290a100156
|
|
| MD5 |
d1453b320d3a547e93952f03ec80c2d2
|
|
| BLAKE2b-256 |
ca2478c5288d454bdcfa8cf2ef758041f624b2166714f5f86112afe23dc7e3ef
|