Org Chart JS component supports customizing the expand or collapse action behavior. You can manually change the expand/collapse node ids when the expand icons are clicked. This Code sample demonstrates, how to add expand/collapsed button for partners.