Update capacity_check API in broker