from pydantic import BaseModel, Field


class PdfTextExtractRequest(BaseModel):
    dataset_id: str = Field(..., alias="datasetId", description="Dataset ID")
    file_id: str = Field(..., alias="fileId", description="PDF file ID")

    class Config:
        populate_by_name = True


class PdfTextExtractResponse(BaseModel):
    dataset_id: str = Field(..., alias="datasetId", description="Dataset ID")
    source_file_id: str = Field(..., alias="sourceFileId", description="Source PDF file ID")
    text_file_id: str = Field(..., alias="textFileId", description="Generated text file ID")
    text_file_name: str = Field(..., alias="textFileName", description="Generated text file name")
    text_file_path: str = Field(..., alias="textFilePath", description="Generated text file path")
    text_file_size: int = Field(..., alias="textFileSize", description="Generated text file size")

    class Config:
        populate_by_name = True