Fine-tuning/backend/app/schemas/sample_center.py

from pydantic import BaseModel


class KnowledgeBaseItem(BaseModel):
    id: str
    name: str
    parent_table: str
    child_table: str
    document_count: int
    status: int
    created_at: str
    created_by: str
    metadata_schema: list[dict] = []


class KnowledgeBaseListResponse(BaseModel):
    total: int
    page: int
    page_size: int
    items: list[KnowledgeBaseItem]


class KnowledgeBaseDetailResponse(BaseModel):
    id: str
    name: str
    description: str = ""
    parent_table: str
    child_table: str
    document_count: int
    status: int
    created_at: str
    updated_at: str = ""
    created_by: str
    metadata_schema: list[dict] = []


class ImportTaskResponse(BaseModel):
    task_id: str
    status: str


class KbImportResponse(BaseModel):
    kb_id: str
    kb_name: str
    document_count: int
    metadata_schema: list[dict] = []
    parent_table: str
    child_table: str