Update app.py

app.py

@@ -167,11 +167,22 @@ async def fast_ai(user: str, model: str = "llama3-70b", system: str = "Answer as
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error during Snova AI request: {e}")
 
+from pydantic import BaseModel
+import asyncio
+
+app = FastAPI()
+
+# Define a Pydantic model for the request payload
+class FastAIRequest(BaseModel):
+    user: str
+    model: str = "llama3-70b"
+    system: str = "Answer as concisely as possible."
+
 @app.post("/api/fastAI-post")
-async def fast_ai(user: str, model: str = "llama3-70b", system: str = "Answer as concisely as possible."):
+async def fast_ai(request: FastAIRequest):
     """Get a response from the Snova AI service."""
     try:
-        response = await asyncio.to_thread(fastai, user, model, system)
+        response = await asyncio.to_thread(fastai, request.user, request.model, request.system)
         return JSONResponse(content={"response": response})
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error during Snova AI request: {e}")