We should maybe validate the function args that are returned by the model using pydantic.
We should maybe validate the function args that are returned by the model using pydantic.