app/db/models/__init__.py

# app/db/models/__init__.py
from app.db.models.user import UserDocument, UserRole
from app.db.models.scan_result import ScanResultDocument, RiskLevel
from app.db.models.alert import AlertDocument, AlertSeverity, AlertStatus

__all__ = [
    "UserDocument", "UserRole",
    "ScanResultDocument", "RiskLevel",
    "AlertDocument", "AlertSeverity", "AlertStatus",
]