Code knowledge graph over MCP: token-budgeted context for AI coding agents.
io.github.pmgarg/cgraphy MCP Server
The io.github.pmgarg/cgraphy Model Context Protocol (MCP) server provides a code knowledge graph intended for AI coding agents, using a token-budgeted context approach. It focuses on code analysis and knowledge-graph representations, aligning with MCP and tooling commonly used in agent workflows.
π οΈ Key Features
Code knowledge graph for MCP-based agent context
Token-budgeted context for AI coding agents
Emphasis on code analysis and knowledge-graph structure
π Use Cases
Supplying AI coding agents with structured code context
Supporting code-analysis workflows via a knowledge-graph model
Integrating MCP context into development tools
β‘ Developer Benefits
Reusable MCP server for token-limited code context
cgraphy is a Python code knowledge graph and Model Context Protocol (MCP) server for AI coding agents such as Claude Code, Codex CLI, Cursor, and Gemini CLI.
cgraphy indexes any codebase into a knowledge graph β functions, classes and
files as nodes; calls, imports, inheritance and git co-change history as
edges β and serves compact, token-budgeted slices of it to AI assistants
through the Model Context Protocol. Instead
of re-reading dozens of files to orient itself on every prompt, an agent asks
the graph and gets the relevant subgraph in a couple of thousand tokens.
Any language. Full-fidelity extraction (calls, imports, inheritance) for
Python, TypeScript/JavaScript, Java, Go, C, C++ and Rust; generic
definition-level extraction for 20+ more via tree-sitter; config and docs
files participate through summaries.
Importance-ranked. PageRank over the code graph puts load-bearing
symbols first in every answer.
Token-budgeted.cgraphy_context expands the graph greedily around a
symbol and stops exactly at your token budget β cost scales with the
question, not the repo.
Git-aware.--git-history mines commit history for files that change
together (logical coupling), an edge type static analysis can't see.
No API key. Semantic summaries are written by the host agent itself
through the enrich loop; summaries survive re-indexing via content hashing.
Zero infrastructure. One SQLite file in .cgraphy/. No services, no
daemons, no vector database.
Proven end-to-end. In 400+ controlled agent runs on SWE-bench Lite,
the deployed configuration resolved 14 vs 8 of 57 real GitHub issues
(official Docker harness). Indexes kubernetes (26K files, 219K nodes) in
49s, keeps it fresh in 1.6s cycles, answers queries in 1β152ms.
All benchmarks and predictions are in this repo (Research paper and benchmarks).
Install
bash
pip install cgraphy # or: uv tool install cgraphy
Quick start
bash
cd your-repo
cgraphy init # one command: MCP config + agent steering + index
cgraphy init does three things:
Writes a project-scoped .mcp.json β picked up automatically by Claude
Code in all its forms: CLI, VSCode extension, and the desktop app.
Appends a steering block to CLAUDE.md and AGENTS.md telling agents to
consult the graph (cgraphy_overview β cgraphy_search β
cgraphy_context) before reading files β this is what makes the graph
actually replace bulk file reading. (Agents can't be forced, only steered:
instruction files + persuasive tool descriptions + the tools being
genuinely faster is the mechanism, and it works.)
Builds the index with git co-change history.
Or register the MCP server manually with your assistant:
Claude Code
bash
claude mcp add cgraphy -- uvx cgraphy serve /path/to/repo
The working git diff mapped to touched symbols, their users, and covering tests
before committing / when resuming work
Enrichment:
Tool
Returns
The agent uses itβ¦
cgraphy_enrich
Batch of symbols that still need one-line summaries
when asked to "enrich the graph"
cgraphy_store_summaries
Confirmation + remaining count
to save the summaries it wrote
Retrieval is usage-aware: symbols an agent repeatedly asks about get a small,
capped boost in future context expansion (telemetry stays in the local
SQLite file; nothing leaves your machine).
Semantic search (optional)
bash
pip install "cgraphy[semantic]"
Adds tiny static embeddings (model2vec, CPU-only, no torch) fused with FTS5
by reciprocal-rank fusion β closes the vocabulary gap between issue-style
prose ("login broken") and code identifiers (validate_jwt).
The graph self-heals: tools detect stale files and re-index incrementally
(changed files only) before answering.
Enriching the graph
Structure is extracted automatically; meaning comes from summaries. Tell
your agent once:
enrich the cgraphy graph
It will loop cgraphy_enrich β cgraphy_store_summaries until every symbol
has a one-line semantic summary. Summaries are keyed to a hash of each
symbol's source, so editing one function invalidates only that summary.
For CI, cgraphy index --summarize pre-bakes summaries with your own
Anthropic API key (pip install cgraphy[summarize], ANTHROPIC_API_KEY set).
Viewer
bash
cgraphy view . # http://localhost:8787
A dependency-free local page (bundled Cytoscape.js): search, color by kind,
click for details, double-click to expand neighbors; co-change edges shown
dashed.
Mines fix-like commits from the repo's history (subject = query, touched
files = ground truth, co-change mining excludes evaluated commits), then
scores an ablation ladder β FTS-only, +PageRank, +graph expansion, Β±co-change
edges β on hit@5/hit@10/MRR and token cost. No LLM calls, no human grading,
fully reproducible. Results and a paper draft live in paper/.
How it works
cgraphy index walks the repo (respecting .gitignore +
.cgraphyignore), parses each file with tree-sitter, and stores nodes and
edges in .cgraphy/graph.db (SQLite + FTS5). Re-indexing is incremental by
content hash.
A resolver links cross-file references (calls, imports, inheritance) by
qualified name, best-effort; unresolved names are kept, never dropped.
PageRank runs over the edge graph; every query surfaces important symbols
first. Search blends FTS5 relevance with rank.
cgraphy serve exposes the five MCP tools over stdio.
Optional: --git-history adds weighted co-change edges mined from
git log.
If you use cgraphy in research or tooling, please cite the paper:
bibtex
@misc{garg2026cgraphy,
author = {Garg, Prateek Mohan},
title = {Which Graph Signals Pay for Their Tokens? cgraphy: A Token-Budgeted
Code Knowledge Graph as a Portable Context Layer for AI Coding Agents},
year = {2026},
doi = {10.5281/zenodo.21422935},
url = {https://doi.org/10.5281/zenodo.21422935},
note = {Preprint}
}