src/Response.py

from pydantic import BaseModel


class ChatResponse(BaseModel):
    response: str


class FileResponse(BaseModel):
    filename: str
    role_name_list: list[str] = []
    role_name_dict: dict[str, str] = {}
    md5: str = None