Hardik commited on
Commit
84a9f0b
·
1 Parent(s): d939d85

Include prediction errors in response for debugging

Browse files
Files changed (1) hide show
  1. ml_service/app/api/routes.py +7 -1
ml_service/app/api/routes.py CHANGED
@@ -64,7 +64,13 @@ async def predict(request: PredictRequest):
64
  models=models_result,
65
  timing=TimingInfo(total_ms=total_ms),
66
  metadata=PredictMetadata(model_versions=MODEL_VERSIONS)
67
- )
 
 
 
 
 
 
68
 
69
  @router.get("/health")
70
  async def health_check() -> Dict[str, Any]:
 
64
  models=models_result,
65
  timing=TimingInfo(total_ms=total_ms),
66
  metadata=PredictMetadata(model_versions=MODEL_VERSIONS)
67
+ ) if not errors else {
68
+ "overall": OverallPrediction(label=majority_label, agreement=agreement).model_dump(),
69
+ "models": {k: v.model_dump() for k, v in models_result.items()},
70
+ "timing": TimingInfo(total_ms=total_ms).model_dump(),
71
+ "metadata": PredictMetadata(model_versions=MODEL_VERSIONS).model_dump(),
72
+ "errors": errors,
73
+ }
74
 
75
  @router.get("/health")
76
  async def health_check() -> Dict[str, Any]: