trustgraph/tests/unit/test_retrieval
Sunny Yang 2bdc930b2a
feat: hybrid retrieval (BM25 + vector RRF fusion) for document-RAG (#875) (#1030)
Adds a sparse keyword retrieval path beside the existing vector path in
document-RAG, fused by weighted Reciprocal Rank Fusion on chunk_id, behind
--retrieval-mode (vector | keyword | hybrid, default vector).

The keyword index is a new pluggable service (KeywordIndexService /
KeywordIndexClientSpec); the first backend is SQLite FTS5, consuming Chunk
messages off the ingestion stream and answering BM25 queries from one
process, since the index is a single local file. Query text is sanitized
into per-term quoted phrases (raw text is not valid FTS5 syntax), which
also makes dotted clause numbers and error codes exact-match without a
trigram index. Indexes are scoped per (workspace, collection) and dropped
on collection deletion.

The keyword-index client spec is only registered when the sparse path is
enabled, so existing flow definitions without keyword-index queues are
untouched; with retrieval_mode=vector the retrieval path is unchanged. In
hybrid mode a keyword-path failure degrades to vector-only.
2026-07-07 12:54:02 +01:00
..
test_structured_diag feat: workspace-based multi-tenancy, replacing user as tenancy axis (#840) 2026-04-21 23:23:01 +01:00
test_document_rag.py feat: add cross-encoder reranking to Document-RAG with two-limit control (#878) (#1011) 2026-07-02 09:50:13 +01:00
test_document_rag_diversity_selection.py fix: remove test import (#1017) 2026-07-03 13:44:09 +01:00
test_document_rag_hybrid.py feat: hybrid retrieval (BM25 + vector RRF fusion) for document-RAG (#875) (#1030) 2026-07-07 12:54:02 +01:00
test_document_rag_provenance_integration.py Fixed document-rag workspace problem (#866) 2026-05-06 14:55:21 +01:00
test_document_rag_rerank.py Add diversity-aware selection after Document-RAG reranking (#1014) 2026-07-03 13:35:42 +01:00
test_document_rag_reranker_wiring.py feat: add cross-encoder reranking to Document-RAG with two-limit control (#878) (#1011) 2026-07-02 09:50:13 +01:00
test_document_rag_service.py feat: add cross-encoder reranking to Document-RAG with two-limit control (#878) (#1011) 2026-07-02 09:50:13 +01:00
test_graph_rag.py feat: replace LLM edge scoring with cross-encoder reranker in GraphRAG (#1005) 2026-06-30 14:36:37 +01:00
test_graph_rag_direction_aware_text.py feat: filter and cap GraphRAG reranker input across full stack (#1021) 2026-07-03 15:51:04 +01:00
test_graph_rag_explain_forwarding.py feat: workspace-based multi-tenancy, replacing user as tenancy axis (#840) 2026-04-21 23:23:01 +01:00
test_graph_rag_provenance_integration.py feat: replace LLM edge scoring with cross-encoder reranker in GraphRAG (#1005) 2026-06-30 14:36:37 +01:00
test_graph_rag_service.py feat: workspace-based multi-tenancy, replacing user as tenancy axis (#840) 2026-04-21 23:23:01 +01:00
test_nlp_query.py feat: workspace-based multi-tenancy, replacing user as tenancy axis (#840) 2026-04-21 23:23:01 +01:00
test_structured_query.py feat: workspace-based multi-tenancy, replacing user as tenancy axis (#840) 2026-04-21 23:23:01 +01:00