from pydantic import BaseModel
from typing import Optional
from datetime import datetime

class KbBase(BaseModel):
    name: str
    description: Optional[str] = None

class KbCreate(KbBase):
    pass

class KbUpdate(KbBase):
    pass

class Kb(KbBase):
    id: int
    create_time: datetime
    update_time: datetime

    class Config:
        orm_mode = True

class KbFileBase(BaseModel):
    kb_id: int
    file_name: str
    file_path: str
    file_size: int
    file_type: str
    state: int = 0

class KbFileCreate(KbFileBase):
    pass

class KbFileUpdate(KbFileBase):
    pass

class KbFile(KbFileBase):
    id: int
    create_time: datetime
    update_time: datetime

    class Config:
        orm_mode = True

class KbModel(BaseModel):
    kb_name: str
    short_name: str
    is_delete: Optional[int] = 0

class KbFileModel(BaseModel):
    file_name: str
    ext_name: str
    kb_id: int
    is_delete: Optional[int] = 0
    state: Optional[int] = 0