This study presents a novel battery-aware approach for connectivity maintenance for multi-agent systems. It extends a classical connectivity controller by incorporating battery states into the computation of control inputs, enhancing the network’s transition response in the event of agent removals by anticipating battery depletion. Additionally, strategies for agent addition are introduced to maintain the desired level of connectivity within the multi-agent network. The proposed approach is thoroughly tested in the Gymnasium (previously OpenAI Gym) environment using a specific mission scenario. The results demonstrate the controller’s ability to anticipate agent removals and maintain robust performance during transitional stages. The approach is further validated with Crazyflie UAVs in both simulation and real-world environments, with a fleet management system. The system is designed to be modular and easily adaptable to different flight controllers, demonstrating the applicability of the proposed algorithms to real-world multi-robot missions.