updated bot chat api

api/models/responses.py

@@ -3,10 +3,8 @@ from pydantic import BaseModel
 from typing import Dict
 class ChatResponse(BaseModel):
     status: str
-    message: str
+    message: Dict
    
-
-
 class SurveyAgentResponse(BaseModel):
     status: str
     result: Dict

api/routes/chat.py

@@ -25,13 +25,12 @@ async def chat_ai(
         
         # Initialize DroneBot with history
         bot = DroneBot(history=history, use_openai_as_fallback=True)
-        
-        # Get response
         result = bot.chat(request.query)
-
+        message = json.loads(result["final_message"])
+        print(result)
         return ChatResponse(
             status="success",
-            message=result["final_message"]
+            message=message
         )
     except Exception as e:
         raise HTTPException(