Update app.py

app.py

@@ -21,7 +21,7 @@ class ChatRequest(BaseModel):
 
 class ChatResponse(BaseModel):
     model_status: str
-    response: dict
+    response: str
     response_time: float
 
 @app.post("/chat", response_model=ChatResponse)
@@ -34,16 +34,13 @@ async def chat(request: ChatRequest):
             {'role': 'user', 'content': request.message}
         ]
 
-        # Simulate getting the response from the model
         response = ollama.chat(model=model_name, messages=messages)
-        response_dict = vars(response)
-        assistant_response = response_dict["message"]["content"]
-        # response = "pending"
+        response = str(response)
 
         end_time = time.time()
         response_time = end_time - start_time
 
-        return {"model_status": model.status, "response_time": response_time, "response": assistant_response}
+        return {"model_status": model.status, "response_time": response_time, "response": response}
         
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))