"feat-add-entity-category-for-config-entities"

reachy_mini_ha_voice/entity.py

@@ -286,6 +286,7 @@ class NumberEntity(ESPHomeEntity):
         icon: str = "",
         unit_of_measurement: str = "",
         mode: int = 0,  # 0 = auto, 1 = box, 2 = slider
+        entity_category: int = 0,  # 0 = none, 1 = config, 2 = diagnostic
         value_getter: Optional[Callable[[], float]] = None,
         value_setter: Optional[Callable[[float], None]] = None,
     ) -> None:
@@ -299,6 +300,7 @@ class NumberEntity(ESPHomeEntity):
         self.icon = icon
         self.unit_of_measurement = unit_of_measurement
         self.mode = mode
+        self.entity_category = entity_category
         self._value_getter = value_getter
         self._value_setter = value_setter
         self._value = min_value
@@ -329,6 +331,7 @@ class NumberEntity(ESPHomeEntity):
                 step=self.step,
                 unit_of_measurement=self.unit_of_measurement,
                 mode=self.mode,
+                entity_category=self.entity_category,
             )
         elif isinstance(msg, (SubscribeHomeAssistantStatesRequest, SubscribeStatesRequest)):
             yield self._get_state_message()

reachy_mini_ha_voice/entity_extensions.py

@@ -121,6 +121,7 @@ class SwitchEntity(ESPHomeEntity):
         object_id: str,
         icon: str = "",
         device_class: str = "",
+        entity_category: int = 0,  # 0 = none, 1 = config, 2 = diagnostic
         value_getter: Optional[Callable[[], bool]] = None,
         value_setter: Optional[Callable[[bool], None]] = None,
     ) -> None:
@@ -130,6 +131,7 @@ class SwitchEntity(ESPHomeEntity):
         self.object_id = object_id
         self.icon = icon
         self.device_class = device_class
+        self.entity_category = entity_category
         self._value_getter = value_getter
         self._value_setter = value_setter
         self._value = False
@@ -154,6 +156,7 @@ class SwitchEntity(ESPHomeEntity):
                 name=self.name,
                 icon=self.icon,
                 device_class=self.device_class,
+                entity_category=self.entity_category,
             )
         elif isinstance(msg, (SubscribeHomeAssistantStatesRequest, SubscribeStatesRequest)):
             yield self._get_state_message()

reachy_mini_ha_voice/satellite.py

@@ -690,6 +690,7 @@ class VoiceSatelliteProtocol(APIServer):
             icon="mdi:volume-high",
             unit_of_measurement="%",
             mode=2,  # Slider mode
+            entity_category=1,  # config
             value_getter=self.reachy_controller.get_speaker_volume,
             value_setter=self.reachy_controller.set_speaker_volume,
         )
@@ -1259,6 +1260,7 @@ class VoiceSatelliteProtocol(APIServer):
             icon="mdi:microphone",
             unit_of_measurement="%",
             mode=2,  # Slider mode
+            entity_category=1,  # config
             value_getter=self.reachy_controller.get_microphone_volume,
             value_setter=self.reachy_controller.set_microphone_volume,
         )
@@ -1384,6 +1386,7 @@ class VoiceSatelliteProtocol(APIServer):
             object_id="agc_enabled",
             icon="mdi:tune-vertical",
             device_class="switch",
+            entity_category=1,  # config
             value_getter=self.reachy_controller.get_agc_enabled,
             value_setter=self.reachy_controller.set_agc_enabled,
         )
@@ -1401,6 +1404,7 @@ class VoiceSatelliteProtocol(APIServer):
             icon="mdi:volume-plus",
             unit_of_measurement="dB",
             mode=2,
+            entity_category=1,  # config
             value_getter=self.reachy_controller.get_agc_max_gain,
             value_setter=self.reachy_controller.set_agc_max_gain,
         )
@@ -1418,6 +1422,7 @@ class VoiceSatelliteProtocol(APIServer):
             icon="mdi:volume-off",
             unit_of_measurement="%",
             mode=2,
+            entity_category=1,  # config
             value_getter=self.reachy_controller.get_noise_suppression,
             value_setter=self.reachy_controller.set_noise_suppression,
         )