"""
Module de base de données pour la persistance des données du pipeline.

Ce module contient les modèles SQLAlchemy, la configuration de la base de données,
et les utilitaires pour la gestion de la persistance.
"""

from pipeline_mco_pmsi.database.base import Base, get_engine, get_session
from pipeline_mco_pmsi.database.models import (
    AuditRecordDB,
    ClinicalDocumentDB,
    ClinicalFactDB,
    CodeDB,
    EvidenceDB,
    GroupageResultDB,
    QuestionDB,
    ReferentielVersionDB,
    StayDB,
    TIMCorrectionDB,
    ValidationIssueDB,
    VerificationResultDB,
)

__all__ = [
    "Base",
    "get_engine",
    "get_session",
    "StayDB",
    "ClinicalDocumentDB",
    "ClinicalFactDB",
    "EvidenceDB",
    "CodeDB",
    "ValidationIssueDB",
    "QuestionDB",
    "AuditRecordDB",
    "ReferentielVersionDB",
    "TIMCorrectionDB",
    "VerificationResultDB",
    "GroupageResultDB",
]
