REFACTOR ALL APIs fix 1

services/training/app/api/training_operations.py

@@ -478,6 +478,18 @@ async def training_progress_websocket(
     await connection_manager.connect(websocket, job_id, connection_id)
     logger.info(f"WebSocket connection established for job {job_id}, user {user_id}")
 
+    # Send immediate connection confirmation to prevent gateway timeout
+    try:
+        await websocket.send_json({
+            "type": "connected",
+            "job_id": job_id,
+            "message": "WebSocket connection established",
+            "timestamp": str(datetime.now())
+        })
+        logger.debug(f"Sent connection confirmation for job {job_id}")
+    except Exception as e:
+        logger.error(f"Failed to send connection confirmation for job {job_id}: {e}")
+
     consumer_task = None
     training_completed = False