from enum import Enum

from pydantic import BaseModel, Field


class DatasetFormat(str, Enum):
    JSONL = "jsonl"
    CSV = "csv"
    PARQUET = "parquet"
    JSON = "json"


class DatasetUploadResponse(BaseModel):
    id: str
    name: str
    format: DatasetFormat
    record_count: int
    file_path: str
    created_at: str


class DatasetPreviewRow(BaseModel):
    row_index: int
    data: dict


class DatasetPreviewResponse(BaseModel):
    total_records: int
    preview_rows: list[DatasetPreviewRow]
    columns: list[str]


class DatasetValidationResult(BaseModel):
    is_valid: bool
    errors: list[str] = []
    warnings: list[str] = []