src/retrieval/__init__.py

"""Retrieval components for hybrid search."""

# Only import memory retriever by default (no OpenSearch dependency)
from .memory_retriever import MemoryRetriever

__all__ = ["MemoryRetriever"]

# Optional OpenSearch retriever (only if needed)
try:
    from .hybrid_retriever import HybridRetriever
    __all__.append("HybridRetriever")
except ImportError:
    pass  # OpenSearch dependencies not installed

# Optional query rewriter
try:
    from .query_rewriter import QueryRewriter, RewrittenQueries
    __all__.extend(["QueryRewriter", "RewrittenQueries"])
except ImportError:
    pass  # PydanticAI not installed