Cómo usar Async y Await con AWS SDK Javascript

Estoy trabajando con el SDK de AWS con la biblioteca KMS. Me gustaría usar async y esperar en lugar de devoluciones de llamada.

import AWS, { KMS } from "aws-sdk";

this.kms = new AWS.KMS();

const key = await this.kms.generateDataKey();

Sin embargo, esto no funciona, cuando se envuelve en una función asíncrona.

¿Cómo puedo usar async y esperar aquí?