"""
Admin Training API Routes

FastAPI endpoints for training task management and scheduling.
"""

from fastapi import APIRouter

from ._utils import _validate_uuid
from .tasks import register_task_routes
from .documents import register_document_routes
from .export import register_export_routes
from .datasets import register_dataset_routes
from .models import register_model_routes
from .pool import register_pool_routes


def create_training_router() -> APIRouter:
    """Create training API router."""
    router = APIRouter(prefix="/admin/training", tags=["Admin Training"])

    register_task_routes(router)
    register_document_routes(router)
    register_export_routes(router)
    register_dataset_routes(router)
    register_model_routes(router)
    register_pool_routes(router)

    return router


__all__ = ["create_training_router", "_validate_uuid"]
