File size: 601 Bytes
c4ef1cf
 
 
 
 
 
 
 
 
9513cca
c4ef1cf
9513cca
c4ef1cf
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
"""
Retrieval module - Search and retrieval strategies.

Components:
- TwoStageRetriever: Pooled prefetch → MaxSim reranking (our novel contribution)
- SingleStageRetriever: Direct multi-vector or pooled search
"""

from visual_rag.retrieval.multi_vector import MultiVectorRetriever
from visual_rag.retrieval.single_stage import SingleStageRetriever
from visual_rag.retrieval.three_stage import ThreeStageRetriever
from visual_rag.retrieval.two_stage import TwoStageRetriever

__all__ = [
    "TwoStageRetriever",
    "SingleStageRetriever",
    "MultiVectorRetriever",
    "ThreeStageRetriever",
]