How can I get the episode info dict from the on_sample_end callback?

Question

I need to get the episode info dict from the on_sample_end callback so that I can display some custom metrics every time a rollout has finished. How can I do that?