Update app/routers/chat.py

app/routers/chat.py

@@ -0,0 +1,35 @@
+from fastapi import APIRouter
+from fastapi import Depends
+from fastapi import HTTPException
+
+from app.dependencies import get_assistant
+from app.schemas.chat import ChatRequest
+from app.schemas.chat import ChatResponse
+
+router = APIRouter()
+
+
+@router.post("/", response_model=ChatResponse)
+async def chat(
+    request: ChatRequest,
+    assistant=Depends(get_assistant)
+):
+
+    try:
+
+        response, sources, contexts = (
+            assistant.answer(request.message)
+        )
+
+        return ChatResponse(
+            response=response,
+            sources=sources,
+            contexts=contexts
+        )
+
+    except Exception as e:
+
+        raise HTTPException(
+            status_code=500,
+            detail=str(e)
+        )