database_module/__init__.py

# database_module/__init__.py
from .db import init_db
from .mcp_tools import (
    get_all_models_handler,
    search_models_handler,
    save_diagnostic_data,
    get_baseline_diagnostics,
    save_drift_score,
    register_model_with_capabilities
)