MCP server that retrieves the most relevant source code for a query, within a token budget
Project description
fittok
Retrieve only the relevant source code for a question — instead of the model reading whole files — so an LLM answers codebase questions on a small, focused slice of context. Less input = fewer tokens, lower cost, faster answers.
Works three ways from one install: an MCP server, a CLI, and a Python library — plus a Claude Code plugin that injects context automatically.
How it works
codebase ──▶ graphify ──▶ slurp ──▶ readable slice ──▶ LLM answers
(parse) (select) (trim to budget)
- graphify — parses the repo with tree-sitter into a knowledge graph of functions / classes / methods (Python, JS, JSX, TS, TSX, Java, Go, Rust).
- slurp — scores every node against the question with **semantic embeddings
- TF-IDF + PageRank**, then selects only the genuinely relevant nodes via a relevance cliff (no budget-padding with noise).
- readable output — returns the actual source code of those nodes, top-ranked in full and the supporting tail as signatures, trimmed to a budget. The model answers directly from it.
Note: an earlier design compressed the slice with LLMLingua, but that produced unreadable token-salad the model ignored (then re-read the files). fittok returns real, readable code instead. LLMLingua remains available only as the standalone
compress_contexttool.
Graphs and embeddings are cached on disk (~/.cache/fittok), keyed by content —
so after a code change only the changed functions re-embed.
Install & use
fittok runs through uv — one tool for everything
below, with no manual pip install. Install it once:
brew install uv # macOS
# or any OS: curl -LsSf https://astral.sh/uv/install.sh | sh
As an MCP server (recommended — Claude Code / Cursor / Windsurf)
Claude Code — one command:
claude mcp add fittok -- uvx fittok # add -s user to enable it in every project
Restart Claude Code, then run /mcp to confirm fittok shows connected.
Cursor / Windsurf / any MCP client — add to its MCP config:
{ "mcpServers": { "fittok": { "command": "uvx", "args": ["fittok"] } } }
Then ask codebase questions normally. To make fittok trigger without mentioning
it, add one line to your client's CLAUDE.md:
"For any codebase question, call fittok first and answer from its output."
As a CLI (no MCP, no install — uvx fetches + runs it)
uvx fittok index <repo> # optional one-time pre-warm
uvx fittok query <repo> "how does auth work" # prints the relevant code slice
As a library
uv add fittok # in a uv project (or: uv pip install fittok in a venv)
from fittok import optimize
result = optimize("/path/to/repo", "how does authentication work")
print(result["optimized_context"])
First query on a repo auto-indexes (~15s once, cached); after that it's instant.
Token savings — honest numbers
fittok cuts the input/exploration cost of a codebase question. On a real
Next.js/TS repo (~5k functions) it returns a ~1.5–3.5k-token slice instead of
the model reading 15–20k+ tokens of files — an ~80–90% reduction on input,
deterministic and reported in the tool's savings footer.
How to measure it honestly:
- ✅ Use the
savingsfooter (e.g.84% — 2,494 vs 15,631 tokens) or your API bill (total tokens — which counts the subagent crawls fittok avoids). - ⚠️ Do not judge by Claude Code's
/context"Messages" number — it excludes subagent tokens and is dominated by the model's own reasoning, which fittok doesn't touch. On thorough models the real saving (e.g. ~84k → ~27k total tokens, by avoiding an Explore subagent) is invisible there but clear on the bill.
Where it shines: broad / multi-file questions, large files, unfamiliar repos, and thorough models that would otherwise explore heavily. On a tiny question a capable model can answer from one small file, so the win is marginal there.
Configuration (env vars)
| Variable | Default | Purpose |
|---|---|---|
FITTOK_SHOW_SAVINGS |
false |
Append a 🪙 saved X% footer to answers |
FITTOK_EMBED_MODEL |
all-MiniLM-L6-v2 |
Embedding model |
FITTOK_DEVICE |
auto |
auto / cuda / mps / cpu |
FITTOK_CACHE_DIR |
~/.cache/fittok |
Cache location |
Requirements
Python ≥ 3.10. First run downloads a ~90 MB embedding model. Optional extras:
uv pip install "fittok[ui]" (graph visualizer), "fittok[gpu]" (torch/CUDA).
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 fittok-0.3.1.tar.gz.
File metadata
- Download URL: fittok-0.3.1.tar.gz
- Upload date:
- Size: 227.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
b2939e98314f2fd463dd7d3b1ebd77655a2a87e6792218791d9d24b664e19ce6
|
|
| MD5 |
244683268fcb43b533647640d3f22c65
|
|
| BLAKE2b-256 |
528a748710f5e2bf45a443d9019089b0366284286cbaa9ba3879e8225d12ff01
|
File details
Details for the file fittok-0.3.1-py3-none-any.whl.
File metadata
- Download URL: fittok-0.3.1-py3-none-any.whl
- Upload date:
- Size: 43.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
531e2084e0980eb95e6340153d480e9082641936448d0f10ff9a52a4d8dcdd4e
|
|
| MD5 |
1ae3922d62f1a02fdbca2c70825a8375
|
|
| BLAKE2b-256 |
0eed891295aabf0ce744192f43c3d0af2d41aaef8aeb91bb07397207e75ac813
|