Demos / backend /models /rag_models.py
nikhile-galileo's picture
Adding finance protect demo
e68d535
raw
history blame
161 Bytes
from pydantic import BaseModel
class RAGRequest(BaseModel):
query: str
protect_enabled: bool = False
class RAGResponse(BaseModel):
response: str