The trick with the AirGradient is to use one of the ESPHome configs available [1] instead of their crappy Arduino sketch. Then it integrates with Home Assistant perfectly.

[1] e.g. https://github.com/MallocArray/airgradient_esphome