diff --git a/homeassistant/helpers/selector.py b/homeassistant/helpers/selector.py index 2e7df07cf04..afd38bf7636 100644 --- a/homeassistant/helpers/selector.py +++ b/homeassistant/helpers/selector.py @@ -189,8 +189,6 @@ class DeviceFilterSelectorConfig(TypedDict, total=False): integration: str manufacturer: str model: str - entity: EntityFilterSelectorConfig | list[EntityFilterSelectorConfig] - filter: DeviceFilterSelectorConfig | list[DeviceFilterSelectorConfig] class ActionSelectorConfig(TypedDict): @@ -546,14 +544,12 @@ class DateTimeSelector(Selector[DateTimeSelectorConfig]): return data -class DeviceSelectorConfig(TypedDict, total=False): +class DeviceSelectorConfig(DeviceFilterSelectorConfig, total=False): """Class to represent a device selector config.""" - integration: str - manufacturer: str - model: str entity: EntityFilterSelectorConfig | list[EntityFilterSelectorConfig] multiple: bool + filter: DeviceFilterSelectorConfig | list[DeviceFilterSelectorConfig] @SELECTORS.register("device") @@ -622,6 +618,7 @@ class EntitySelectorConfig(EntityFilterSelectorConfig, total=False): exclude_entities: list[str] include_entities: list[str] multiple: bool + filter: EntityFilterSelectorConfig | list[EntityFilterSelectorConfig] @SELECTORS.register("entity")