[Backport 2.12] Fix assistant failed cases because of change of ml-commons

[Backport 2.12] Fix assistant failed cases because of change of ml-commons #1355