AI data centers should be operated according to two limiting factors.
1) No energy from grid. Can't use coal or fossil fuel energy sources. Must have plan to provide excess TO grid.
2) No use of fresh water from municipal or fresh groundwater for cooling. Can use waste water. Must transition to providing excess fresh water to common supply.
No loopholes. Massive penalties for use of loopholes or breaking rules, not limited to but including complete shutdown of data center.
Those two limits will spur innovation AND prevent AI being criticized for energy use. These rules would hard burn improvements in energy storage and renewables as well as other methods of energy production.
Give them five years to comply to some useful progress percentage. Plenty of time to come up with a transition plan and show sufficient progress to justify further extensions. Realistically it will take 20 years at least to fully realize this plan.
Don't bring up cost. If you do, let me remind everyone that the climate change issue is real enough to hurt now. There's the very real cost of not pursuing these rules. AI has had plenty of time to bootstrap off grid. Now it can begin to migrate to something else instead.
Those with experience with energy generation will realize this plan has ridiculously high reward for those who follow it. Have your cake and eat it too definitely applies.