ideally we'll attach this to the pydantic model as well
ideally we'll attach this to the pydantic model as well