Fix to use new ThreadSafeContext API

Turn `add_output` into a manual co-routine
so that we can get out of the caller scope mid-call
to release the threadsafecontext lock early.

Author: yuyichao