Update models.py

models.py

@@ -17,7 +17,8 @@ class QueryRequest(BaseModel):
     bot_id: str
     chat_id: str
     query: str
-
+    prompt_type: str
+    
 class QueryResponse(BaseModel):
     response: str
     web_sources: List[str]