Is there a way to map build steps to kubernetes pod/container?

I’m trying to write a little janitor to make sure hanging builds are restartet etc, since the kubernetes support in Drone-CI is a bit unstable. However, I’m struggling to find something to match build steps to the individual containers in the build pod, other than relying on their order (so pod 2 = step 2 etc). Is there any way for me to map build steps to containers with more certainty?