...

• The code changes can be seen here in the open PR

Rejected alternatives

• It was mentioned to fix the parameter "max-messsage-size" (a different feature) as it is not working properly, it was discarded due to the potential confusion with the current implementation, see more here.
• Use number of batches instead of byes, you can see why this was discarded here

...