diff --git a/homeassistant/components/xiaomi_miio/const.py b/homeassistant/components/xiaomi_miio/const.py index e27fb4d2110..a1f2414b713 100644 --- a/homeassistant/components/xiaomi_miio/const.py +++ b/homeassistant/components/xiaomi_miio/const.py @@ -195,8 +195,25 @@ MODELS_LIGHT = ( + MODELS_LIGHT_BULB + MODELS_LIGHT_MONO ) -MODELS_VACUUM = [ROCKROBO_V1, ROCKROBO_S5, ROCKROBO_S6, ROCKROBO_S6_MAXV, ROCKROBO_S7] -MODELS_VACUUM_WITH_MOP = [ROCKROBO_S5, ROCKROBO_S6, ROCKROBO_S6_MAXV, ROCKROBO_S7] + +# TODO: use const from pythonmiio once new release with the constant has been published. # pylint: disable=fixme +ROCKROBO_S5_MAX = "roborock.vacuum.s5e" +MODELS_VACUUM = [ + ROCKROBO_V1, + ROCKROBO_S5, + ROCKROBO_S5_MAX, + ROCKROBO_S6, + ROCKROBO_S6_MAXV, + ROCKROBO_S7, +] +MODELS_VACUUM_WITH_MOP = [ + ROCKROBO_S5, + ROCKROBO_S5_MAX, + ROCKROBO_S6, + ROCKROBO_S6_MAXV, + ROCKROBO_S7, +] + MODELS_AIR_MONITOR = [ MODEL_AIRQUALITYMONITOR_V1, MODEL_AIRQUALITYMONITOR_B1,