from pydantic import BaseModel, Field


class RagDatasetItemVO(BaseModel):
    id: int = Field(...)
    name: str = Field(...)
    description: str = Field("")
    area: str = Field("")
    isPublic: bool = Field(False)
    isDefault: bool = Field(False)
    documentCount: int = Field(0)
    totalChunks: int = Field(0)
    status: int = Field(1)


class RagDatasetPageVO(BaseModel):
    data: list[RagDatasetItemVO] = Field(default_factory=list)
    total: int = Field(0)
