api/schema.py

from pydantic import BaseModel


class PredictRequest(BaseModel):
	distance_km: float


class PredictResponse(BaseModel):
	cost_soles: float