Add label/KPA-related schemas to models and update documentation in label routes

models/__init__.py

@@ -8,6 +8,15 @@ from .stance import (
     BatchStanceResponse,
 )
 
+# Import label/KPA-related schemas
+from .label import (
+    PredictionRequest,
+    PredictionResponse,
+    BatchPredictionRequest,
+    BatchPredictionResponse,
+    HealthResponse as KPAHealthResponse,
+)
+
 # Import health-related schemas
 from .health import (
     HealthResponse,
@@ -19,6 +28,12 @@ __all__ = [
     "StanceResponse",
     "BatchStanceRequest",
     "BatchStanceResponse",
+    # Label/KPA schemas
+    "PredictionRequest",
+    "PredictionResponse",
+    "BatchPredictionRequest",
+    "BatchPredictionResponse",
+    "KPAHealthResponse",
     # Health schemas
     "HealthResponse",
 ]

routes/label.py

@@ -77,7 +77,7 @@ async def predict_kpa(request: PredictionRequest):
 @router.post("/batch-predict", response_model=BatchPredictionResponse, tags=["KPA"])
 async def batch_predict_kpa(request: BatchPredictionRequest):
     """
-    Predict keypoint–argument matching for multiple argument/keypoint pairs.
+    Predict keypoint-argument matching for multiple argument/keypoint pairs.
 
     - **pairs**: List of items to classify