Update models.py

models.py

@@ -1,6 +1,7 @@
 from pydantic import BaseModel, EmailStr, Field, validator
 from typing import List, Optional
 
+# === Trainer Models ===
 class InitializeBotResponse(BaseModel):
     bot_id: str
 
@@ -19,10 +20,10 @@ class NewChatResponse(BaseModel):
     chat_id: str
 
 class QueryRequest(BaseModel):
+    bot_id: str
     chat_id: str
     query: str
 
-
 class QueryResponse(BaseModel):
     response: str
     web_sources: List[str]