models/requests.py

from typing import Dict
from pydantic import BaseModel


class PredictRequest(BaseModel):
    """Request model for /predict endpoint.

    Contains a mapping of feature names to numeric values that the model
    will consume. Keep it minimal so imports from `models.requests`
    succeed when the app starts.
    """
    features: Dict[str, float]