schema/apiSchema.py

"""
Python Backend API to chat with private data  

08/14/2023
D.M. Theekshana Samaradiwakara
"""

from typing import Optional, List, Any, Dict
from pydantic import BaseModel


class Document(BaseModel):
    name: Optional[str]
    page_content: str
    metadata: Dict[str, Any]


class QueryModel(BaseModel):
    model: str
    dataset: str
    question: str
    history: list = None


class ResponseModel(BaseModel):
    success: str = None
    error: str = None
    documents: List[Document] # = None