from .core import (
    BayesianIndex,
    BayesianRetrievalConfig,
    DocumentRecord,
    RetrievalHit,
    ExtractiveGroundedGenerator,
)

__all__ = [
    "BayesianIndex",
    "BayesianRetrievalConfig",
    "DocumentRecord",
    "RetrievalHit",
    "ExtractiveGroundedGenerator",
]

try:
    from .retriever import BayesianRetriever
    from .rag import BayesianRAG
    __all__ += ["BayesianRetriever", "BayesianRAG"]
except ImportError:
    # Core retrieval/evaluation remains usable before optional framework deps are installed.
    pass