-
Notifications
You must be signed in to change notification settings - Fork 34
ColPali-based RAG pipeline for PDF #186
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: master
Are you sure you want to change the base?
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pull Request Overview
This PR adds a ColPali-based document retrieval system with three main components: PDF processing, embedding indexing, and query retrieval. The implementation uses ColPali embeddings with Milvus for vector storage and supports API, batch, and single-query retrieval modes.
Key Changes
- Introduces PDF processing pipeline that converts PDFs to embeddings using ColPali models
- Implements Milvus-based indexing system for storing and searching ColPali embeddings
- Adds retrieval functionality with API server, batch processing, and single-query modes
Reviewed Changes
Copilot reviewed 7 out of 8 changed files in this pull request and generated 13 comments.
Show a summary per file
| File | Description |
|---|---|
| src/mmore/colpali/run_retriever.py | Implements query retrieval with multiple modes (API/batch/single) |
| src/mmore/colpali/run_process.py | Processes PDFs to generate ColPali embeddings and stores them in Parquet |
| src/mmore/colpali/run_index.py | Indexes embeddings from Parquet into Milvus database |
| src/mmore/colpali/milvuscolpali.py | Core Milvus operations manager for ColPali embeddings |
| examples/colpali/config_retrieval.yml | Configuration for retrieval operations |
| examples/colpali/config_process.yml | Configuration for PDF processing |
| examples/colpali/config_index.yml | Configuration for indexing operations |
Comments suppressed due to low confidence (1)
src/mmore/colpali/run_retriever.py:2
- Import of 'concurrent' is not used.
import concurrent.futures
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
Co-authored-by: Copilot <[email protected]>
Co-authored-by: Copilot <[email protected]>
Co-authored-by: Copilot <[email protected]>
Co-authored-by: Copilot <[email protected]>
Co-authored-by: Copilot <[email protected]>
Co-authored-by: Copilot <[email protected]>
Co-authored-by: Copilot <[email protected]>
Co-authored-by: Copilot <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pull request overview
Copilot reviewed 12 out of 13 changed files in this pull request and generated 18 comments.
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
This PR introduces an independent, end-to-end RAG pipeline for PDF document, built to showcase the ease-of-use of the ColPali model for embedding and retrieval.
The pipeline demonstrates: