Add new function to get traffic

gateway/app/routes/training.py

@@ -13,38 +13,6 @@ from app.core.config import settings
 logger = logging.getLogger(__name__)
 router = APIRouter()
 
-@router.post("/train")
-async def start_training(request: Request):
-    """Proxy training request to training service"""
-    try:
-        body = await request.body()
-        auth_header = request.headers.get("Authorization")
-        
-        async with httpx.AsyncClient(timeout=30.0) as client:
-            response = await client.post(
-                f"{settings.TRAINING_SERVICE_URL}/train",
-                content=body,
-                headers={
-                    "Content-Type": "application/json",
-                    "Authorization": auth_header
-                }
-            )
-            
-            return JSONResponse(
-                status_code=response.status_code,
-                content=response.json()
-            )
-            
-    except httpx.RequestError as e:
-        logger.error(f"Training service unavailable: {e}")
-        raise HTTPException(
-            status_code=503,
-            detail="Training service unavailable"
-        )
-    except Exception as e:
-        logger.error(f"Training error: {e}")
-        raise HTTPException(status_code=500, detail="Internal server error")
-
 @router.get("/status/{training_job_id}")
 async def get_training_status(training_job_id: str, request: Request):
     """Get training job status"""
@@ -130,3 +98,35 @@ async def get_training_jobs(
     except Exception as e:
         logger.error(f"Get training jobs error: {e}")
         raise HTTPException(status_code=500, detail="Internal server error")
+
+@router.post("/jobs")
+async def start_training_job(request: Request):
+    """Start a new training job - Proxy to training service"""
+    try:
+        body = await request.body()
+        auth_header = request.headers.get("Authorization")
+        
+        async with httpx.AsyncClient(timeout=30.0) as client:
+            response = await client.post(
+                f"{settings.TRAINING_SERVICE_URL}/training/jobs",  # Correct path
+                content=body,
+                headers={
+                    "Content-Type": "application/json",
+                    "Authorization": auth_header
+                }
+            )
+            
+            return JSONResponse(
+                status_code=response.status_code,
+                content=response.json()
+            )
+            
+    except httpx.RequestError as e:
+        logger.error(f"Training service unavailable: {e}")
+        raise HTTPException(
+            status_code=503,
+            detail="Training service unavailable"
+        )
+    except Exception as e:
+        logger.error(f"Start training job error: {e}")
+        raise HTTPException(status_code=500, detail="Internal server error")