"""
知识库文件模型
"""
from datetime import datetime
from typing import Optional
from pydantic import BaseModel, Field


class KnowledgeBaseFile(BaseModel):
    """知识库文件模型"""
    id: Optional[int] = None
    knowledge_base_id: int
    user_id: int
    file_name: str = Field(..., max_length=255)
    file_path: str = Field(..., max_length=500)
    file_size: int
    file_type: str = Field(default="pdf", max_length=50)
    status: str = Field(default="processing", max_length=20)
    chunk_count: int = 0
    created_at: Optional[datetime] = None
    updated_at: Optional[datetime] = None
    is_deleted: bool = False
    deleted_at: Optional[datetime] = None

    class Config:
        from_attributes = True


class KnowledgeBaseChunk(BaseModel):
    """知识库文档块模型"""
    id: Optional[int] = None
    file_id: int
    knowledge_base_id: int
    chunk_index: int
    content: str
    metadata: Optional[dict] = None
    vector_id: Optional[str] = None
    created_at: Optional[datetime] = None

    class Config:
        from_attributes = True


class FileUploadResponse(BaseModel):
    """文件上传响应模型"""
    id: int
    file_name: str
    file_size: int
    status: str
    chunk_count: int
    created_at: datetime
    file_url: Optional[str] = Field(None, description="文件访问 URL（OSS 或本地路径）")

    class Config:
        from_attributes = True


class FileListResponse(BaseModel):
    """文件列表响应模型"""
    total: int = Field(..., description="总数量")
    items: list[FileUploadResponse] = Field(..., description="文件列表")

    class Config:
        from_attributes = True