Hey
@Luke Diliberto - I filed an issue to track this:
https://github.com/dagster-io/dagster/issues/4607. I think it would be cool to add.
I'm not sure on when we'll be able to get to it - if you happen to have time to work on it, I'd be happy to assist.