* refactor: Replace API call with build_cls_from_json_with_retry function * fix lint error * fix lint errors * lint * trigger
* Agent Infra * Update test/utils/test_agent_infra.py