Add option to use a callback to retrieve a model or collection from the component’s props