ModelFinder

ModelFinder.cs

public class ModelFinder : IModelFinder
{
    private readonly BehaviorGraph _graph;

    public ModelFinder(BehaviorGraph graph)
    {
        _graph = graph;
    }

    public Type FindRelatedModel(Type type)
    {
        var post = _graph
            .BehaviorFor(type)
            .FirstCall();

        // The handler Type is the CreateEndpoint so grab the Get method
        var getModelType = post
            .HandlerType
            .GetMethod("Get", BindingFlags.Public | BindingFlags.Instance)
            .GetParameters()[0]
            .ParameterType;

        return getModelType;
    }
}