src/colpali/__init__.py

"""
ColPali Visual Document Retrieval Module

This module implements visual document retrieval using ColPali (ColBERT-style multi-vector embeddings)
for processing PDF documents as images.

All components are self-contained within src/colpali/ - no external dependencies on colpali_colab_package.
"""

# Core inference components
from .processor import ColPaliProcessor
from .search import VisualDocumentSearch
from .visualizer import generate_saliency_maps   # data
from .visual_search import VisualSearchAdapter, VisualSearchResult, create_visual_search_adapter


__all__ = [
    "ColPaliProcessor",
    "VisualSearchResult",
    "VisualDocumentSearch",
    "VisualSearchAdapter",
    "generate_saliency_maps",
    "create_visual_search_adapter"
]