Update main.py

main.py

@@ -497,6 +497,14 @@ async def chat(request: ChatRequest):
         
         # Save SINGLE chat directly to MongoDB
         save_chat_to_db(request.space_id, chat)
+        
+        # ADD THIS RETURN BLOCK:
+        return {
+            "chat_id": chat_id,
+            "response": response,
+            "sources": sources,
+            "timestamp": timestamp
+        }
     
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))