Add type hints in worker template

We should add type hints in the worker template