diff --git a/homeassistant/components/esphome/entry_data.py b/homeassistant/components/esphome/entry_data.py index ae0dca49411..da0dae52569 100644 --- a/homeassistant/components/esphome/entry_data.py +++ b/homeassistant/components/esphome/entry_data.py @@ -47,6 +47,7 @@ from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback from homeassistant.helpers import entity_registry as er from homeassistant.helpers.dispatcher import async_dispatcher_send from homeassistant.helpers.storage import Store +from homeassistant.util.signal_type import SignalType from .const import DOMAIN from .dashboard import async_get_dashboard @@ -147,9 +148,9 @@ class RuntimeEntryData: ) @property - def signal_static_info_updated(self) -> str: + def signal_static_info_updated(self) -> SignalType[list[EntityInfo]]: """Return the signal to listen to for updates on static info.""" - return f"esphome_{self.entry_id}_on_list" + return SignalType(f"esphome_{self.entry_id}_on_list") @callback def async_register_static_info_callback( diff --git a/homeassistant/components/esphome/update.py b/homeassistant/components/esphome/update.py index 5a565f9914d..3e5a82bbd0b 100644 --- a/homeassistant/components/esphome/update.py +++ b/homeassistant/components/esphome/update.py @@ -139,7 +139,9 @@ class ESPHomeUpdateEntity(CoordinatorEntity[ESPHomeDashboard], UpdateEntity): ) @callback - def _handle_device_update(self, static_info: EntityInfo | None = None) -> None: + def _handle_device_update( + self, static_info: list[EntityInfo] | None = None + ) -> None: """Handle updated data from the device.""" self._update_attrs() self.async_write_ha_state()