from typing import Optional

from pydantic import BaseModel


class PreviewDetail(BaseModel):
    content: str
    child_chunks: Optional[list[str]] = None


class QAPreviewDetail(BaseModel):
    question: str
    answer: str


class IndexingEstimate(BaseModel):
    total_segments: int
    preview: list[PreviewDetail]
    qa_preview: Optional[list[QAPreviewDetail]] = None


class PipelineDataset(BaseModel):
    id: str
    name: str
    description: str
    chunk_structure: str


class PipelineDocument(BaseModel):
    id: str
    position: int
    data_source_type: str
    data_source_info: Optional[dict] = None
    name: str
    indexing_status: str
    error: Optional[str] = None
    enabled: bool


class PipelineGenerateResponse(BaseModel):
    batch: str
    dataset: PipelineDataset
    documents: list[PipelineDocument]