"""Build-time model prefetch utility for reducing cold-start downloads."""

import asyncio

from app.core.config import settings
from app.core.logging import get_logger, setup_logging
from app.services.model_registry import get_model_registry


setup_logging()
logger = get_logger(__name__)


async def main() -> None:
    """Download fusion and submodel repositories into the configured HF cache."""
    logger.info("Starting build-time model prefetch for %s", settings.HF_FUSION_REPO_ID)
    registry = get_model_registry()
    await registry.load_from_fusion_repo(settings.HF_FUSION_REPO_ID, force_reload=True)
    logger.info("Build-time model prefetch completed")


if __name__ == "__main__":
    asyncio.run(main())