Spaces:
Sleeping
Sleeping
Hardik commited on
Commit ·
84a9f0b
1
Parent(s): d939d85
Include prediction errors in response for debugging
Browse files
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]:
|