Metadata-Version: 2.4
Name: kb-arena
Version: 0.10.0
Summary: Compare retrieval architectures on your documentation and choose with reproducible evidence
Project-URL: Homepage, https://github.com/xmpuspus/kb-arena
Project-URL: Documentation, https://github.com/xmpuspus/kb-arena/tree/main/docs
Project-URL: Repository, https://github.com/xmpuspus/kb-arena
Project-URL: Issues, https://github.com/xmpuspus/kb-arena/issues
Author-email: Xavier Puspus <36430014+xmpuspus@users.noreply.github.com>
License-Expression: MIT
License-File: LICENSE
Keywords: benchmark,chromadb,document-retrieval,evaluation,knowledge-graph,llm,neo4j,rag,rag-evaluation,retrieval,vector-search
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: <3.14,>=3.11
Requires-Dist: anthropic==0.42.0
Requires-Dist: beautifulsoup4==4.12.3
Requires-Dist: chromadb==0.5.23
Requires-Dist: click==8.4.2
Requires-Dist: fastapi==0.141.1
Requires-Dist: httpx==0.28.1
Requires-Dist: jellyfish==1.1.3
Requires-Dist: neo4j==5.27.0
Requires-Dist: networkx==3.4.2
Requires-Dist: openai==2.24.0
Requires-Dist: pydantic-settings==2.7.1
Requires-Dist: pydantic==2.10.4
Requires-Dist: pyyaml==6.0.2
Requires-Dist: rank-bm25==0.2.2
Requires-Dist: rich==13.9.4
Requires-Dist: scipy>=1.11.0
Requires-Dist: sse-starlette==2.2.1
Requires-Dist: starlette==1.4.1
Requires-Dist: tiktoken==0.13.0
Requires-Dist: typer==0.27.1
Requires-Dist: uvicorn[standard]==0.34.0
Provides-Extra: all-formats
Requires-Dist: httpx==0.28.1; extra == 'all-formats'
Requires-Dist: mammoth==1.8.0; extra == 'all-formats'
Requires-Dist: pymupdf==1.25.3; extra == 'all-formats'
Provides-Extra: dev
Requires-Dist: httpx2==2.9.1; extra == 'dev'
Requires-Dist: pytest-asyncio==0.24.0; extra == 'dev'
Requires-Dist: pytest==8.3.4; extra == 'dev'
Requires-Dist: ruff==0.8.6; extra == 'dev'
Provides-Extra: docx
Requires-Dist: mammoth==1.8.0; extra == 'docx'
Provides-Extra: pdf
Requires-Dist: pymupdf==1.25.3; extra == 'pdf'
Provides-Extra: quantum
Requires-Dist: qiskit-aer==0.17.2; extra == 'quantum'
Requires-Dist: qiskit==2.4.2; extra == 'quantum'
Requires-Dist: scikit-learn==1.9.0; extra == 'quantum'
Provides-Extra: rerank
Requires-Dist: sentence-transformers<6,>=5.0; extra == 'rerank'
Provides-Extra: web
Requires-Dist: httpx==0.28.1; extra == 'web'
Description-Content-Type: text/markdown

# KB Arena

Compare retrieval architectures on your own documentation and choose with evidence.

[![PyPI](https://img.shields.io/pypi/v/kb-arena)](https://pypi.org/project/kb-arena/)
[![Python](https://img.shields.io/pypi/pyversions/kb-arena)](https://pypi.org/project/kb-arena/)
[![CI](https://github.com/xmpuspus/kb-arena/actions/workflows/ci.yml/badge.svg)](https://github.com/xmpuspus/kb-arena/actions/workflows/ci.yml)
[![License](https://img.shields.io/badge/license-MIT-blue)](LICENSE)
[![DOI](https://zenodo.org/badge/1182030516.svg)](https://zenodo.org/badge/latestdoi/1182030516)

KB Arena runs lexical, dense, graph, hybrid, hierarchical, and reranked retrieval on the same
corpus and question set. It records retrieval quality, answer quality, latency, cost, and run
artifacts so you can decide which design fits your data.

Use it before you commit to a retrieval architecture, or as a regression lab after your corpus,
chunking, model, or index changes.

![Historical KB Arena retrieval result](https://raw.githubusercontent.com/xmpuspus/kb-arena/main/docs/demo.gif)

## Explore a checked result

The packaged demo uses precomputed AWS Compute results. It needs no API key, Docker service, or
Neo4j instance.

```bash
pip install kb-arena
kb-arena demo
```

Open the URL printed by the command. The dashboard exposes the benchmark table, per-tier results,
source drill-down, run comparisons, and the Retriever Lab.

## What KB Arena helps decide

| Question | Comparison |
|---|---|
| Do exact terms matter more than semantic similarity? | BM25 against dense retrieval |
| Does document context improve chunk retrieval? | Naive against contextual vector |
| Do cross-document relationships justify a graph? | Dense against graph and hybrid |
| Does hierarchy help on broad questions? | Dense against RAPTOR and PageIndex |
| Is a reranker worth its latency and cost? | Dense against reranked dense |
| Is a proposed method meaningfully different? | Paired scores, confidence intervals, and rank overlap |

KB Arena does not give a universal leaderboard. A result applies to the corpus, questions,
ground truth, configuration, and models named in that run.

## Evidence included in this repository

The tracked Retriever Lab run `855aac4e` is a historical, reproducible example:

- Corpus: `aws-compute`, three documents and 1,549 words
- Run date: 2026-04-26
- Questions: 75 across five tiers
- Cutoff: top 5 chunks
- Chunk-level labels: 35 questions have labels, and 40 do not
- Scope: eight strategies from the version available on the run date

| Strategy | Recall@5 | MRR | NDCG@5 |
|---|---:|---:|---:|
| Contextual Vector | 0.355 | 0.433 | 0.388 |
| Naive Vector | 0.352 | 0.414 | 0.367 |
| RAPTOR | 0.352 | 0.414 | 0.367 |
| BM25 | 0.275 | 0.352 | 0.278 |
| Hybrid | 0.080 | 0.093 | 0.086 |
| PageIndex | 0.061 | 0.111 | 0.076 |

![Historical retrieval metrics from run 855aac4e](https://raw.githubusercontent.com/xmpuspus/kb-arena/main/docs/benchmark-evidence.png)

Source: [tracked report](https://github.com/xmpuspus/kb-arena/blob/main/results/run_855aac4e/retriever_lab.md) and
[run artifact](https://github.com/xmpuspus/kb-arena/blob/main/results/run_855aac4e/retriever_lab.json).

These numbers show the report format and calculation path. The corpus is too small, and
its chunk labels are too incomplete, to support a general winner claim. Q&A Pairs and Knowledge
Graph from that run have zero chunk-level scores because their retrieved identifiers did not map
to the available labels. Those zeroes reflect gaps in the current evaluation, not evidence that the
methods cannot retrieve useful context.

## A larger public corpus for method development

The repository includes a deterministic NIST SP 800-171 Revision 3 corpus built from the
official publication. It has 130 control documents and 80 questions across direct,
paraphrased, scenario, boundary, and multi-control categories. Each question maps to source control
sections, with 48 development, 12 validation, and 20 holdout items.

The question set is not a human-approved benchmark but a machine-generated draft. Do not publish a
strategy winner from it until a qualified reviewer checks the questions, answers, constraints, and
holdout isolation. See the [corpus notes](https://github.com/xmpuspus/kb-arena/blob/main/datasets/nist-800-171-r3/README.md),
[source manifest](https://github.com/xmpuspus/kb-arena/blob/main/datasets/nist-800-171-r3/source-manifest.json), and
[evaluation method](https://github.com/xmpuspus/kb-arena/blob/main/docs/methodology.md).

## Run it on your documents

### Local models with Ollama

Install and start [Ollama](https://ollama.com/), then pull both generation and embedding models:

```bash
ollama pull llama3.1:8b
ollama pull nomic-embed-text

export KB_ARENA_LLM_PROVIDER=ollama
export KB_ARENA_EMBEDDING_PROVIDER=ollama
```

Create a corpus, place files in `raw/`, and run the pipeline:

```bash
pip install 'kb-arena[all-formats]'
kb-arena init-corpus my-docs
cp -R /path/to/docs/. datasets/my-docs/raw/
kb-arena run --corpus my-docs --skip-graph
```

![Create and ingest an example documentation corpus](https://raw.githubusercontent.com/xmpuspus/kb-arena/main/docs/demo-own-docs.gif)

`run` ingests a populated `raw/` directory automatically. Remove `--skip-graph` after starting
Neo4j when you want graph and hybrid comparisons.

### Hosted generation and embeddings

The generation and embedding providers are independent:

```bash
export KB_ARENA_LLM_PROVIDER=anthropic
export KB_ARENA_ANTHROPIC_API_KEY=...
export KB_ARENA_EMBEDDING_PROVIDER=openai
export KB_ARENA_OPENAI_API_KEY=...

kb-arena run --corpus my-docs
```

Supported embedding providers are OpenAI, Voyage, Cohere, Gemini, local BGE, and Ollama. See the
[getting-started guide](https://github.com/xmpuspus/kb-arena/blob/main/docs/getting-started.md) for formats, Neo4j setup, checkpoints, and provider
configuration.

## Evaluation paths

Use the retrieval-only path when you need to isolate the index and ranking behavior:

```bash
kb-arena label-chunks --corpus my-docs
kb-arena retriever-lab --corpus my-docs --top-k 5
```

It reports Recall@k, Precision@k, Hit@k, MRR, NDCG@k, MAP, R-Precision, bpref,
bootstrap confidence intervals, and per-tier breakdowns.

Use the full benchmark for generated-answer scoring, source attribution, latency,
cost, and reliability:

```bash
kb-arena benchmark --corpus my-docs --top-k 5
kb-arena report --corpus my-docs --format markdown
```

Use optimization after you define a development split. Keep a separate holdout for the published
comparison:

```bash
kb-arena optimize \
  --corpus my-docs \
  --split development \
  --strategies bm25,naive_vector,contextual_vector,raptor \
  --top-ks 3,5,10 \
  --metric ndcg
```

After you choose the configuration, run the public comparison with
`kb-arena benchmark --split holdout`.

Optimization remains retrieval-only. QnA Pairs and RAPTOR reuse their prebuilt indexes and sweep
top-k only. They do not regenerate pairs or summaries during a search.

Read [the evaluation method](https://github.com/xmpuspus/kb-arena/blob/main/docs/methodology.md) before interpreting small score differences or
synthetic question sets.

## Strategy catalog

The catalog includes the strategies below. The default `all` benchmark excludes Rerank Vector and
SQR. Rerank Vector needs the dependency for its selected backend, and SQR needs the optional
`quantum` dependency group. The API reports loaded and unavailable strategies at `GET /strategies`.

| Strategy | Architecture | Default | Notes |
|---|---|:---:|---|
| Naive Vector | Dense | Yes | Chunk, embed, cosine retrieval |
| Contextual Vector | Dense | Yes | Adds parent context before embedding |
| Q&A Pairs | Generated index | Yes | Creates likely questions at index time |
| Knowledge Graph | Graph | Yes | Retrieves through Neo4j entities and relationships |
| Hybrid | Hybrid | Yes | Routes and fuses vector and graph results with RRF |
| RAPTOR | Hierarchical | Yes | Retrieves chunks and recursive summaries |
| PageIndex | Hierarchical | Yes | Uses document structure and LLM tree traversal |
| BM25 | Lexical | Yes | Keyless keyword baseline |
| Rerank Vector | Reranked dense | No | The BGE backend uses `kb-arena[rerank]`. |
| QISS | Experimental | Yes | Pure NumPy fidelity reranker over dense candidates |
| SQR | Experimental | No | Qiskit Aer SWAP-test reranker, install `kb-arena[quantum]` |

See [strategy details](https://github.com/xmpuspus/kb-arena/blob/main/docs/strategy-catalog.md) and the
[plugin guide](https://github.com/xmpuspus/kb-arena/blob/main/CONTRIBUTING.md#add-a-strategy).

## Data and method limits

- Auto-generated questions help expand coverage, but production queries and human review give
  stronger deployment evidence.
- An LLM judge can introduce model bias. Use a different judge family, keep the prompts and model
  versions, and inspect disagreements.
- Architecture-native indexes do not always return the same chunk identifiers. Validate qrel
  mappings before comparing retrieval metrics.
- Cost and latency depend on provider, model, cache state, hardware, concurrency, and region.
- Tune on development data. Publish results only from a sealed holdout.
- Quantum strategies are experiments. The AWS sample does not show a Recall@5 gain over the dense
  baseline.

The [method guide](https://github.com/xmpuspus/kb-arena/blob/main/docs/methodology.md) defines the evidence that belongs with a public result.

## Project references

- [Getting started](https://github.com/xmpuspus/kb-arena/blob/main/docs/getting-started.md)
- [Evaluation method](https://github.com/xmpuspus/kb-arena/blob/main/docs/methodology.md)
- [Retriever Lab](https://github.com/xmpuspus/kb-arena/blob/main/docs/retriever-lab.md)
- [Strategy catalog](https://github.com/xmpuspus/kb-arena/blob/main/docs/strategy-catalog.md)
- [Changelog](https://github.com/xmpuspus/kb-arena/blob/main/CHANGELOG.md)
- [Security policy](https://github.com/xmpuspus/kb-arena/blob/main/SECURITY.md)
- [Contributing](https://github.com/xmpuspus/kb-arena/blob/main/CONTRIBUTING.md)

## Development

```bash
git clone https://github.com/xmpuspus/kb-arena
cd kb-arena
pip install -e '.[dev]'
ruff check .
ruff format --check .
pytest tests/ -q --ignore=tests/live
```

The frontend uses Next.js 16 and needs Node.js 20.9 or later:

```bash
cd web
npm ci
npm run lint
npm run build
```

## Citation

The canonical citation metadata is in [CITATION.cff](https://github.com/xmpuspus/kb-arena/blob/main/CITATION.cff). GitHub can export it through
the repository's **Cite this repository** action. The archived software record is available through
the DOI badge above.

## License

[MIT](https://github.com/xmpuspus/kb-arena/blob/main/LICENSE)
