fix: align ML service response schemas

This commit is contained in:
Asep Haryana Saputra
2026-05-23 09:36:20 +00:00
parent dce4d4f21c
commit ee872c644f
2 changed files with 7 additions and 33 deletions
-31
View File
@@ -17,17 +17,6 @@ pub struct MetadataResponse {
pub labels: Vec<String>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct PredictionResponse {
pub predictions: Vec<PredictionResult>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct PredictionResult {
pub label: String,
pub confidence: f32,
}
pub fn health_response(model_loaded: bool) -> HealthResponse {
HealthResponse {
status: "ok".to_string(),
@@ -85,24 +74,4 @@ mod tests {
vec!["Bercak Daun", "Daun Sehat", "Karat Daun", "Hawar Daun"]
);
}
#[test]
fn prediction_response_serializes() {
let response = PredictionResponse {
predictions: vec![
PredictionResult {
label: "Bercak Daun".to_string(),
confidence: 0.95,
},
PredictionResult {
label: "Daun Sehat".to_string(),
confidence: 0.05,
},
],
};
let json = serde_json::to_string(&response).expect("failed to serialize");
assert!(json.contains("Bercak Daun"));
assert!(json.contains("0.95"));
}
}