Ensuring reliable operation and maximum the output is crucial for energy production systems. Traditional time-based availability (TBA) warranty policies often overlook some factors, such as energy loss and the experience gained during the maintenance activities. In this paper, an innovative warranty policy which focuses on the energy-based availability (EBA) is proposed, where imperfect preventive maintenance (IPM) and minimal repair (MR) are taken into account, and hybrid hazard rate model is adopted to describe the effect of preventive maintenance. In addition, the learning-forgetting effect during the maintenance is considered. On this basis, six types of single-objective and multi-objective models are established, and they are solved by genetic algorithm (GA) and non-dominated sorting genetic algorithm-II (NSGA-II), respectively. To illustrate the effectiveness of the proposed warranty policy, a numerical case of wind turbine gearbox is conducted. The results show that the proposed EBA warranty policy can gain around 0.29% energy more than TBA policy. Compared to single-objective models, multi-objective models can provide more selectable maintenance options. Additionally, sensitivity analysis indicates that by considering the learning-forgetting effect, the gearbox can achieve higher EBA and lower warranty cost.