Yassine Mhirsi
Add label/KPA-related schemas to models and update documentation in label routes
3afc11b
| """Pydantic models and schemas for request/response validation""" | |
| # Import stance-related schemas | |
| from .stance import ( | |
| StanceRequest, | |
| StanceResponse, | |
| BatchStanceRequest, | |
| BatchStanceResponse, | |
| ) | |
| # Import label/KPA-related schemas | |
| from .label import ( | |
| PredictionRequest, | |
| PredictionResponse, | |
| BatchPredictionRequest, | |
| BatchPredictionResponse, | |
| HealthResponse as KPAHealthResponse, | |
| ) | |
| # Import health-related schemas | |
| from .health import ( | |
| HealthResponse, | |
| ) | |
| __all__ = [ | |
| # Stance schemas | |
| "StanceRequest", | |
| "StanceResponse", | |
| "BatchStanceRequest", | |
| "BatchStanceResponse", | |
| # Label/KPA schemas | |
| "PredictionRequest", | |
| "PredictionResponse", | |
| "BatchPredictionRequest", | |
| "BatchPredictionResponse", | |
| "KPAHealthResponse", | |
| # Health schemas | |
| "HealthResponse", | |
| ] | |