100 lines
3.8 KiB
Python
100 lines
3.8 KiB
Python
"""
|
|
Training routes for gateway - FIXED VERSION
|
|
"""
|
|
|
|
from fastapi import APIRouter, Request, HTTPException, Query, Response
|
|
from fastapi.responses import JSONResponse
|
|
import httpx
|
|
import logging
|
|
from typing import Optional
|
|
|
|
from app.core.config import settings
|
|
|
|
logger = logging.getLogger(__name__)
|
|
router = APIRouter()
|
|
|
|
async def _proxy_training_request(request: Request, target_path: str, method: str = None):
|
|
"""Proxy request to training service with user context"""
|
|
|
|
# Handle OPTIONS requests directly for CORS
|
|
if request.method == "OPTIONS":
|
|
return Response(
|
|
status_code=200,
|
|
headers={
|
|
"Access-Control-Allow-Origin": settings.CORS_ORIGINS_LIST,
|
|
"Access-Control-Allow-Methods": "GET, POST, PUT, DELETE, OPTIONS",
|
|
"Access-Control-Allow-Headers": "Content-Type, Authorization, X-Tenant-ID",
|
|
"Access-Control-Allow-Credentials": "true",
|
|
"Access-Control-Max-Age": "86400" # Cache preflight for 24 hours
|
|
}
|
|
)
|
|
|
|
try:
|
|
url = f"{settings.TRAINING_SERVICE_URL}{target_path}"
|
|
|
|
# Forward headers AND add user context from gateway auth
|
|
headers = dict(request.headers)
|
|
headers.pop("host", None) # Remove host header
|
|
|
|
# ✅ ADD USER CONTEXT FROM GATEWAY AUTHENTICATION
|
|
# Gateway middleware already verified the token and added user to request.state
|
|
if hasattr(request.state, 'user'):
|
|
headers["X-User-ID"] = str(request.state.user.get("user_id"))
|
|
headers["X-User-Email"] = request.state.user.get("email", "")
|
|
headers["X-Tenant-ID"] = str(request.state.user.get("tenant_id"))
|
|
headers["X-User-Roles"] = ",".join(request.state.user.get("roles", []))
|
|
headers["X-User-Permissions"] = ",".join(request.state.user.get("permissions", []))
|
|
|
|
# Get request body if present
|
|
body = None
|
|
request_method = method or request.method
|
|
if request_method in ["POST", "PUT", "PATCH"]:
|
|
body = await request.body()
|
|
|
|
async with httpx.AsyncClient(timeout=30.0) as client:
|
|
response = await client.request(
|
|
method=request_method,
|
|
url=url,
|
|
headers=headers,
|
|
content=body,
|
|
params=dict(request.query_params)
|
|
)
|
|
|
|
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 service 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"""
|
|
return await _proxy_training_request(request, f"/training/status/{training_job_id}", "GET")
|
|
|
|
@router.get("/models")
|
|
async def get_trained_models(request: Request):
|
|
"""Get trained models"""
|
|
return await _proxy_training_request(request, "/training/models", "GET")
|
|
|
|
@router.get("/jobs")
|
|
async def get_training_jobs(
|
|
request: Request,
|
|
limit: Optional[int] = Query(10, ge=1, le=100),
|
|
offset: Optional[int] = Query(0, ge=0)
|
|
):
|
|
"""Get training jobs"""
|
|
return await _proxy_training_request(request, f"/training/jobs?limit={limit}&offset={offset}", "GET")
|
|
|
|
@router.post("/jobs")
|
|
async def start_training_job(request: Request):
|
|
"""Start a new training job - Proxy to training service"""
|
|
return await _proxy_training_request(request, "/training/jobs", "POST") |