Add support for including a partition key with text messages