HUE-2330 [beeswax] Collapse previously expanded tables on assist

Added support for expanding/collapsing on all the Assists