Are you frustrated with vector database retrieval accuracy for long professional documents? Traditional vector-based RAG relies on semantic similarity rather than true relevance. But similarity β relevance β what we truly need in retrieval is relevance, and that requires reasoning. When working with professional documents that demand domain expertise and multi-step reasoning, similarity search often falls short.
π§ Reasoning-based RAG offers a better alternative: enabling LLMs to think and reason their way to the most relevant document sections. Inspired by AlphaGo, we use tree search to perform structured document retrieval.
PageIndex is a document indexing system that builds search tree structures from long documents, making them ready for reasoning-based RAG.
Self-host it with this open-source repo, or try our βοΈ Cloud service β no setup required, with advanced features like OCR for complex and scanned PDFs.
PageIndex powers a reasoning-based RAG system that achieved 98.7% accuracy on FinanceBench, demonstrating state-of-the-art performance in financial document analysis.
Built by Vectify AI.
PageIndex can transform lengthy PDF documents into a semantic tree structure, similar to a "table of contents" but optimized for use with Large Language Models (LLMs). It's ideal for: financial reports, regulatory filings, academic textbooks, legal or technical manuals, and any document that exceeds LLM context limits.
-
Hierarchical Tree Structure
Enables LLMs to traverse documents logically β like an intelligent, LLM-optimized table of contents. -
Chunk-Free Segmentation
No arbitrary chunking. Nodes follow the natural structure of the document. -
Precise Page Referencing
Every node contains its summary and start/end page physical index, allowing pinpoint retrieval. -
Scales to Massive Documents
Designed to handle hundreds or even thousands of pages with ease.
Here is an example output. See more example documents and generated trees.
...
{
"title": "Financial Stability",
"node_id": "0006",
"start_index": 21,
"end_index": 22,
"summary": "The Federal Reserve ...",
"nodes": [
{
"title": "Monitoring Financial Vulnerabilities",
"node_id": "0007",
"start_index": 22,
"end_index": 28,
"summary": "The Federal Reserve's monitoring ..."
},
{
"title": "Domestic and International Cooperation and Coordination",
"node_id": "0008",
"start_index": 28,
"end_index": 31,
"summary": "In 2023, the Federal Reserve collaborated ..."
}
]
}
...
A bug introduced on April 18 has now been fixed.
If you pulled the repo between April 18β23, please update to the latest version:
git pull origin main
Thanks for your understanding π
Follow these steps to generate a PageIndex tree from a PDF document.
pip3 install -r requirements.txt
Create a .env
file in the root directory and add your API key:
CHATGPT_API_KEY=your_openai_key_here
python3 run_pageindex.py --pdf_path /path/to/your/document.pdf
You can customize the processing with additional optional arguments:
--model OpenAI model to use (default: gpt-4o-2024-11-20)
--toc-check-pages Pages to check for table of contents (default: 20)
--max-pages-per-node Max pages per node (default: 10)
--max-tokens-per-node Max tokens per node (default: 20000)
--if-add-node-id Add node ID (yes/no, default: yes)
--if-add-node-summary Add node summary (yes/no, default: no)
--if-add-doc-description Add doc description (yes/no, default: yes)
Don't want to host it yourself? Try our hosted API for PageIndex. The hosted service leverages our custom OCR model for more accurate PDF recognition, delivering better tree structures for complex documents. Ideal for rapid prototyping, production environments, and documents requiring advanced OCR.
You can also upload PDFs from your browser and explore results visually with our web Dashboard β no coding needed.
Leave your email in this form to receive 1,000 pages for free.
Mafin 2.5 is a state-of-the-art reasoning-based RAG model designed specifically for financial document analysis. Powered by PageIndex, it achieved a market-leading 98.7% accuracy on the FinanceBench benchmark β significantly outperforming traditional vector-based RAG systems.
PageIndex's hierarchical indexing enabled precise navigation and extraction of relevant content from complex financial reports, such as SEC filings and earnings disclosures.
π See the full benchmark results and our blog post for detailed comparisons and performance metrics.
Use PageIndex to build reasoning-based retrieval systems without relying on semantic similarity. Great for domain-specific tasks where nuance matters (more examples).
- Process documents using PageIndex to generate tree structures.
- Store the tree structures and their corresponding document IDs in a database table.
- Store the contents of each node in a separate table, indexed by node ID and tree ID.
- Query Preprocessing:
- Analyze the query to identify the required knowledge
- Document Selection:
- Search for relevant documents and their IDs
- Fetch the corresponding tree structures from the database
- Node Selection:
- Search through tree structures to identify relevant nodes
- LLM Generation:
- Fetch the corresponding contents of the selected nodes from the database
- Format and extract the relevant information
- Send the assembled context along with the original query to the LLM
- Generate contextually informed responses
prompt = f"""
You are given a question and a tree structure of a document.
You need to find all nodes that are likely to contain the answer.
Question: {question}
Document tree structure: {structure}
Reply in the following JSON format:
{{
"thinking": <reasoning about where to look>,
"node_list": [node_id1, node_id2, ...]
}}
"""
π For more examples, see the PageIndex Dashboard.
- Detailed examples of document selection, node selection, and RAG pipelines
- Integration of reasoning-based retrieval and semantic-based retrieval
- Efficient tree search methods introduction
- Technical report on the design of PageIndex
This project is in its early beta development, and all progress will remain open and transparent. We welcome you to raise issues, reach out with questions, or contribute directly to the project.
Due to the diverse structures of PDF documents, you may encounter instability during usage. For a more accurate and stable version with a leading OCR integration, please try our hosted API for PageIndex. Leave your email in this form to receive 1,000 pages for free.
Together, let's push forward the revolution of reasoning-based RAG systems.
-
Does PageIndex support other LLMs besides OpenAI?
Currently optimized for GPT models, but future versions will support more. -
Can PageIndex handle scanned PDFs?
Yes! Our Cloud API includes advanced OCR specifically for scanned and complex PDFs.
Need customized support for your documents or reasoning-based RAG system?
π’ Join our Discord
βοΈ Leave us a message