Cosmos: Review tracing span handling across await boundaries

A `tracing` span can be held across await boundaries, but it requires some [special handling](https://docs.rs/tracing/latest/tracing/trait.Instrument.html#method.instrument). We haven't done that yet and should review the driver code to ensure its handling spans correctly.