Add type annotations for producer