def METHOD_NAME(self, client_info):
    # Precompute the wrapped methods.
    self._wrapped_methods = {
        self.generate_message: gapic_v1.method.wrap_method(
            self.generate_message,
            default_timeout=None,
            client_info=client_info,
        ),
        self.count_message_tokens: gapic_v1.method.wrap_method(
            self.count_message_tokens,
            default_timeout=None,
            client_info=client_info,
        ),
    }