It's been a while since AWS Step Functions adopts JSONata as the default query language on its console. However, there is still no update in this SDK to catch up the latest feature of AWS Step Functions. There seems to be no workaround as well, except writing the whole definition for the state machine, which is inconvenient.