from pydantic import BaseModel, Field


class RagChatAttachmentVO(BaseModel):
    attachmentId: str = Field(..., description="临时附件ID")
    conversationId: str = Field(..., description="会话ID")
    fileName: str = Field(..., description="原始文件名")
    contentType: str = Field("", description="文件 MIME 类型")
    fileSize: int = Field(0, description="文件大小")
    indexingStatus: str = Field("waiting", description="索引状态")
    indexingError: str | None = Field(None, description="索引错误")
    chunkCount: int = Field(0, description="分段数量")
    collectionName: str = Field("", description="临时向量集合名")
    expiresAt: int = Field(0, description="过期时间戳")
    createdAt: int = Field(0, description="创建时间戳")


class RagChatAttachmentDeleteVO(BaseModel):
    result: str = Field("success")