update

app.py

@@ -1,30 +1,17 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import Response
-from pydantic import BaseModel
 import ollama
+
 app = FastAPI()
 
 @app.get('/')
 def home():
     return "Hello world"
 
-# @app.post("/chat")
-# async def chat(query: str):
-#     if not query:
-#         return {"error": "Message cannot be empty."}
-
-#     response = ollama.chat(model="llama3", messages=[{"role": "user", "content": query}])
-#     return {"response": response['message']['content']}
-
-
-
-class ChatRequest(BaseModel):
-    query: str
-
 @app.post("/chat")
-async def chat(request: ChatRequest):
-    if not request.query.strip():
-        raise HTTPException(status_code=400, detail="Message cannot be empty.")
+async def chat(query: str):
+    if not query:
+        return {"error": "Message cannot be empty."}
 
-    response = ollama.chat(model="llama3", messages=[{"role": "user", "content": request.query}])
+    response = ollama.chat(model="llama3", messages=[{"role": "user", "content": query}])
     return {"response": response['message']['content']}