Add back `actionID` to `stateKeys()`

[RodrigoVillar]

As per the conversation linked, we most likely want to add back actionID argument to the stateKeys() interface.

This change would probably make sense to work on after #1644 is merged