A new context database for reasoning-driven retrieval via tree search.
Fast, context-aware retrieval at scale with up to 70% less token cost.
ConDB (Context Database) is a tree-structured context database that uses LLM-powered reasoning-based retrieval via tree search instead of vector similarity β no vector DB, no chunking. It accepts PageIndex-compatible document trees, ChatIndex conversation trees, filesystem trees, and custom hierarchical JSON β with no runtime dependency on either. The LLM reasons over the tree, like a human expert using a table of contents, to locate relevant content.
- Similarity β relevance β vector search retrieves what looks similar, not what is truly relevant. Similar-looking chunks may differ in intent (low accuracy), while truly relevant information may be expressed in very different language and get missed entirely (low recall). True relevance requires reasoning
- Chunking breaks semantic continuity β documents must be split into fixed-size segments to fit embedding models, causing context fragmentation that destroys their natural structure and cross-section relationships
- Retrieval is blind to context β embedding models encode the query alone, ignoring conversational history, user intent, and other contextual signals
ConDB replaces this with reasoning-based tree search: the LLM performs node-level relevance classification over a hierarchical index, incorporating full context β making retrieval adaptive, explainable, and traceable.
- Fast tree search at scale β reasoning-driven tree search with block partitioning and parallel processing, supporting complex, context-aware retrieval over large hierarchical structures
- KV-cache native β the first database designed around LLM KV-cache reuse. By caching intermediate results during tree search, ConDB reduces token usage by up to 70% with no loss in accuracy. The same efficiency gains extend to memory systems for long-context reasoning at scale
- Unified long-context infrastructure β a single system for both static and dynamic long-context workloads
Structured, persistent knowledge β documents (via PageIndex), file systems, and codebases. Scalable retrieval within large, organized hierarchies.
Evolving, runtime context β agent memory, long conversations (via ChatIndex), and autoresearch. Systems can continuously update, retrieve, and reason over newly generated information.
- Hierarchical storage β document trees, chat trees, and custom hierarchical JSON in SQLite
- Multiple retrieval strategies β beam search for small trees, block retrieval for large documents
- Multi-provider LLM support β Anthropic (Claude) and OpenAI (GPT) out of the box
- Extensible β plug in custom storage backends, LLM providers, or retrieval strategies
pip install -r requirements.txtimport contextdb
# Open database
db = contextdb.open("my_docs.sqlite")
# Configure LLM
db.set_llm(provider="anthropic", model="claude-sonnet-4-6")
# Store a document tree
tree_id = db.store(document_tree_json, format="document")
# Query with LLM reasoning
result = db.query(tree_id, "What are the key findings?")
print(result.contents)from contextdb import ContextTree
def build_markdown_tree(path: str) -> dict:
...
ct = ContextTree("context.sqlite")
tree_id = ct.index_markdown_file("doc.md", tree_builder=build_markdown_tree)
# You can also generate a tree out of process and call:
# tree_id = ct.index_document_tree(document_tree_json)
ct.close()Create a .env file with your API keys:
ANTHROPIC_API_KEY=sk-...
OPENAI_API_KEY=sk-...
Model and provider settings live in contextdb/config/config.yaml:
llm:
provider: anthropic # anthropic or openai
model: claude-sonnet-4-6 # any model the provider supports
context_limit: 100000
max_concurrent: 10
retriever:
beam_size: 3
max_turns: 5Override at runtime with environment variables:
LLM_MODEL=claude-opus-4-6 python your_script.pyConDB automatically selects the best retrieval strategy based on tree size:
| Strategy | Best for | How it works |
|---|---|---|
| Beam | Small trees (< 50 nodes) |
LLM evaluates and selects promising branches at each depth level |
| Block | Large documents (50+ nodes) |
Splits tree into token-bounded blocks, LLM reasons over each block. KV-cache native β caches intermediate block results to cut token usage by up to 70% |
You can also specify a strategy explicitly:
result = db.query(tree_id, "question", strategy="block", beam_size=3)Two benchmarks live under bench/.
Runs on AmuroEita/SWEBench-FileTree,
a path-only version of SWE-bench code retrieval:
- 500 GitHub issues as queries
- 475
(repo, commit)repository snapshots as independent retrieval universes - 58,058 file paths; no source code, no file summaries
Given an issue and one snapshot's file tree, return the file(s) the fix
touches. Specification: notes/condb_swebench_filetree_bench.md.
export ANTHROPIC_API_KEY=sk-ant-...
python bench/run_swebench_filetree.py --tier mediumTiers (by retriever difficulty; lower difficulty = more path signal in query):
easy 107 queries gold path appears in query text (sanity check)
medium 133 queries gold filename appears in query (main report)
hard 261 queries gold module stem appears (fuzzy matching)
all 500 queries no filter, includes ~48% path-signal-less queries
Output goes to bench/runs/<timestamp>__<tier>/: report.md, summary.json,
per_query.jsonl.
Block mode can optionally rerank only the cross-block merge candidates before the file/directory split:
python bench/run_swebench_filetree.py --tier medium --strategy block --ranker vectorAvailable rankers are none, bm25, and vector. The vector ranker uses
LiteLLM embeddings (--embedding-provider, --embedding-model) and leaves
the default ranker=none unchanged.
Claude Sonnet 4.6, --ranker none, --top-k 10, 500 queries, 0 failures.
Block (ConDB) is compared against a Vertical baseline β a per-beam variant
that expands each parent's children into separate subtree blocks (AβB,
AβC), one LLM call per branch, losing the cross-branch view Block keeps.
| variant | recall@gold | exact@gold | MRR | nDCG@10 | avg returned | avg latency |
|---|---|---|---|---|---|---|
| Vertical (baseline) | 0.382 | 0.366 | 0.466 | 0.481 | 3.00 | ~24 s |
| Block (ConDB) | 0.711 | 0.672 | 0.805 | 0.813 | 7.20 | ~8 s |
Block gains +0.33 recall@gold at ~3Γ lower latency.
Block per-gold-count breakdown β the cutoff is the query's gold-file count
(6+ row contains one 6-gold query and one 21-gold outlier):
| gold files | queries | cutoff | recall@gold | exact@gold | found@gold | avg returned |
|---|---|---|---|---|---|---|
| 1 | 430 | 1 | 0.749 | 0.749 | 0.75 | 7.00 |
| 2 | 48 | 2 | 0.521 | 0.271 | 1.04 | 8.31 |
| 3 | 13 | 3 | 0.410 | 0.077 | 1.23 | 8.77 |
| 4 | 6 | 4 | 0.417 | 0.000 | 1.67 | 9.17 |
| 5 | 1 | 5 | 0.200 | 0.000 | 1.00 | 2.00 |
| 6+ | 2 | gold | 0.274 | 0.000 | 2.00 | 10.00 |
Reproduce:
python bench/run_swebench_filetree.py --tier all --strategy block --ranker none --top-k 10
python bench/run_swebench_filetree.py --tier all --strategy vertical --ranker none --top-k 10Compares retriever algorithms (Block / Beam / Vertical / ...) on one hierarchical document. Reports time, LLM calls, token usage with prompt caching, and USD cost.
python bench/run_document_bench.py \
--doc examples/large_doc.json \
--config bench/queries.jsonQueries live in the config JSON as {"queries": ["...", "..."]}. Swap in
any --doc and any --config to benchmark a different document.
contextdb/
βββ api/
β βββ condb.py # ConDB β main entry point
β βββ context_tree.py # ContextTree β tree indexing + query API
βββ core/
β βββ storage.py # TreeDB (SQLite), StorageProtocol
βββ adapter/
β βββ base.py # DocumentTree, ChatIndex, Generic adapters
βββ retriever/
β βββ base.py # Retriever protocols
β βββ algorithm/ # Beam, Block retrieval strategies
βββ llm.py # LLMClient (Anthropic, OpenAI)
βββ config/ # YAML configs for retrievers
βββ prompts/ # Jinja2 prompt templates
Custom Storage Backend
from contextdb import StorageProtocol
class MyStorage:
def get_node(self, tree_id, node_id): ...
def get_children(self, tree_id, node_id): ...
# implement StorageProtocol methods
ct = ContextTree(storage=MyStorage())Custom LLM Provider
from contextdb import LLMProtocol
class MyLLM:
def chat(self, messages, system="", tools=None):
return {"content": [...], "stop_reason": "..."}
ct = ContextTree("db.sqlite", llm=MyLLM())./run_tests.sh all- PageIndex β vectorless, reasoning-based RAG that builds hierarchical tree indexes from long documents
- ChatIndex β tree indexing for long conversations, enabling reasoning-based retrieval over chat histories
- AgentFS β filesystem for AI agents
Licensed under Apache 2.0.
Β© 2026 Vectify AI