Logging / probes are specified only for by generation methods

[lukepmccombs]

Every probe specifically checks for and uses the generation key. This makes them unusable for steady_state, as it uses the birth counter instead.

[lukepmccombs]

I'm making a fix for this by creating a util function that does all of the checks for generation and birth in the context and returns whichever is appropriate, then swapping out all usages with that.