Add support for tracing functions which return `Awaitable`s.