from app.checks import http, shell

CHECK_MODULES = {
    "http": http,
    "shell": shell,
}

def load_check_module(check_type: str):
    if check_type not in CHECK_MODULES:
        raise ValueError(f"Unsupported check type: {check_type}")
    return CHECK_MODULES[check_type]