The full upstream README, mirrored here for reference. Install config, tool schemas, adoption signals, and an original overview live on the AI Architect MCP Codebase listing page.
Cross-platform codebase intelligence for Codex, Gemini CLI, Claude Code, Cursor, VS Code, Zed, and any stdio MCP host.
One read-only Rust server, host-specific installation packages, and the same evidence-graded graph answers everywhere.
What An Agent Can Ask · Getting Started · Pipeline · Tools · Architecture · Zetetic Standard
Companion projects:
Hypermnesia MCP — persistent memory that consolidates and reconsolidates across sessions
zetetic-team-subagents — 97 genius reasoning agents + 18 team specialists
AI Architect Spec — TypeScript PRD generator that consumes our graph intelligence
Every AI coding assistant hits the same wall: you ask it to change handle_tool_call, and it either hallucinates a function that was renamed last week, edits something in the wrong community of the codebase, or silently breaks a call chain three modules away. Agents operate on strings; codebases have structure. The gap is where bugs live.
ai-architect-mcp-codebase is a cross-platform Rust MCP server for Codex, Gemini CLI, Claude Code, Cursor, VS Code, Zed, and other stdio MCP hosts. It indexes any Rust, Python, TypeScript, Java, Kotlin, Swift, Objective-C, C, C++, or Go codebase into a LadybugDB property graph (Ruby is dispatched on the shallow path — node-kind rows, no deep extraction — for 11 languages in total), resolves imports and call chains across files, detects functional communities via Leiden-class community detection, traces available call-graph paths from detected entry points, builds a hybrid BM25 + sparse TF-IDF + RRF search index, and exposes all of it through 26 MCP tools.
It is the codebase intelligence layer that sits between a finding ("this bug exists") and a PRD ("here is the fix, here is what it affects, here is what it must never break"). It is read-only intelligence — it never writes code, opens PRs, or runs CI. It supplies source-linked structural evidence and analysis limitations for the next stage to inspect.
One pipeline stage = one MCP tool. 10 stages. 26 tools. 12,000+ lines of Rust. 1500+ tests. Compiler and Clippy checks are part of CI.
For inferred Rust receiver calls, pass lsp: true to analyze_codebase and
install rust-analyzer. The response's lsp_status.state distinguishes
disabled, completed, and failed; failures retain their error and analysis
continues on the available graph, which may contain partial LSP results.
lsp_resolve retains the pass's counts; resolve.phase = "static" identifies
the separate static-resolution receipt. Completion does not mean every call
was resolved.
Analysis persists its coverage report for query_graph(graph="missed") and
returns the same summary. Rust processes use explicit #[test] and
#[kani::proof] attributes, with separate test and proof entry kinds.
These are source declarations, not evidence of execution or successful proof.
(Rust testing attributes,
Kani proof attributes.)
Graphs created before entry metadata was stored require a full reindex:
analyze_codebase rebuilds them, and index_codebase automatically falls back
to a full index when its incremental compatibility check detects the old schema.
These tools establish different kinds of evidence. Stage 2's verified receipt
means schema checks, clarification completeness and caller acknowledgement passed;
its transcript digest binds the recorded bytes, not the truth of the finding.
gates_passed means no critical flag was emitted by the available security checks.
Inspect report.assessment_complete as well: it is false for an empty symbol list, skipped
checks, or changed symbols that could not be resolved. Review those items and warnings even when
gates_passed is true. The unresolved-import gate reports unresolved imports in
a changed symbol's file; a single graph snapshot cannot establish when they were
introduced.
A semantic-diff clean verdict requires a structural regression score below
the configured threshold and no positive unresolved-import delta. Any increase
in unresolved imports produces at least concerning, even below that threshold.
A clean result does not establish behavioral equivalence. Tests, compiler checks
or formal proofs must establish that separate property.
Impact and process results depend on the relationships the graph captured. Process traversal stops at depth 20; it is graph reachability, not an observed runtime trace or an exhaustive account of effects. Preserve coverage and resolution qualifiers, and confirm important absence claims against source even when the coverage report contains no flagged files.
rust-toolchain.toml, so rustup installs and selects it for you; the same compiler builds CI and the releasesThe repo ships a .mcp.json that Claude Code picks up automatically when you open the directory:
Or register globally (recommended agent setup — the core profile):
The server registers one of two tool sets, chosen once at startup:
| Profile | Tools | Who it's for |
|---|---|---|
core | 8 — health_check · analyze_codebase · search_codebase · get_context · get_symbol · get_impact · query_graph · detect_changes | Recommended for agents. The read-only code-intelligence surface: analyze once, then search, inspect symbols, and measure blast radius. |
full | all 26 | The ai-architect pipeline orchestrator — adds the internal finding → PRD stages (1/2/4/6/8/9) and the manual graph passes (index_codebase, resolve_graph, cluster_graph, lsp_resolve, get_processes, index_history). |
Select with the --profile flag or the AP_PROFILE environment variable (the flag wins):
The default stays full until the next major version — shrinking the default tool surface is a breaking change. New agent installations should opt into core: analyze_codebase already runs index + resolve + cluster in one call, so the 18 hidden tools are pipeline plumbing an agent never needs, and hiding them keeps the tool prompt small.
The server is a self-contained stdio binary — any MCP host can launch it. Install once:
One command detects your installed hosts and writes the right MCP config for each — never clobbering the rest of the file:
It configures the top six hosts it detects: Claude Code (~/.claude.json), Codex CLI (~/.codex/config.toml), Gemini CLI (~/.gemini/settings.json), Cursor (~/.cursor/mcp.json), VS Code (Code/User/mcp.json), and Zed (~/.config/zed/settings.json).
ai-architect entry is added or updated; every other server survives. A file it cannot safely parse is never overwritten — it prints the exact entry to paste by hand.settings.json allows comments, which strict JSON editing would destroy, so install refuses to edit it in place and prints the snippet + instructions instead (your comments stay byte-for-byte).toml_edit).--dry-run (print planned changes, write nothing), --only <host> / --skip <host> (filter; --only forces a host even if undetected), --with-hooks (also register the Grep/Glob PreToolUse hook, see below). Re-running is idempotent (a second run reports "no change").ai-architect-mcp-codebase uninstall removes exactly our entries (and the hook), leaving everything else intact.Binary → first query. Measured on this machine (2026-07): install completes in ~1.3 s (dominated by process/DB startup; the config write itself is sub-second); analyze_codebase on this repo's own src/ (114 files → 16.5k nodes, 16.3k edges — index + resolve + cluster) takes ~12 s wall; the first search_codebase returns instantly. So once the binary exists, install → analyze → first graph query is ~15 s — well under the 2-minute target. The one-time cargo build --release (~5 min, compiling the LadybugDB C++ core) is a separate, before-the-clock step.
ai-architect-mcp-codebase install --with-hooks registers a Claude Code PreToolUse hook (matcher Grep|Glob) that runs ai-architect-mcp-codebase hook-augment. Before a Grep/Glob in a project that has an ai-architect graph, it injects a one-line suggestion to consider search_codebase/query_graph first. Cardinal rule: it never blocks the tool call — no graph, an unparseable payload, or any error → it prints nothing and exits 0. Hook registration is opt-in (the --with-hooks flag), never default.
The CLI commands below assume ~/.cargo/bin is on your PATH. GUI hosts (Cursor, Windsurf, VS Code) may not inherit your shell PATH — in the JSON configs, replace ai-architect-mcp-codebase with the output of which ai-architect-mcp-codebase. Use the core profile (8 read-only tools) for agent hosts.
Gemini CLI
Or install as an extension (this repo ships a gemini-extension.json):
The extension also exposes three host-native workflows from skills/:
understand-codebase, impact-analysis, and validate-change-plan.
They use only the eight tools in the core profile and explicitly surface
index coverage gaps before accepting negative graph results.
Claude Code plugin (primary interface)
Fresh marketplace installs require GitHub CLI 2.68 or newer. The bootstrap
verifies the release's attached Sigstore bundle against the fixed
cdeust/ai-architect-mcp-codebase/.github/workflows/release.yml signer before
installing any executable; it never accepts a manifest-provided trust anchor.
The bundle avoids a Rekor transparency-log lookup, but gh can still need the
network to refresh Sigstore's TUF trust root on a cold cache.
This protects the official package and makes a minimal-diff fork that changes
only metadata fail closed; it cannot make arbitrary code from a hostile fork
trustworthy, because such a fork can also replace the bootstrap itself. Verify
that the marketplace slug is exactly cdeust/ai-architect-mcp-codebase.
bin/ensure-binary.sh pins the installed binary to a verified release digest
(see Security) — that pin rejects any binary it did not download
and verify itself, including one you legitimately rebuilt from source. Set
AI_ARCHITECT_SOURCE_CHECKOUT=1 to opt out of the pin for a local dev build.
The bootstrap accepts two shapes under this flag, both requiring the explicit
opt-in — it is never inferred from metadata:
$CLAUDE_PLUGIN_ROOT itself contains .git
(you registered a clone directly as the plugin root).target/release/ai-architect-mcp-codebase is a symlink whose fully
resolved target lies outside $CLAUDE_PLUGIN_ROOT and sits inside its own
.git-bearing checkout (e.g. a marketplace cache whose binary was replaced
with a symlink into a separate dev clone, so you can iterate without
reinstalling the plugin after every rebuild). Added in
#208 —
a plain .git-at-root check cannot see this shape, because a marketplace
cache has no .git of its own.What the flag skips, precisely: only the release-binary digest
verification (sha256sum against the cached/pinned digest) and, for a fresh
install, the download + Sigstore provenance check — for that one launch. It
does not skip the Cargo.toml / plugin.json presence checks (still
fatal if either file is missing), and for a plain source checkout it still
runs the freshness rebuild (cargo build --release when src/ is newer than
the binary). For the montage shape specifically, nothing rebuilds the
binary — the bootstrap trusts the already-built binary the symlink resolves
to, as-is.
Threat model. This is an explicit, user-set opt-in, never something
packaged metadata can trigger. An attacker who can already write to your
plugin cache — replacing the installed binary with a symlink to force this
path — can just as easily replace bin/ensure-binary.sh or
bin/launch-plugin.sh themselves, so the digest pin was never a defense
against that attacker; it defends the default path (flag unset) where the
bootstrap is the thing standing between a marketplace download and your
shell. The default path is unchanged by this hatch and remains a hard
fatal on any digest mismatch. Every accepted bypass is announced on
stderr even in quiet mode:
Diagnosing the failure mode without the flag. If a marketplace-cache
binary is replaced by a montage symlink and AI_ARCHITECT_SOURCE_CHECKOUT is
not set, the plugin dies silently from Claude Code's point of view — you only
see MCP error -32000: Connection closed. The real cause is on stderr, which
Claude Code does not surface for a failed MCP launch; run the launcher by
hand with CLAUDE_PLUGIN_ROOT set to the plugin cache directory to see it:
Operational gotcha: export AI_ARCHITECT_SOURCE_CHECKOUT=1 in
~/.zshrc alone is not enough. ~/.zshrc is read only by interactive
shells; the Claude Code plugin launcher and its hooks run in non-interactive
ones and never see it. Put the export in ~/.zshenv (or your shell's
equivalent non-interactive startup file) instead.
If the former Automatised Pipeline plugin is installed, remove it before installing the canonical package:
Claude MCP allowlists and permissions must also replace every prefix listed in
revoked_claude_tool_prefixes in the contract with
mcp__plugin_ai-architect-mcp-codebase_ai-architect__<tool>. The final
ai-architect segment is intentionally stable: it is the MCP server key, not
the plugin's distribution name. The machine-readable source of truth is
mcp-contract.json; consumer repositories validate their
allowlists against its derived claude_tool_prefix instead of maintaining an
independent spelling.
Contract schema 1 requires distribution, claude_plugin,
claude_marketplace, mcp_server, claude_tool_prefix, and
revoked_claude_tool_prefixes. Consumers must pin the raw contract URL to the
full commit SHA (tags can be moved), validate that the prefix equals
mcp__plugin_<claude_plugin>_<mcp_server>__, and remove revoked prefixes from
allowlists rather than retaining them as aliases. Consumer PRs record the full
producer commit in their contract URL; the v0.11.1 release must not be assumed
available until its verified-release workflow completes.
The same contract is included in the crate, MCPB, and signed release assets.
OpenAI Codex CLI (also picked up by the ChatGPT desktop app and Codex IDE extension — they share ~/.codex/config.toml)
Or in ~/.codex/config.toml:
Or install the packaged Codex plugin and its three matching skills from this repository's marketplace:
The Codex package lives under plugins/ai-architect-mcp-codebase/, with its own
.mcp.json fixed to --profile core. This isolation is intentional: the
root .mcp.json remains the existing Claude project configuration and keeps
the server's backward-compatible full default.
For Gemini CLI, uninstall the former extension identity before reinstalling from the renamed repository:
Cursor — .cursor/mcp.json (project) or ~/.cursor/mcp.json (global):
Windsurf — ~/.codeium/windsurf/mcp_config.json: same mcpServers block as Cursor.
VS Code — .vscode/mcp.json:
OpenAI Agents SDK (Python)
Every stage is a tool. Stages build on each other but are independently callable. The pipeline is serial in logical order but MCP calls are stateless — you can re-run stages 3a-3d on a fresh codebase without re-running stages 1-2.
| # | Tool(s) | What it does |
|---|---|---|
| 0 | health_check | Handshake + protocol + tool count |
| 1 | extract_finding, refine_finding | Deterministic finding extraction + orchestrator-aware prompt refinement |
| 2 | start_verification, append_clarification, finalize_verification, abort_verification | Human-gated clarification loop with SHA-256 transcript digest, atomic single-file session state |
| 3a | index_codebase, query_graph, get_symbol | tree-sitter AST → LadybugDB graph (16 node labels, 36+ relationship tables); user-configurable exclude_dirs and graceful skip of unreadable directories (issue #249) |
| 3b | resolve_graph, lsp_resolve | Import/call/impl resolution with confidence scoring + optional LSP deep resolution (rust-analyzer / pyright / typescript-language-server) |
| 3c | cluster_graph, get_processes, get_impact | Leiden-class community detection (Louvain + C2 repair) + BFS execution-flow tracing from entry points |
| 3d | search_codebase, get_context, analyze_codebase, detect_changes | Hybrid BM25 + sparse TF-IDF + RRF search · 360° symbol view · all-in-one analysis · git-diff impact |
| 4 | prepare_prd_input | Bundle verified finding + graph intel → artifact for ai-architect-mcp-spec |
| 6 | validate_prd_against_graph | Symbol hallucination · community consistency · process-impact contradiction |
| 8 | check_security_gates | Auth-critical community · unsafe symbol · public-API change · unresolved-import presence · test-coverage gap |
| 9 | verify_semantic_diff | Before/after graph diff with Tarjan SCC cycle detection and regression scoring |
Stages 5 (PRD generation), 7 (implementation), 10 (benchmark), 11 (deployment), 12 (PR) belong to other systems in the pipeline: ai-architect-mcp-spec, the coding agent, CI, and
gh. This project is the read-only intelligence half.
Every tool takes structured JSON arguments via the MCP protocol and returns a structured JSON response. No LLM is called from inside any tool — intelligence is the agent's job; the tool's job is safe, fast data movement with invariants.
Each tool has a JSON Schema enforced at the wire, reason codes on error (no cryptic protocol errors), and a receipt-style response with timing and counts.
Agent installs rarely need all 26 — the
coreprofile (see Tool profiles) registers just the 8 code-intelligence tools.
index_codebase can commit a compressed snapshot of the graph so teammates who
clone the repo never have to cold-index it.
index_codebase with "export_artifact": true writes, after a successful
index, a tar → zstd snapshot to <path>/.ai-architect-mcp-codebase/graph.zst
plus a graph.meta.json sidecar (schema version, git sha, tool version,
node/edge counts). It also appends a .gitattributes entry
(.ai-architect-mcp-codebase/graph.zst binary merge=ours) so the committed
binary never produces merge conflicts across branches. Commit both files.
A repo indexed before the project rename (issue #195) carries this snapshot
under the old .automatised-pipeline/ directory name; the first touch of
the artifact (export, bootstrap, or even a hook-augment Grep/Glob check)
migrates it to the current name in place — a one-shot rename, not a
permanent dual-path read.
index_codebase with "bootstrap": true — when there is no local graph at
<output_dir>/graph but a committed artifact is present — decompresses the
snapshot instead of cold-indexing. Staleness is checked first by comparing
the artifact's git sha with the repo's current HEAD:
source='artifact_bootstrap',
graph_state='fresh';source='artifact_bootstrap_fill',
graph_state='filled_to_working_tree', carrying fill_method and
{changed, added, deleted, renamed, unchanged} counts;"accept_stale": true → import the stale snapshot anyway and skip the
fill, and the response carries a stale_artifact
{artifact_sha, head_sha, commits_behind} report so a stale graph is
never mistaken for a fresh one.A fill that fails (no git diff and no bundled manifest) falls back to a
full index, as does an import failure — both explicit (logged to stderr,
never a silent partial graph) and reported via a bootstrap_skipped note.
Both index_codebase and analyze_codebase accept "exclude_dirs"
(default []) — directory names or paths to prune from the walk in
addition to the built-in build/dependency skip list (node_modules,
.venv, vendor, target, …). This is for directories that must never be
read (a secrets folder, a credentials mount), not a performance prune:
"secrets") is a bare name,
matched anywhere in the tree — like the built-in list."config/secrets") is a path
relative to path, matched as exactly one subtree. No glob support.dependency_scope tier, including full —
it is checked before, and independently of, dependency-directory descent.skipped with reason user_excluded, and the
response's coverage.skipped.user_excluded_count carries the exact count.exclude_dirs on an existing graph requires "full": true — like
dependency_scope, the incremental-index manifest does not capture it.Independent of exclude_dirs, a directory the OS refuses to read
(EACCES/PermissionDenied) no longer aborts the whole index: it is
recorded in the coverage sidecar with reason unreadable and the walk
continues past it, so one locked-down subdirectory can no longer discard an
otherwise-successful index.
All three flags default to false, so existing behavior and the core/core8
profiles are unchanged. The artifact is entirely optional: without it,
index_codebase cold-indexes exactly as before.
Post-import incremental fill (re-index only the
artifact_commit..HEADdiff instead of a full re-index) is tracked in #62 — it needs a changed-files-only indexer, which AP does not yet have.
Rust MCP server, hand-rolled stdio JSON-RPC 2.0 (no SDK — we own the wire). Clean Architecture with module boundaries.
Sixteen crates. Nothing speculative; everything justified.
| Crate | Purpose | License | Why |
|---|---|---|---|
serde + serde_json | Wire serialization | MIT | JSON-RPC, artifact persistence |
sha2 | Stage-2 transcript digest | MIT | Tamper detection |
lbug (LadybugDB) | Embedded property graph + Cypher | MIT | Native Cypher, FTS-ready, the Kùzu successor |
tree-sitter | Incremental parser runtime | MIT | First-class Rust bindings |
tree-sitter-rust · -python · -typescript · -java · -kotlin-ng · -swift · -objc · -c · -cpp · -go | Language grammars (10) | MIT / Apache-2.0 | Semantic structure without a compiler |
tantivy | Lucene-grade BM25 | MIT | Real ranked text search, <10ms startup |
Deliberately not included: async runtime (we're stdio-blocking), HTTP client, LLM SDK, embedding model runtime (sparse TF-IDF replaces it at zero dep cost).
Graphs are per-finding by design (Lamport's isolation invariant): each finding gets its own LadybugDB instance at <output_dir>/runs/<run_id>/findings/<finding_id>/graph/. Zero-coordination concurrency, trivial cleanup, no cross-finding state leakage. Redundant indexing for shared codebases is acknowledged and mitigated in a later optional cache layer — not shoehorned into the core.
max_db_sizeEvery LadybugDB Database this crate opens reserves virtual address space up front, sized by max_db_size. lbug's own default (SystemConfig::default()) is 1 << 43 = 8 TiB per instance; with graph_cache's MAX_CACHED_GRAPHS = 8 entries live in the read-path cache at once, that is a 64 TiB worst case (issue #25). src/graph_store.rs::system_config() is the single choke point every GraphStore::open_or_create call resolves through, in this precedence order:
AP_LBUG_TEST_MAX_DB_SIZE — test-only, set for every cargo test process via .cargo/config.toml's [env] table (512 MiB / 2^29, issue #21). Always wins when present, so cargo test behavior is independent of the production knob below.AP_LBUG_MAX_DB_SIZE — production override, unset by default. Bytes, must be a power of two and at least 8 MiB (lbug's own BufferManager::verifySizeParams floor). An invalid value is rejected with an actionable error at GraphStore::open_or_create time — never a silent fallback.1 << 43 bytes) when neither var is set — lbug's own DEFAULT_VM_REGION_MAX_SIZE, the engine's per-database VM-region ceiling on every 64-bit desktop/server platform (lbug-0.19.1/lbug-src/src/include/common/constants.h). This is an address-space reservation, not an allocation: disk and memory grow only with the data actually written. An earlier release capped the default at 8 GiB (issue #25, sized from the measurement table below); that cap aborted any ingestion whose graph outgrew it and was repealed on 2026-08-14 — an index must complete regardless of corpus size, multi-TiB included.Set AP_LBUG_MAX_DB_SIZE to bound the reservation in address-space-constrained environments (e.g. containers with a low RLIMIT_AS); the historical measurement table below documents typical graph sizes.
Measured graph sizes (2026-07-15, du -k on every graph file found under ~/.cache/cortex/code-graphs/*/graph, ~/.cortex/ap_graph/graph, and **/.prd-gen/graphs/*/graph), top 10 of 75:
| Graph | Size |
|---|---|
repro-cortex-viz-deps (cortex-viz + node_modules) | 473 MiB |
bench-c2-viz-deps (cortex-viz + deps) | 472 MiB |
bench-c3-viz-pubapi (cortex-viz, public API surface) | 460 MiB |
wt-windows-launcher-96-97-* (Cortex worktree) | 147 MiB |
wt-homeostatic-* (Cortex worktree) | 144 MiB |
wt-tools-drift-* (Cortex worktree) | 143 MiB |
Cortex-wt-wiki-titles-* | 142 MiB |
wt-findings-provenance-* | 126 MiB |
anthropic-partnership-Cortex | 126 MiB |
wt-ingest-provenance-* | 124 MiB |
Total across all 75 measured graphs: ~4.87 GiB. Every graph other than the top 3 (which include node_modules) is under 150 MiB — the node_modules-inclusive runs are the actual worst case driving the sizing rule above.
Inherited from zetetic-team-subagents. Not a prompt suggestion — an enforcement rule that holds in code.
| Pillar | Question |
|---|---|
| Logical | Is it consistent? |
| Critical | Is it true? |
| Rational | Is it useful? |
| Essential | Is it necessary? |
In this codebase it concretely means:
RRF_K = 60 cites Cormack 2009. BULK_BATCH_SIZE = 500 cites Kùzu/LadybugDB tuning. PARSE_TIMEOUT_MICROS = 5_000_000 is justified in the block above it.unsafe finding_id (spec §5.1.4, §9.3 Q4): must match [A-Za-z0-9._-]+ — callers see which rule they violated.lsp_resolve on a stub binary returns lsp_probe_failed: found on PATH but didn't respond as an LSP server (stdout closed immediately; likely a stub, proxy, or non-LSP binary) — not a cryptic protocol error.Four CRITICAL, four HIGH, three MEDIUM findings were surfaced by a security-auditor agent pass and fixed in commit 512d683:
insert_edge → centralized cypher_str() escaping (\ first, then ')validate_git_ref rejects --, newlines, NUL; -- separator before refslsp_command → strict allowlist (rust-analyzer, pyright, pyright-langserver, typescript-language-server)fs::symlink_metadata + MAX_DEPTHMAX_FILES=100_000, MAX_FILE_BYTES=10 MB, MAX_TOTAL_BYTES=2 GB, MAX_DEPTH=64set_timeout_micros(5_000_000) + MAX_PARSE_BYTES=1 MBquery_graph read-only → two layers over disjoint statement families (see below)graph_path filesystem safety → validate_graph_path_safe() before any remove_dir_allrootUri → RFC 3986 percent-encodingDIFF_LINE_MAX = u64::MAX / 2 guardEach fix has a test that asserts the exploit is now rejected. Run cargo test to see 1500+ tests pass including the exploit-regression suite.
query_graph is kept read-onlyTwo layers, covering disjoint statement families. Neither subsumes the other.
| Layer | Refuses | Mechanism |
|---|---|---|
Engine (GraphStore::execute_read_only_query) | every database write and DDL — CREATE, MERGE, SET, DELETE/DETACH DELETE, DROP, ALTER, however spelled | PreparedStatement::is_read_only(): the verdict comes from the compiled plan, so a mutation written in syntax no keyword scan enumerates is still refused |
Lexical (FORBIDDEN_CYPHER_KEYWORDS) | filesystem and database movement — COPY … TO, EXPORT/IMPORT DATABASE, ATTACH, DETACH, USE, LOAD FROM | whole-word, case-insensitive scan of the query with string literals, backticked identifiers and comments masked out first |
Lexical (READ_ONLY_PROCEDURES) | every CALL naming anything but TABLE_INFO / SHOW_TABLES — including the CALL <setting> = <value> configuration form | per-PROCEDURE classification of the identifier after each CALL token |
The lexical layer is load-bearing, not defence in depth. lbug's
StatementReadWriteAnalyzer overrides visitCopyFrom but leaves six
statements at the base visitor's no-op — visitCopyTo, visitExportDatabase,
visitImportDatabase, visitAttachDatabase, visitDetachDatabase and
visitUseDatabase (parsed_statement_visitor.h:51, 57-61 on lbug 0.19.1) — so all
six are classified read-only. DETACH/USE were added to the denylist on
2026-08-25 after a mechanical re-audit against those headers; before it, both
passed the lexical filter AND the engine filter. Measured 2026-08-24 against
lbug 0.19.1 on both available engine gates — is_read_only() and a database
opened with SystemConfig::read_only(true), which reaches the same predicate via
ClientContext::validateTransaction — COPY (…) TO 'f.csv' and
EXPORT DATABASE 'd' execute and write the filesystem, while both correctly refuse
CREATE NODE TABLE. Pinned by engine_gate_does_not_cover_filesystem_writes and,
for the whole family, engine_classifies_every_filesystem_statement_as_read_only.
CALL is classified per procedure rather than refused wholesale, so schema
introspection (CALL table_info('Function') RETURN *) is reachable. That
distinction is load-bearing too: the same analyzer returns readOnly = true from
visitStandaloneCall, so CALL threads = 8 — a configuration write — is
engine-read-only, and this lexical layer is the only barrier that exists against
it. Relaxing the KEYWORD rather than allowlisting the PROCEDURE would remove that
barrier entirely.
A keyword introduced by : or . is an identifier, not a clause, so queries over
this schema's own Import node table work unchanged:
MATCH (f:File)-[:Defines_File_Import]->(n:Import) WHERE n.is_resolved = false RETURN n.path.
The gate does not extend that exemption to an alias (AS <keyword>), where the
clause detectors do. The asymmetry is deliberate: on the gate an exemption can only
ever let a keyword through, so it fails closed and a bare use/create pattern
variable is refused (backtick it); on the clause detectors the expensive direction
is reversed, because reading AS limit as a clause would suppress the LIMIT
injection. A masked literal or backticked identifier is treated as a TOKEN, never
as whitespace, so no look-back can walk across one.
query_graph executes one statement per call. A trailing ; is accepted; a
;-chained request is refused with reason multi_statement_not_supported, because
the read-only classification, LIMIT injection, ORDER BY detection and the offset
cursor are all properties of a single statement.
The full security argument — threat model, trust boundaries, what each claim rests on, and where it stops — is in docs/ASSURANCE-CASE.md. Reporting process and response SLA: SECURITY.md. How the project is run and what happens if the maintainer stops: GOVERNANCE.md. Where it is going: docs/ROADMAP.md. OpenSSF Best Practices answers, criterion by criterion: .bestpractices.json.
Re-measured 2026-07-28 on the current dependency (lbug 0.18, rustc 1.95.0,
macOS 26.5.1 arm64) by re-running the dba agent's nine compile-and-run probes
— cargo test --release --test lbug_bulk_investigation -- --nocapture, 199
edges per strategy. The ranking is the same one the original 0.15.3 run found;
the absolute figures are not comparable across the two runs, because both the
engine version and the machine changed.
| Strategy | ms/edge |
|---|---|
| Raw string per edge (naive) | 9.658 |
| Prepared statement, no transaction | 6.924 |
BEGIN TRANSACTION + prepared + COMMIT | 0.328 |
UNWIND + typed LogicalType::Struct | 0.127 |
The chosen path is 76× faster than the naive one on this measurement.
The bulk-insert path uses UNWIND with a typed struct schema (the engineer who wrote the first version used LogicalType::Any which fails the binder — the typed struct form works). Prepared statements are cached in a RefCell<HashMap<query, PreparedStatement>> on the GraphStore. Sparse TF-IDF replaces the dense N × V × 4B matrix — 30.5× smaller on our own codebase (108 KB vs 3.2 MB) and scales linearly with non-zero terms rather than vocab size. Clustering eliminated probe_node_label_for_process (per-node Cypher round-trip) in favor of a single in-memory HashMap<id, label> population pass.
500-file synthetic Rust fixture indexes in ~38 seconds end-to-end (parse + resolve + cluster + search index), down from the pre-audit implied "5 min – 1 hour" bracket.
This offline retrieval evaluation compares graph queries with a fixed
substring-search/full-file-read protocol on an authored 4-language corpus
(Python, TypeScript, Go, Rust): 20 questions across five capability dimensions.
benchmarks/eval_headtohead/PRE_REGISTRATION.md records the hypotheses and
protocol. The current results below are the post-#92 run in
benchmarks/eval_headtohead/results.json; earlier runs remain separately saved.
See that folder's MANIFEST.md for provenance and reproduce.sh for the command.
The deterministic evaluation needs no API key or external corpus; building it
requires the Rust toolchain and dependencies to be available.
| metric (mean ± sample stdev, n=20) | AP graph tools | Grep/Glob/Read baseline | source field |
|---|---|---|---|
| retrieval precision | 1.00 ± 0.00 | 0.65 ± 0.33 | aggregate.{graph,explorer}.precision |
| retrieval recall | 1.00 ± 0.00 | 1.00 ± 0.00 | aggregate.*.recall |
| payload token proxy | 43.14 ± 17.26 | 550.36 ± 330.28 | aggregate.*.tokens |
| modeled tool calls | 1.00 ± 0.00 | 5.20 ± 1.64 | aggregate.*.tool_calls |
| mean per-question token ratio (baseline / graph) | 14.26× | — | aggregate.token_ratio_explorer_over_graph.mean |
| mean per-question tool-call ratio | 5.20× | — | aggregate.toolcall_ratio_explorer_over_graph.mean |
All four hypotheses H1–H4 are SUPPORTED in the current run under this
protocol. The original run FALSIFIED H4: graph recall was 0.825 against 1.00.
Its five losses (go-D3, go-D4, rs-D2, rs-D4, ts-D4) remain in
raw_results.2026-07-26-pre-fix-87.json; fixes #87 and #92 closed these gaps.
The corpus informed those fixes, so the current result is a regression benchmark,
not an unseen generalization test.
Costs are modeled, not observed AI-client bills or tool traces. The graph leg
serializes a benchmark-specific compact envelope of symbol identities; the
baseline counts its substring-hit transcript plus full matching files. Both use
a payload-size / 4 token proxy. Graph calls are assigned one per question;
baseline calls are assigned two plus the number of matching files. Indexing,
client prompts, actual MCP response envelopes and model reasoning are excluded.
The 14.26× figure is a mean of per-question ratios; dividing aggregate payload
volumes gives 12.76×, a different statistic. The optional answer-quality judge
(AP_EVAL_JUDGE_CMD) did not run. These measurements establish file-retrieval
results and protocol costs, not AI-agent success, hallucination reduction or
real-world token savings.
stages/*.md traces to an agent dispatch.stage-4.prd_input.json artifact via disk or MCP-to-MCP query of search_codebase / get_context / get_impact. Each in its ideal language: our performance-critical graph work in Rust, their document generation in TypeScript.Every stage has an integration test with fixture data. The lbug_bulk_investigation test is intentionally preserved — it's the compile-and-run proof that dba's UNWIND pattern works, kept for regression protection and documentation.
Every major architectural decision was made by a genius agent with a specific problem shape. Stored in stages/decisions/*.md and in Cortex.
| Decision | Agent | Verdict |
|---|---|---|
| Rust vs C/C++ for the glue layer | Popper | Conjecture "Rust is the right language" is unfalsified. lbug + tree-sitter already run native C/C++; Rust is the glue where the borrow checker pays the most. |
| Graph-per-finding vs graph-per-codebase | Lamport | Per-finding. Isolation holds by construction with zero coordination; the redundant-indexing cost is mitigable in an optional cache layer later. |
| Stage 3a decomposition | Simon | Five steps, satisficed against the growth rule; first useful query at step 4. |
| DB backend choice | dba | LadybugDB (evaluated at lbug 0.15.3, now on 0.18) — only option simultaneously maintained, native Cypher, embedded, with FTS + vector + algo extensions. |
| Stage 2 clarification loop shape | Shannon | Four-tool state machine with atomic single-file session (no crash window between separate files), unconditional one-round-minimum before finalize. |
| lbug UNWIND pattern | dba | LogicalType::Struct { fields } works; LogicalType::Any fails the binder — 38× speedup verified by compile-and-run probes. |
Agents are spawned via zetetic-team-subagents; each genius is a reasoning pattern (not a persona) with canonical moves and primary-source citations.
Public repo, MIT licensed. Security audit fixes are in, correctness fixes are in, scale fixes are in, stages 4/6/8/9 are live, but every capability marked "live" above has been verified end-to-end on this machine, not yet in a production context.
What works today: indexing Rust, Python, TypeScript, Java, Kotlin, Swift, Objective-C, C, C++, and Go codebases end-to-end, resolving cross-file relationships, clustering into communities, tracing processes from entry points, hybrid search, PRD input preparation, PRD claim validation, security gate checking, before/after regression detection.
What's deferred:
is_unsafe extraction in the Rust parser (stage 8 S2 runs in info-skip mode pending this)group_*)Published on crates.io as ai-architect-mcp-codebase and listed in the MCP Registry under the name below (this line doubles as the registry's package-ownership proof):
mcp-name: io.github.cdeust/ai-architect-mcp-codebase
MIT — see LICENSE.
This software is the independent work of Clément Deust. It was developed outside any employment relationship and is not affiliated with, endorsed by, or owned by any past or present employer. It is part of the ai-architect ecosystem (Cortex, zetetic-team-subagents, AI Architect Spec).
The graph-theoretic and information-retrieval algorithms used here (Louvain
community detection with C2 repair, BM25, RRF rank fusion, tree-sitter AST
parsing, Tarjan strongly-connected-components) are sourced from published
research; citations are documented inline via // source: annotations and in
docs/. The MIT license covers this implementation; it does not assert
ownership over the underlying algorithms, which remain attributable to their
original authors.
Built by cdeust. Every stage designed by a genius agent. Every constant sourced.