"""
API models for request/response validation.
"""

from .api_models import (
    AnalyzeDatasetRequest,
    BuildDatasetRequest,
    CurateDatasetRequest,
    DatasetAnalysisResponse,
    DatasetValidationResponse,
    DeviceType,
    DownloadDatasetRequest,
    DownloadDatasetResponse,
    EvaluateBAAgreementRequest,
    HealthResponse,
    JobResponse,
    JobStatus,
    ModelsResponse,
    PretrainRequest,
    TrainRequest,
    TrainUnifiedRequest,
    UploadDatasetRequest,
    UploadDatasetResponse,
    UseCase,
    ValidateARKitRequest,
    ValidateDatasetRequest,
    ValidateSequenceRequest,
    ValidationStats,
    VisualizeRequest,
)
from .intermediate_artifacts import (
    ArraySequenceRef,
    ArtifactURI,
    AuditArtifactBundle,
    CalibrationParams,
    InferenceArtifactBundle,
    IntermediateSchemaVersion,
    LandmarkRef,
    LandmarkSet,
    PoseRef,
    PoseSet,
    Provenance,
    TeacherArtifactBundle,
    TrackObservation,
    TrackRef,
    TrackSet,
    Units,
)
from .run_models import RunError, RunResult, Stage

# Alias for backward compatibility
EvaluateRequest = EvaluateBAAgreementRequest

__all__ = [
    "AnalyzeDatasetRequest",
    "BuildDatasetRequest",
    "CurateDatasetRequest",
    "DatasetAnalysisResponse",
    "DatasetValidationResponse",
    "DeviceType",
    "DownloadDatasetRequest",
    "DownloadDatasetResponse",
    "EvaluateBAAgreementRequest",
    "EvaluateRequest",
    "HealthResponse",
    "JobResponse",
    "JobStatus",
    "ModelsResponse",
    "PretrainRequest",
    "TrainRequest",
    "TrainUnifiedRequest",
    "UploadDatasetRequest",
    "UploadDatasetResponse",
    "UseCase",
    "ValidateARKitRequest",
    "ValidateDatasetRequest",
    "ValidateSequenceRequest",
    "ValidationStats",
    "VisualizeRequest",
    # Intermediate artifacts (Phase 0 contract)
    "ArraySequenceRef",
    "AuditArtifactBundle",
    "ArtifactURI",
    "CalibrationParams",
    "InferenceArtifactBundle",
    "IntermediateSchemaVersion",
    "LandmarkRef",
    "LandmarkSet",
    "PoseRef",
    "PoseSet",
    "Provenance",
    "TeacherArtifactBundle",
    "TrackObservation",
    "TrackRef",
    "TrackSet",
    "Units",
    # Run contracts
    "RunError",
    "RunResult",
    "Stage",
]