Revert for Agent and Assistant

When i give an Agent or an assistant a task i can only revert to right after they made some change.

It would be really helpful if i could just revert the whole session.

The cut off for one message in seems like the wrong place, I think the logical cut of is the start.