from pydantic import BaseModel, Field


class PdfTextExtractRequest(BaseModel):
    dataset_id: str = Field(..., alias="datasetId", description="数据集ID")
    file_id: str = Field(..., alias="fileId", description="源文件ID")

    class Config:
        populate_by_name = True


class PdfTextExtractResponse(BaseModel):
    dataset_id: str = Field(..., alias="datasetId", description="数据集ID")
    source_file_id: str = Field(..., alias="sourceFileId", description="源文件ID")
    text_file_id: str = Field(..., alias="textFileId", description="解析后的文本文件ID")
    text_file_name: str = Field(..., alias="textFileName", description="解析后的文本文件名")
    text_file_path: str = Field(..., alias="textFilePath", description="解析后的文本文件路径")
    text_file_size: int = Field(..., alias="textFileSize", description="解析后的文本文件大小")

    class Config:
        populate_by_name = True
