helper/model_load.py

from pydantic import BaseModel
from typing import Optional, List

class UploadDocRequest(BaseModel):
    doc_type: str
    content: Optional[str] = None
    file_name: Optional[str] = None

class ChatRequest(BaseModel):
    query: str
    collection: str

class ChatResponse(BaseModel):
    answer: str
    sources: Optional[List[str]] = []