homeassistant-stubs 2024.6.4__py3-none-any.whl → 2024.7.0b0__py3-none-any.whl

homeassistant-stubs/components/shelly/switch.pyi

@@ -1,18 +1,15 @@
-from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, DOMAIN as DOMAIN, GAS_VALVE_OPEN_STATES as GAS_VALVE_OPEN_STATES, MOTION_MODELS as MOTION_MODELS
+from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, MOTION_MODELS as MOTION_MODELS
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry, ShellyRpcCoordinator as ShellyRpcCoordinator
-from .entity import BlockEntityDescription as BlockEntityDescription, ShellyBlockAttributeEntity as ShellyBlockAttributeEntity, ShellyBlockEntity as ShellyBlockEntity, ShellyRpcEntity as ShellyRpcEntity, ShellySleepingBlockAttributeEntity as ShellySleepingBlockAttributeEntity, async_setup_block_attribute_entities as async_setup_block_attribute_entities, async_setup_entry_attribute_entities as async_setup_entry_attribute_entities
+from .entity import BlockEntityDescription as BlockEntityDescription, ShellyBlockEntity as ShellyBlockEntity, ShellyRpcEntity as ShellyRpcEntity, ShellySleepingBlockAttributeEntity as ShellySleepingBlockAttributeEntity, async_setup_entry_attribute_entities as async_setup_entry_attribute_entities
 from .utils import async_remove_shelly_entity as async_remove_shelly_entity, get_device_entry_gen as get_device_entry_gen, get_rpc_key_ids as get_rpc_key_ids, is_block_channel_type_light as is_block_channel_type_light, is_rpc_channel_type_light as is_rpc_channel_type_light, is_rpc_thermostat_internal_actuator as is_rpc_thermostat_internal_actuator, is_rpc_thermostat_mode as is_rpc_thermostat_mode
 from _typeshed import Incomplete
 from aioshelly.block_device import Block as Block
 from dataclasses import dataclass
-from homeassistant.components.automation import automations_with_entity as automations_with_entity
-from homeassistant.components.script import scripts_with_entity as scripts_with_entity
 from homeassistant.components.switch import SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
 from homeassistant.const import EntityCategory as EntityCategory, STATE_ON as STATE_ON
 from homeassistant.core import HomeAssistant as HomeAssistant, State as State, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.entity_registry import RegistryEntry as RegistryEntry
-from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue
 from homeassistant.helpers.restore_state import RestoreEntity as RestoreEntity
 from typing import Any
 
@@ -20,7 +17,6 @@ from typing import Any
 class BlockSwitchDescription(BlockEntityDescription, SwitchEntityDescription):
     def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, unit_fn, value, available, removal_condition, extra_state_attributes) -> None: ...
 
-GAS_VALVE_SWITCH: Incomplete
 MOTION_SWITCH: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ShellyConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
@@ -38,18 +34,6 @@ class BlockSleepingMotionSwitch(ShellySleepingBlockAttributeEntity, RestoreEntit
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     async def async_added_to_hass(self) -> None: ...
 
-class BlockValveSwitch(ShellyBlockAttributeEntity, SwitchEntity):
-    entity_description: BlockSwitchDescription
-    _attr_translation_key: str
-    control_result: Incomplete
-    def __init__(self, coordinator: ShellyBlockCoordinator, block: Block, attribute: str, description: BlockSwitchDescription) -> None: ...
-    @property
-    def is_on(self) -> bool: ...
-    async def async_turn_on(self, **kwargs: Any) -> None: ...
-    async def async_turn_off(self, **kwargs: Any) -> None: ...
-    async def async_added_to_hass(self) -> None: ...
-    def _update_callback(self) -> None: ...
-
 class BlockRelaySwitch(ShellyBlockEntity, SwitchEntity):
     control_result: Incomplete
     def __init__(self, coordinator: ShellyBlockCoordinator, block: Block) -> None: ...

homeassistant-stubs/components/shelly/valve.pyi

@@ -1,6 +1,6 @@
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry
 from .entity import BlockEntityDescription as BlockEntityDescription, ShellyBlockAttributeEntity as ShellyBlockAttributeEntity, async_setup_block_attribute_entities as async_setup_block_attribute_entities
-from .utils import get_device_entry_gen as get_device_entry_gen
+from .utils import async_remove_shelly_entity as async_remove_shelly_entity, get_device_entry_gen as get_device_entry_gen
 from _typeshed import Incomplete
 from aioshelly.block_device import Block as Block
 from dataclasses import dataclass

homeassistant-stubs/components/simplisafe/alarm_control_panel.pyi

@@ -2,9 +2,9 @@ from . import SimpliSafe as SimpliSafe, SimpliSafeEntity as SimpliSafeEntity
 from .const import ATTR_ALARM_DURATION as ATTR_ALARM_DURATION, ATTR_ALARM_VOLUME as ATTR_ALARM_VOLUME, ATTR_CHIME_VOLUME as ATTR_CHIME_VOLUME, ATTR_ENTRY_DELAY_AWAY as ATTR_ENTRY_DELAY_AWAY, ATTR_ENTRY_DELAY_HOME as ATTR_ENTRY_DELAY_HOME, ATTR_EXIT_DELAY_AWAY as ATTR_EXIT_DELAY_AWAY, ATTR_EXIT_DELAY_HOME as ATTR_EXIT_DELAY_HOME, ATTR_LIGHT as ATTR_LIGHT, ATTR_VOICE_PROMPT_VOLUME as ATTR_VOICE_PROMPT_VOLUME, DOMAIN as DOMAIN, LOGGER as LOGGER
 from .typing import SystemType as SystemType
 from _typeshed import Incomplete
-from homeassistant.components.alarm_control_panel import AlarmControlPanelEntity as AlarmControlPanelEntity, AlarmControlPanelEntityFeature as AlarmControlPanelEntityFeature, CodeFormat as CodeFormat
+from homeassistant.components.alarm_control_panel import AlarmControlPanelEntity as AlarmControlPanelEntity, AlarmControlPanelEntityFeature as AlarmControlPanelEntityFeature
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_CODE as CONF_CODE, STATE_ALARM_ARMED_AWAY as STATE_ALARM_ARMED_AWAY, STATE_ALARM_ARMED_HOME as STATE_ALARM_ARMED_HOME, STATE_ALARM_ARMING as STATE_ALARM_ARMING, STATE_ALARM_DISARMED as STATE_ALARM_DISARMED, STATE_ALARM_PENDING as STATE_ALARM_PENDING, STATE_ALARM_TRIGGERED as STATE_ALARM_TRIGGERED
+from homeassistant.const import STATE_ALARM_ARMED_AWAY as STATE_ALARM_ARMED_AWAY, STATE_ALARM_ARMED_HOME as STATE_ALARM_ARMED_HOME, STATE_ALARM_ARMING as STATE_ALARM_ARMING, STATE_ALARM_DISARMED as STATE_ALARM_DISARMED, STATE_ALARM_PENDING as STATE_ALARM_PENDING, STATE_ALARM_TRIGGERED as STATE_ALARM_TRIGGERED
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -23,12 +23,11 @@ WEBSOCKET_EVENTS_TO_LISTEN_FOR: Incomplete
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class SimpliSafeAlarm(SimpliSafeEntity, AlarmControlPanelEntity):
-    _attr_supported_features: Incomplete
+    _attr_code_arm_required: bool
     _attr_name: Incomplete
-    _attr_code_format: Incomplete
+    _attr_supported_features: Incomplete
     _last_event: Incomplete
     def __init__(self, simplisafe: SimpliSafe, system: SystemType) -> None: ...
-    def _is_code_valid(self, code: str | None, state: str) -> bool: ...
     _attr_state: Incomplete
     def _set_state_from_system_data(self) -> None: ...
     async def async_alarm_disarm(self, code: str | None = None) -> None: ...

homeassistant-stubs/components/siren/__init__.pyi

@@ -4,16 +4,17 @@ from functools import cached_property as cached_property
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import SERVICE_TOGGLE as SERVICE_TOGGLE, SERVICE_TURN_OFF as SERVICE_TURN_OFF, SERVICE_TURN_ON as SERVICE_TURN_ON
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall
-from homeassistant.helpers.config_validation import PLATFORM_SCHEMA as PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE as PLATFORM_SCHEMA_BASE
 from homeassistant.helpers.deprecation import all_with_deprecated_constants as all_with_deprecated_constants, check_if_deprecated_constant as check_if_deprecated_constant, dir_with_deprecated_constants as dir_with_deprecated_constants
 from homeassistant.helpers.entity import ToggleEntity as ToggleEntity, ToggleEntityDescription as ToggleEntityDescription
 from homeassistant.helpers.entity_component import EntityComponent as EntityComponent
-from homeassistant.helpers.typing import ConfigType as ConfigType
+from homeassistant.helpers.typing import ConfigType as ConfigType, VolDictType as VolDictType
 from typing import Any, TypedDict
 
 _LOGGER: Incomplete
+PLATFORM_SCHEMA: Incomplete
+PLATFORM_SCHEMA_BASE: Incomplete
 SCAN_INTERVAL: Incomplete
-TURN_ON_SCHEMA: Incomplete
+TURN_ON_SCHEMA: VolDictType
 
 class SirenTurnOnServiceParameters(TypedDict, total=False):
     tone: int | str

homeassistant-stubs/components/sql/sensor.pyi

@@ -5,7 +5,7 @@ from _typeshed import Incomplete
 from homeassistant.components.recorder import CONF_DB_URL as CONF_DB_URL, SupportedDialect as SupportedDialect, get_instance as get_instance
 from homeassistant.components.sensor import CONF_STATE_CLASS as CONF_STATE_CLASS
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_DEVICE_CLASS as CONF_DEVICE_CLASS, CONF_ICON as CONF_ICON, CONF_NAME as CONF_NAME, CONF_UNIQUE_ID as CONF_UNIQUE_ID, CONF_UNIT_OF_MEASUREMENT as CONF_UNIT_OF_MEASUREMENT, CONF_VALUE_TEMPLATE as CONF_VALUE_TEMPLATE, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
+from homeassistant.const import CONF_DEVICE_CLASS as CONF_DEVICE_CLASS, CONF_ICON as CONF_ICON, CONF_NAME as CONF_NAME, CONF_UNIQUE_ID as CONF_UNIQUE_ID, CONF_UNIT_OF_MEASUREMENT as CONF_UNIT_OF_MEASUREMENT, CONF_VALUE_TEMPLATE as CONF_VALUE_TEMPLATE, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, MATCH_ALL as MATCH_ALL
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import TemplateError as TemplateError
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
@@ -31,6 +31,7 @@ def _validate_and_get_session_maker_for_db_url(db_url: str) -> scoped_session |
 def _generate_lambda_stmt(query: str) -> StatementLambdaElement: ...
 
 class SQLSensor(ManualTriggerSensorEntity):
+    _unrecorded_attributes: Incomplete
     _query: Incomplete
     _template: Incomplete
     _column_name: Incomplete

homeassistant-stubs/components/starlink/const.pyi

@@ -1 +1,2 @@
 DOMAIN: str
+ATTR_ALTITUDE: str

homeassistant-stubs/components/starlink/device_tracker.pyi

@@ -1,4 +1,4 @@
-from .const import DOMAIN as DOMAIN
+from .const import ATTR_ALTITUDE as ATTR_ALTITUDE, DOMAIN as DOMAIN
 from .coordinator import StarlinkData as StarlinkData
 from .entity import StarlinkEntity as StarlinkEntity
 from _typeshed import Incomplete
@@ -9,6 +9,7 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity import EntityDescription as EntityDescription
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from typing import Any
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
@@ -16,7 +17,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_e
 class StarlinkDeviceTrackerEntityDescription(EntityDescription):
     latitude_fn: Callable[[StarlinkData], float]
     longitude_fn: Callable[[StarlinkData], float]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, latitude_fn, longitude_fn) -> None: ...
+    altitude_fn: Callable[[StarlinkData], float]
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, latitude_fn, longitude_fn, altitude_fn) -> None: ...
 
 DEVICE_TRACKERS: Incomplete
 
@@ -28,3 +30,5 @@ class StarlinkDeviceTrackerEntity(StarlinkEntity, TrackerEntity):
     def latitude(self) -> float | None: ...
     @property
     def longitude(self) -> float | None: ...
+    @property
+    def extra_state_attributes(self) -> dict[str, Any]: ...

homeassistant-stubs/components/statistics/__init__.pyi

@@ -1,5 +1,11 @@
 from _typeshed import Incomplete
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import Platform as Platform
+from homeassistant.core import HomeAssistant as HomeAssistant
 
 DOMAIN: str
 PLATFORMS: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None: ...

homeassistant-stubs/components/statistics/config_flow.pyi

@@ -0,0 +1,23 @@
+import voluptuous as vol
+from . import DOMAIN as DOMAIN
+from .sensor import CONF_KEEP_LAST_SAMPLE as CONF_KEEP_LAST_SAMPLE, CONF_MAX_AGE as CONF_MAX_AGE, CONF_PERCENTILE as CONF_PERCENTILE, CONF_PRECISION as CONF_PRECISION, CONF_SAMPLES_MAX_BUFFER_SIZE as CONF_SAMPLES_MAX_BUFFER_SIZE, CONF_STATE_CHARACTERISTIC as CONF_STATE_CHARACTERISTIC, DEFAULT_NAME as DEFAULT_NAME, DEFAULT_PRECISION as DEFAULT_PRECISION, STATS_BINARY_SUPPORT as STATS_BINARY_SUPPORT, STATS_NUMERIC_SUPPORT as STATS_NUMERIC_SUPPORT
+from _typeshed import Incomplete
+from collections.abc import Mapping
+from homeassistant.const import CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_NAME as CONF_NAME
+from homeassistant.core import split_entity_id as split_entity_id
+from homeassistant.helpers.schema_config_entry_flow import SchemaCommonFlowHandler as SchemaCommonFlowHandler, SchemaConfigFlowHandler as SchemaConfigFlowHandler, SchemaFlowError as SchemaFlowError, SchemaFlowFormStep as SchemaFlowFormStep
+from homeassistant.helpers.selector import BooleanSelector as BooleanSelector, DurationSelector as DurationSelector, DurationSelectorConfig as DurationSelectorConfig, EntitySelector as EntitySelector, EntitySelectorConfig as EntitySelectorConfig, NumberSelector as NumberSelector, NumberSelectorConfig as NumberSelectorConfig, NumberSelectorMode as NumberSelectorMode, SelectSelector as SelectSelector, SelectSelectorConfig as SelectSelectorConfig, SelectSelectorMode as SelectSelectorMode, TextSelector as TextSelector
+from typing import Any
+
+async def get_state_characteristics(handler: SchemaCommonFlowHandler) -> vol.Schema: ...
+async def validate_options(handler: SchemaCommonFlowHandler, user_input: dict[str, Any]) -> dict[str, Any]: ...
+
+DATA_SCHEMA_SETUP: Incomplete
+DATA_SCHEMA_OPTIONS: Incomplete
+CONFIG_FLOW: Incomplete
+OPTIONS_FLOW: Incomplete
+
+class StatisticsConfigFlowHandler(SchemaConfigFlowHandler, domain=DOMAIN):
+    config_flow = CONFIG_FLOW
+    options_flow = OPTIONS_FLOW
+    def async_config_entry_title(self, options: Mapping[str, Any]) -> str: ...

homeassistant-stubs/components/statistics/sensor.pyi

@@ -3,7 +3,8 @@ from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from datetime import datetime, timedelta
 from homeassistant.components.recorder import get_instance as get_instance, history as history
-from homeassistant.components.sensor import DEVICE_CLASS_STATE_CLASSES as DEVICE_CLASS_STATE_CLASSES, PLATFORM_SCHEMA as PLATFORM_SCHEMA, SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorStateClass as SensorStateClass
+from homeassistant.components.sensor import DEVICE_CLASS_STATE_CLASSES as DEVICE_CLASS_STATE_CLASSES, SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorStateClass as SensorStateClass
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_DEVICE_CLASS as ATTR_DEVICE_CLASS, ATTR_UNIT_OF_MEASUREMENT as ATTR_UNIT_OF_MEASUREMENT, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_NAME as CONF_NAME, CONF_UNIQUE_ID as CONF_UNIQUE_ID, PERCENTAGE as PERCENTAGE, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, State as State, callback as callback, split_entity_id as split_entity_id
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -68,8 +69,10 @@ def valid_boundary_configuration(config: dict[str, Any]) -> dict[str, Any]: ...
 def valid_keep_last_sample(config: dict[str, Any]) -> dict[str, Any]: ...
 
 _PLATFORM_SCHEMA_BASE: Incomplete
+PLATFORM_SCHEMA: Incomplete
 
 async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class StatisticsSensor(SensorEntity):
     _attr_should_poll: bool

homeassistant-stubs/components/stream/fmp4utils.pyi

@@ -1,10 +1,10 @@
 from .core import Orientation as Orientation
 from _typeshed import Incomplete
-from collections.abc import Generator
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from io import BufferedIOBase
+from typing_extensions import Generator
 
-def find_box(mp4_bytes: bytes, target_type: bytes, box_start: int = 0) -> Generator[int, None, None]: ...
+def find_box(mp4_bytes: bytes, target_type: bytes, box_start: int = 0) -> Generator[int]: ...
 def get_codec_string(mp4_bytes: bytes) -> str: ...
 def find_moov(mp4_io: BufferedIOBase) -> int: ...
 def read_init(bytes_io: BufferedIOBase) -> bytes: ...

homeassistant-stubs/components/stream/worker.pyi

@@ -6,11 +6,12 @@ from .diagnostics import Diagnostics as Diagnostics
 from .fmp4utils import read_init as read_init
 from .hls import HlsStreamOutput as HlsStreamOutput
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Generator, Iterator, Mapping
+from collections.abc import Callable as Callable, Iterator, Mapping
 from homeassistant.core import HomeAssistant as HomeAssistant
 from io import BytesIO
 from threading import Event
 from typing import Any, Self
+from typing_extensions import Generator
 
 _LOGGER: Incomplete
 NEGATIVE_INF: Incomplete
@@ -71,7 +72,7 @@ class PeekIterator(Iterator):
     def __iter__(self) -> Self: ...
     def __next__(self) -> av.Packet: ...
     def _pop_buffer(self) -> av.Packet: ...
-    def peek(self) -> Generator[av.Packet, None, None]: ...
+    def peek(self) -> Generator[av.Packet]: ...
 
 class TimestampValidator:
     _last_dts: Incomplete

homeassistant-stubs/components/streamlabswater/__init__.pyi

@@ -1,12 +1,9 @@
 from .const import DOMAIN as DOMAIN
 from .coordinator import StreamlabsCoordinator as StreamlabsCoordinator
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, SOURCE_IMPORT as SOURCE_IMPORT
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall
-from homeassistant.data_entry_flow import FlowResultType as FlowResultType
-from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue
-from homeassistant.helpers.typing import ConfigType as ConfigType
 
 ATTR_AWAY_MODE: str
 SERVICE_SET_AWAY_MODE: str
@@ -14,10 +11,8 @@ AWAY_MODE_AWAY: str
 AWAY_MODE_HOME: str
 CONF_LOCATION_ID: str
 ISSUE_PLACEHOLDER: Incomplete
-CONFIG_SCHEMA: Incomplete
 SET_AWAY_MODE_SCHEMA: Incomplete
 PLATFORMS: list[Platform]
 
-async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/streamlabswater/config_flow.pyi

@@ -10,6 +10,5 @@ async def validate_input(hass: HomeAssistant, api_key: str) -> None: ...
 class StreamlabsConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
-    async def async_step_import(self, user_input: dict[str, Any]) -> ConfigFlowResult: ...
 
 class CannotConnect(HomeAssistantError): ...

homeassistant-stubs/components/suez_water/config_flow.pyi

@@ -13,7 +13,6 @@ def validate_input(data: dict[str, Any]) -> None: ...
 class SuezWaterConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
-    async def async_step_import(self, user_input: dict[str, Any]) -> ConfigFlowResult: ...
 
 class CannotConnect(HomeAssistantError): ...
 class InvalidAuth(HomeAssistantError): ...

homeassistant-stubs/components/suez_water/sensor.pyi

@@ -1,21 +1,16 @@
 from .const import CONF_COUNTER_ID as CONF_COUNTER_ID, DOMAIN as DOMAIN
 from _typeshed import Incomplete
-from homeassistant.components.sensor import PLATFORM_SCHEMA as PLATFORM_SCHEMA, SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, SOURCE_IMPORT as SOURCE_IMPORT
-from homeassistant.const import CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME, UnitOfVolume as UnitOfVolume
+from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import UnitOfVolume as UnitOfVolume
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.data_entry_flow import FlowResultType as FlowResultType
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue
-from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
 from pysuez import SuezClient as SuezClient
 
 _LOGGER: Incomplete
-ISSUE_PLACEHOLDER: Incomplete
 SCAN_INTERVAL: Incomplete
 
-async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class SuezSensor(SensorEntity):

homeassistant-stubs/components/switch/__init__.pyi

@@ -5,17 +5,18 @@ from functools import cached_property as cached_property
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import SERVICE_TOGGLE as SERVICE_TOGGLE, SERVICE_TURN_OFF as SERVICE_TURN_OFF, SERVICE_TURN_ON as SERVICE_TURN_ON, STATE_ON as STATE_ON
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.config_validation import PLATFORM_SCHEMA as PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE as PLATFORM_SCHEMA_BASE
 from homeassistant.helpers.deprecation import DeprecatedConstantEnum as DeprecatedConstantEnum, all_with_deprecated_constants as all_with_deprecated_constants, check_if_deprecated_constant as check_if_deprecated_constant, dir_with_deprecated_constants as dir_with_deprecated_constants
 from homeassistant.helpers.entity import ToggleEntity as ToggleEntity, ToggleEntityDescription as ToggleEntityDescription
 from homeassistant.helpers.entity_component import EntityComponent as EntityComponent
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from homeassistant.loader import bind_hass as bind_hass
 
-SCAN_INTERVAL: Incomplete
+_LOGGER: Incomplete
 ENTITY_ID_FORMAT: Incomplete
+PLATFORM_SCHEMA: Incomplete
+PLATFORM_SCHEMA_BASE: Incomplete
+SCAN_INTERVAL: Incomplete
 MIN_TIME_BETWEEN_SCANS: Incomplete
-_LOGGER: Incomplete
 
 class SwitchDeviceClass(StrEnum):
     OUTLET: str

homeassistant-stubs/components/switch/light.pyi

@@ -1,5 +1,5 @@
 from _typeshed import Incomplete
-from homeassistant.components.light import ColorMode as ColorMode, LightEntity as LightEntity, PLATFORM_SCHEMA as PLATFORM_SCHEMA
+from homeassistant.components.light import ColorMode as ColorMode, LightEntity as LightEntity
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_NAME as CONF_NAME, SERVICE_TURN_OFF as SERVICE_TURN_OFF, SERVICE_TURN_ON as SERVICE_TURN_ON, STATE_ON as STATE_ON, STATE_UNAVAILABLE as STATE_UNAVAILABLE
 from homeassistant.core import Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -8,6 +8,7 @@ from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfo
 from typing import Any
 
 DEFAULT_NAME: str
+PLATFORM_SCHEMA: Incomplete
 
 async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
 

homeassistant-stubs/components/switcher_kis/__init__.pyi

@@ -1,9 +1,11 @@
+from .const import DOMAIN as DOMAIN
 from .coordinator import SwitcherDataUpdateCoordinator as SwitcherDataUpdateCoordinator
 from _typeshed import Incomplete
 from aioswitcher.device import SwitcherBase as SwitcherBase
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, Platform as Platform
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers import device_registry as dr
 
 PLATFORMS: Incomplete
 _LOGGER: Incomplete
@@ -11,3 +13,4 @@ SwitcherConfigEntry = ConfigEntry[dict[str, SwitcherDataUpdateCoordinator]]
 
 async def async_setup_entry(hass: HomeAssistant, entry: SwitcherConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: SwitcherConfigEntry) -> bool: ...
+async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: SwitcherConfigEntry, device_entry: dr.DeviceEntry) -> bool: ...

homeassistant-stubs/components/switcher_kis/switch.pyi

@@ -10,14 +10,15 @@ from homeassistant.helpers import entity_platform as entity_platform
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.typing import VolDictType as VolDictType
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from typing import Any
 
 _LOGGER: Incomplete
 API_CONTROL_DEVICE: str
 API_SET_AUTO_SHUTDOWN: str
-SERVICE_SET_AUTO_OFF_SCHEMA: Incomplete
-SERVICE_TURN_ON_WITH_TIMER_SCHEMA: Incomplete
+SERVICE_SET_AUTO_OFF_SCHEMA: VolDictType
+SERVICE_TURN_ON_WITH_TIMER_SCHEMA: VolDictType
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 

homeassistant-stubs/components/synology_dsm/config_flow.pyi

@@ -8,7 +8,7 @@ from homeassistant.const import CONF_DISKS as CONF_DISKS, CONF_HOST as CONF_HOST
 from homeassistant.core import callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
-from homeassistant.helpers.typing import DiscoveryInfoType as DiscoveryInfoType
+from homeassistant.helpers.typing import DiscoveryInfoType as DiscoveryInfoType, VolDictType as VolDictType
 from synology_dsm import SynologyDSM
 from typing import Any
 
@@ -19,7 +19,7 @@ HTTP_SUFFIX: str
 def _discovery_schema_with_defaults(discovery_info: DiscoveryInfoType) -> vol.Schema: ...
 def _reauth_schema() -> vol.Schema: ...
 def _user_schema_with_defaults(user_input: dict[str, Any]) -> vol.Schema: ...
-def _ordered_shared_schema(schema_input: dict[str, Any]) -> dict[vol.Required | vol.Optional, Any]: ...
+def _ordered_shared_schema(schema_input: dict[str, Any]) -> VolDictType: ...
 def format_synology_mac(mac: str) -> str: ...
 
 class SynologyDSMFlowHandler(ConfigFlow, domain=DOMAIN):

homeassistant-stubs/components/systemmonitor/config_flow.pyi

@@ -5,14 +5,12 @@ from _typeshed import Incomplete
 from collections.abc import Mapping
 from homeassistant.config_entries import ConfigFlowResult as ConfigFlowResult
 from homeassistant.core import callback as callback
-from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue
 from homeassistant.helpers.schema_config_entry_flow import SchemaCommonFlowHandler as SchemaCommonFlowHandler, SchemaConfigFlowHandler as SchemaConfigFlowHandler, SchemaFlowFormStep as SchemaFlowFormStep
 from homeassistant.helpers.selector import SelectSelector as SelectSelector, SelectSelectorConfig as SelectSelectorConfig, SelectSelectorMode as SelectSelectorMode
 from homeassistant.util import slugify as slugify
 from typing import Any
 
 async def validate_sensor_setup(handler: SchemaCommonFlowHandler, user_input: dict[str, Any]) -> dict[str, Any]: ...
-async def validate_import_sensor_setup(handler: SchemaCommonFlowHandler, user_input: dict[str, Any]) -> dict[str, Any]: ...
 async def get_sensor_setup_schema(handler: SchemaCommonFlowHandler) -> vol.Schema: ...
 async def get_suggested_value(handler: SchemaCommonFlowHandler) -> dict[str, Any]: ...
 

homeassistant-stubs/components/systemmonitor/sensor.pyi

@@ -6,17 +6,16 @@ from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from datetime import datetime
-from homeassistant.components.sensor import PLATFORM_SCHEMA as PLATFORM_SCHEMA, SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
-from homeassistant.config_entries import SOURCE_IMPORT as SOURCE_IMPORT
-from homeassistant.const import CONF_RESOURCES as CONF_RESOURCES, CONF_TYPE as CONF_TYPE, EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, STATE_OFF as STATE_OFF, STATE_ON as STATE_ON, UnitOfDataRate as UnitOfDataRate, UnitOfInformation as UnitOfInformation, UnitOfTemperature as UnitOfTemperature
+from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
+from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, STATE_OFF as STATE_OFF, STATE_ON as STATE_ON, UnitOfDataRate as UnitOfDataRate, UnitOfInformation as UnitOfInformation, UnitOfTemperature as UnitOfTemperature
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue
-from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType, StateType as StateType
+from homeassistant.helpers.typing import StateType as StateType
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from homeassistant.util import slugify as slugify
-from typing import Any, Literal
+from typing import Literal
 
 _LOGGER: Incomplete
 CONF_ARG: str
@@ -45,13 +44,11 @@ class SysMonitorSensorEntityDescription(SensorEntityDescription):
 
 SENSOR_TYPES: dict[str, SysMonitorSensorEntityDescription]
 
-def check_required_arg(value: Any) -> Any: ...
 def check_legacy_resource(resource: str, resources: set[str]) -> bool: ...
 
 IO_COUNTER: Incomplete
 IF_ADDRS_FAMILY: Incomplete
 
-async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
 async def async_setup_entry(hass: HomeAssistant, entry: SystemMonitorConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class SystemMonitorSensor(CoordinatorEntity[SystemMonitorCoordinator], SensorEntity):

homeassistant-stubs/components/tag/__init__.pyi

@@ -9,7 +9,7 @@ from homeassistant.helpers import collection as collection, entity_registry as e
 from homeassistant.helpers.entity import Entity as Entity
 from homeassistant.helpers.entity_component import EntityComponent as EntityComponent
 from homeassistant.helpers.storage import Store as Store
-from homeassistant.helpers.typing import ConfigType as ConfigType
+from homeassistant.helpers.typing import ConfigType as ConfigType, VolDictType as VolDictType
 from homeassistant.util import slugify as slugify
 from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Any
@@ -21,9 +21,8 @@ STORAGE_KEY = DOMAIN
 STORAGE_VERSION: int
 STORAGE_VERSION_MINOR: int
 TAG_DATA: HassKey[TagStorageCollection]
-SIGNAL_TAG_CHANGED: str
-CREATE_FIELDS: Incomplete
-UPDATE_FIELDS: Incomplete
+CREATE_FIELDS: VolDictType
+UPDATE_FIELDS: VolDictType
 CONFIG_SCHEMA: Incomplete
 
 class TagIDExistsError(HomeAssistantError):
@@ -50,7 +49,7 @@ class TagStorageCollection(collection.DictStorageCollection):
 
 class TagDictStorageCollectionWebsocket(collection.StorageCollectionWebsocket[TagStorageCollection]):
     entity_registry: Incomplete
-    def __init__(self, storage_collection: TagStorageCollection, api_prefix: str, model_name: str, create_schema: ConfigType, update_schema: ConfigType) -> None: ...
+    def __init__(self, storage_collection: TagStorageCollection, api_prefix: str, model_name: str, create_schema: VolDictType, update_schema: VolDictType) -> None: ...
     def ws_list_item(self, hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict) -> None: ...
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...

homeassistant-stubs/components/tami4/__init__.pyi

@@ -1,5 +1,5 @@
 from .const import API as API, CONF_REFRESH_TOKEN as CONF_REFRESH_TOKEN, COORDINATOR as COORDINATOR, DOMAIN as DOMAIN
-from .coordinator import Tami4EdgeWaterQualityCoordinator as Tami4EdgeWaterQualityCoordinator
+from .coordinator import Tami4EdgeCoordinator as Tami4EdgeCoordinator
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant

homeassistant-stubs/components/tami4/coordinator.pyi

@@ -10,16 +10,14 @@ _LOGGER: Incomplete
 
 @dataclass
 class FlattenedWaterQuality:
-    uv_last_replacement: date
     uv_upcoming_replacement: date
-    uv_status: str
-    filter_last_replacement: date
+    uv_installed: bool
     filter_upcoming_replacement: date
-    filter_status: str
+    filter_installed: bool
     filter_litters_passed: float
     def __init__(self, water_quality: WaterQuality) -> None: ...
 
-class Tami4EdgeWaterQualityCoordinator(DataUpdateCoordinator[FlattenedWaterQuality]):
+class Tami4EdgeCoordinator(DataUpdateCoordinator[FlattenedWaterQuality]):
     _api: Incomplete
     def __init__(self, hass: HomeAssistant, api: Tami4EdgeAPI) -> None: ...
     async def _async_update_data(self) -> FlattenedWaterQuality: ...

homeassistant-stubs/components/tami4/sensor.pyi

@@ -1,5 +1,5 @@
 from .const import API as API, COORDINATOR as COORDINATOR, DOMAIN as DOMAIN
-from .coordinator import Tami4EdgeWaterQualityCoordinator as Tami4EdgeWaterQualityCoordinator
+from .coordinator import Tami4EdgeCoordinator as Tami4EdgeCoordinator
 from .entity import Tami4EdgeBaseEntity as Tami4EdgeBaseEntity
 from Tami4EdgeAPI import Tami4EdgeAPI as Tami4EdgeAPI
 from _typeshed import Incomplete
@@ -15,8 +15,8 @@ ENTITY_DESCRIPTIONS: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class Tami4EdgeSensorEntity(Tami4EdgeBaseEntity, CoordinatorEntity[Tami4EdgeWaterQualityCoordinator], SensorEntity):
-    def __init__(self, coordinator: Tami4EdgeWaterQualityCoordinator, api: Tami4EdgeAPI, entity_description: SensorEntityDescription) -> None: ...
+class Tami4EdgeSensorEntity(Tami4EdgeBaseEntity, CoordinatorEntity[Tami4EdgeCoordinator], SensorEntity):
+    def __init__(self, coordinator: Tami4EdgeCoordinator, api: Tami4EdgeAPI, entity_description: SensorEntityDescription) -> None: ...
     _attr_native_value: Incomplete
     def _update_attr(self) -> None: ...
     def _handle_coordinator_update(self) -> None: ...

homeassistant-stubs/components/tautulli/__init__.pyi

@@ -11,9 +11,10 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity as Coordi
 from pytautulli import PyTautulliApiUser as PyTautulliApiUser
 
 PLATFORMS: Incomplete
+TautulliConfigEntry = ConfigEntry[TautulliDataUpdateCoordinator]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, entry: TautulliConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: TautulliConfigEntry) -> bool: ...
 
 class TautulliEntity(CoordinatorEntity[TautulliDataUpdateCoordinator]):
     _attr_has_entity_name: bool

homeassistant-stubs/components/tautulli/coordinator.pyi

@@ -1,6 +1,6 @@
+from . import TautulliConfigEntry as TautulliConfigEntry
 from .const import DOMAIN as DOMAIN, LOGGER as LOGGER
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
@@ -8,7 +8,7 @@ from pytautulli import PyTautulli as PyTautulli, PyTautulliApiActivity as PyTaut
 from pytautulli.models.host_configuration import PyTautulliHostConfiguration as PyTautulliHostConfiguration
 
 class TautulliDataUpdateCoordinator(DataUpdateCoordinator[None]):
-    config_entry: ConfigEntry
+    config_entry: TautulliConfigEntry
     host_configuration: Incomplete
     api_client: Incomplete
     activity: Incomplete

homeassistant-stubs/components/tautulli/sensor.pyi

@@ -1,11 +1,11 @@
-from . import TautulliEntity as TautulliEntity
+from . import TautulliConfigEntry as TautulliConfigEntry, TautulliEntity as TautulliEntity
 from .const import ATTR_TOP_USER as ATTR_TOP_USER, DOMAIN as DOMAIN
 from .coordinator import TautulliDataUpdateCoordinator as TautulliDataUpdateCoordinator
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, SOURCE_IMPORT as SOURCE_IMPORT
+from homeassistant.config_entries import SOURCE_IMPORT as SOURCE_IMPORT
 from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfInformation as UnitOfInformation
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity import EntityDescription as EntityDescription
@@ -30,7 +30,7 @@ class TautulliSessionSensorEntityDescription(SensorEntityDescription):
 SESSION_SENSOR_TYPES: tuple[TautulliSessionSensorEntityDescription, ...]
 
 async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: TautulliConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TautulliSensor(TautulliEntity, SensorEntity):
     entity_description: TautulliSensorEntityDescription

homeassistant-stubs/components/threshold/__init__.pyi

@@ -1,6 +1,7 @@
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import Platform as Platform
+from homeassistant.const import CONF_ENTITY_ID as CONF_ENTITY_ID, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.device import async_remove_stale_devices_links_keep_entity_device as async_remove_stale_devices_links_keep_entity_device
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 async def config_entry_update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None: ...