I want to define custom/user defined partition key to the records in AWS DMS before pushing them to AWS Kinesis.

I have a database, which has 40 tables and I am using AWS DMS to fetch the data changes from those tables and push into AWS KINESIS stream which has 2 shards in it and eventually, my two consumer applications will read the data from the Kinesis and update the respective databases with the help of custom logic.