Spaces:
Sleeping
Sleeping
File size: 1,147 Bytes
72e2b6e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | from pydantic import BaseModel, Field
class PredictRequest(BaseModel):
text: str = Field(..., min_length=1, max_length=500)
model_type: str | None = None
class TopIntent(BaseModel):
intent: str
confidence: float
class PredictResponse(BaseModel):
intent: str
confidence: float
top5: list[TopIntent]
latency_ms: float
is_oos: bool
model_used: str
ab_variant: str | None = None
class BatchRequest(BaseModel):
texts: list[str] = Field(..., min_length=1, max_length=100)
model_type: str | None = None
class BatchResponse(BaseModel):
predictions: list[PredictResponse]
total_latency_ms: float
class ModelInfo(BaseModel):
name: str
alias: str
version: str | None = None
class ModelsResponse(BaseModel):
models: list[ModelInfo]
class HealthResponse(BaseModel):
status: str
models_loaded: list[str]
class DriftResponse(BaseModel):
drift_summary: dict
confidence_drift: dict
oos_rate_drift: dict
class ABStatsResponse(BaseModel):
model_a: dict
model_b: dict
split: float
|