From ca9cbb12bfb015dcdcd5963c225ec14c9d2bcee0 Mon Sep 17 00:00:00 2001
From: Asep Haryana Saputra <90584806+MythEclipse@users.noreply.github.com>
Date: Fri, 22 May 2026 19:28:43 +0000
Subject: [PATCH] feat: add ML service response schemas

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
---
 apps/ml-service/schemas.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 apps/ml-service/schemas.py

diff --git a/apps/ml-service/schemas.py b/apps/ml-service/schemas.py
new file mode 100644
index 0000000..4ef3ccd
--- /dev/null
+++ b/apps/ml-service/schemas.py
@@ -0,0 +1,21 @@
+from pydantic import BaseModel
+
+
+class HealthResponse(BaseModel):
+    status: str
+    model_loaded: bool
+
+
+class MetadataResponse(BaseModel):
+    service_name: str
+    service_version: str
+    model_path: str
+    model_loaded: bool
+    input_size: int
+    labels: list[str]
+
+
+class PredictionResponse(BaseModel):
+    label: str
+    confidence: float
+    probabilities: dict[str, float]