Now we pass `collect_traces` directly to `contextmanager` function at end of file to avoid return type problems.

To fix type hint errors we separate all the possible `collect_traces` function signatures using `@overload` decorator
The implementation of `collect_traces`, now, acts like a factory!

…tead of passing `factory` through the `contextmanager` function

…collect_traces_context_manager_return_type_two` case