Add type annotations for the `log_function` decorator.