File size: 161 Bytes
e68d535
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
from pydantic import BaseModel


class RAGRequest(BaseModel):
    query: str
    protect_enabled: bool = False


class RAGResponse(BaseModel):
    response: str