Update models.py

models.py

@@ -17,6 +17,7 @@ class QueryRequest(BaseModel):
     bot_id: str
     chat_id: str
     query: str
+    prompt_type: str
 
 class QueryResponse(BaseModel):
     response: str