Specifying and viewing Custom task run state Message

We are trying to update the state message for a running task with some custom information(e.g. it’s ETA computed based on our past runtime statistics, or some details on the hardware specifications the task is using). Previously in 1.0 this is done by state handlers. Now in 2.0 we were able to get the same result by posting a new running state within the task function, but this leads to two running states and double counting the run count. Is there a better way to do this? It would be great if there is a way to only update the message of a State without posting a new one.

Also, since in the Prefect UI flow run state message is displayed, could the same thing be added to the task run page?