Add typevar to trace to remove typing errors in VS code

[tbonus]

This change removes Union ambiguity from decorated function and retains information about async and returned types

[erik-megarad]

Awesome, thank you!