setup assisant bot

api/models/requests.py

@@ -0,0 +1,15 @@
+# api/models/requests.py
+from pydantic import BaseModel
+from typing import List
+
+class BaseRequest(BaseModel):
+    pass
+
+class ChatMessage(BaseModel):
+    role: str  # "human" or "ai"
+    content: str
+
+
+class ChatRequest(BaseModel):
+    query: str
+    history: List[ChatMessage] = []

api/models/responses.py

@@ -0,0 +1,8 @@
+# api/models/responses.py
+from pydantic import BaseModel
+
+
+class ChatResponse(BaseModel):
+    status: str
+    message: str
+