[AINode] cherry-pick to version 2.0.6 #25698

Job	Run time
unit-test (17, ubuntu-latest, others)	1m 52s
unit-test (17, ubuntu-latest, datanode)	1m 50s
unit-test (17, windows-latest, datanode)	1m 55s
unit-test (17, windows-latest, others)	1m 57s
	7m 34s

Provide feedback