homeassistant-stubs 2025.2.5__py3-none-any.whl → 2025.3.0__py3-none-any.whl

homeassistant-stubs/components/group/notify.pyi

@@ -5,7 +5,7 @@ from homeassistant.components.notify import ATTR_DATA as ATTR_DATA, ATTR_MESSAGE
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_ACTION as CONF_ACTION, CONF_ENTITIES as CONF_ENTITIES, CONF_SERVICE as CONF_SERVICE, STATE_UNAVAILABLE as STATE_UNAVAILABLE
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
 from typing import Any
 
@@ -24,7 +24,7 @@ class GroupNotifyPlatform(BaseNotificationService):
     def __init__(self, hass: HomeAssistant, entities: list[dict[str, Any]]) -> None: ...
     async def async_send_message(self, message: str = '', **kwargs: Any) -> None: ...
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 @callback
 def async_create_preview_notify(hass: HomeAssistant, name: str, validated_config: dict[str, Any]) -> NotifyGroup: ...
 

homeassistant-stubs/components/group/sensor.pyi

@@ -8,7 +8,7 @@ from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_DEVICE_CL
 from homeassistant.core import HomeAssistant as HomeAssistant, State as State, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity import get_capability as get_capability, get_device_class as get_device_class, get_unit_of_measurement as get_unit_of_measurement
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback, AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue, async_delete_issue as async_delete_issue
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType, StateType as StateType
 from typing import Any
@@ -33,7 +33,7 @@ PLATFORM_SCHEMA: Incomplete
 _LOGGER: 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, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 @callback
 def async_create_preview_sensor(hass: HomeAssistant, name: str, validated_config: dict[str, Any]) -> SensorGroup: ...
 def _has_numeric_state(hass: HomeAssistant, entity_id: str) -> bool: ...

homeassistant-stubs/components/group/switch.pyi

@@ -4,7 +4,7 @@ from homeassistant.components.switch import SwitchEntity as SwitchEntity
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_ENTITIES as CONF_ENTITIES, CONF_NAME as CONF_NAME, CONF_UNIQUE_ID as CONF_UNIQUE_ID, SERVICE_TURN_OFF as SERVICE_TURN_OFF, SERVICE_TURN_ON as SERVICE_TURN_ON, STATE_ON as STATE_ON, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback, AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
 from typing import Any
 
@@ -15,7 +15,7 @@ PLATFORM_SCHEMA: Incomplete
 _LOGGER: 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, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 @callback
 def async_create_preview_switch(hass: HomeAssistant, name: str, validated_config: dict[str, Any]) -> SwitchGroup: ...
 

homeassistant-stubs/components/guardian/__init__.pyi

@@ -5,7 +5,7 @@ from _typeshed import Incomplete
 from aioguardian import Client
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID, CONF_DEVICE_ID as CONF_DEVICE_ID, CONF_FILENAME as CONF_FILENAME, CONF_IP_ADDRESS as CONF_IP_ADDRESS, CONF_PORT as CONF_PORT, CONF_URL as CONF_URL, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError

homeassistant-stubs/components/guardian/binary_sensor.pyi

@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from typing import Any
 
 ATTR_CONNECTED_CLIENTS: str
@@ -29,7 +29,7 @@ class ValveControllerBinarySensorDescription(BinarySensorEntityDescription, Valv
 PAIRED_SENSOR_DESCRIPTIONS: Incomplete
 VALVE_CONTROLLER_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class PairedSensorBinarySensor(PairedSensorEntity, BinarySensorEntity):
     entity_description: PairedSensorBinarySensorDescription

homeassistant-stubs/components/guardian/button.pyi

@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.dispatcher import async_dispatcher_send as async_dispatcher_send
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 
 @dataclass(frozen=True, kw_only=True)
 class ValveControllerButtonDescription(ButtonEntityDescription, ValveControllerEntityDescription):
@@ -25,7 +25,7 @@ async def _async_valve_reset(client: Client) -> None: ...
 
 BUTTON_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class GuardianButton(ValveControllerEntity, ButtonEntity):
     _attr_device_class: Incomplete

homeassistant-stubs/components/guardian/sensor.pyi

@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory, UnitOfElectricCurrent as UnitOfElectricCurrent, UnitOfElectricPotential as UnitOfElectricPotential, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
 from typing import Any
 
@@ -32,7 +32,7 @@ class ValveControllerSensorDescription(SensorEntityDescription, ValveControllerE
 PAIRED_SENSOR_DESCRIPTIONS: Incomplete
 VALVE_CONTROLLER_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class PairedSensorSensor(PairedSensorEntity, SensorEntity):
     entity_description: PairedSensorDescription

homeassistant-stubs/components/guardian/switch.pyi

@@ -11,7 +11,7 @@ from homeassistant.components.switch import SwitchEntity as SwitchEntity, Switch
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from typing import Any
 
 ATTR_AVG_CURRENT: str
@@ -39,7 +39,7 @@ def is_open(data: dict[str, Any]) -> bool: ...
 
 VALVE_CONTROLLER_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ValveControllerSwitch(ValveControllerEntity, SwitchEntity):
     entity_description: ValveControllerSwitchDescription

homeassistant-stubs/components/guardian/valve.pyi

@@ -10,7 +10,7 @@ from enum import StrEnum
 from homeassistant.components.valve import ValveDeviceClass as ValveDeviceClass, ValveEntity as ValveEntity, ValveEntityDescription as ValveEntityDescription, ValveEntityFeature as ValveEntityFeature
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from typing import Any
 
 VALVE_KIND_VALVE: str
@@ -44,7 +44,7 @@ def is_opening(data: dict[str, Any]) -> bool: ...
 
 VALVE_CONTROLLER_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ValveControllerValve(ValveControllerEntity, ValveEntity):
     _attr_supported_features: Incomplete

homeassistant-stubs/components/habitica/__init__.pyi

@@ -1,7 +1,6 @@
 from .const import CONF_API_USER as CONF_API_USER, DOMAIN as DOMAIN, X_CLIENT as X_CLIENT
-from .coordinator import HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry, HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
 from .services import async_setup_services as async_setup_services
-from .types import HabiticaConfigEntry as HabiticaConfigEntry
 from _typeshed import Incomplete
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY, CONF_URL as CONF_URL, CONF_VERIFY_SSL as CONF_VERIFY_SSL, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant

homeassistant-stubs/components/habitica/binary_sensor.pyi

@@ -1,13 +1,13 @@
 from .const import ASSETS_URL as ASSETS_URL
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry
 from .entity import HabiticaBase as HabiticaBase
-from .types import HabiticaConfigEntry as HabiticaConfigEntry
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from enum import StrEnum
 from habiticalib import UserData as UserData
 from homeassistant.components.binary_sensor import BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 
 PARALLEL_UPDATES: int
 
@@ -23,7 +23,7 @@ def get_scroll_image_for_pending_quest_invitation(user: UserData) -> str | None:
 
 BINARY_SENSOR_DESCRIPTIONS: tuple[HabiticaBinarySensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: HabiticaConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: HabiticaConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class HabiticaBinarySensorEntity(HabiticaBase, BinarySensorEntity):
     entity_description: HabiticaBinarySensorEntityDescription

homeassistant-stubs/components/habitica/button.pyi

@@ -1,7 +1,6 @@
 from .const import ASSETS_URL as ASSETS_URL, DOMAIN as DOMAIN
-from .coordinator import HabiticaData as HabiticaData, HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry, HabiticaData as HabiticaData, HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
 from .entity import HabiticaBase as HabiticaBase
-from .types import HabiticaConfigEntry as HabiticaConfigEntry
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from enum import StrEnum
@@ -9,7 +8,7 @@ from habiticalib import HabiticaClass
 from homeassistant.components.button import ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, ServiceValidationError as ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from typing import Any
 
 PARALLEL_UPDATES: int
@@ -42,7 +41,7 @@ class HabiticaButtonEntity(StrEnum):
 BUTTON_DESCRIPTIONS: tuple[HabiticaButtonEntityDescription, ...]
 CLASS_SKILLS: tuple[HabiticaButtonEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: HabiticaConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: HabiticaConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class HabiticaButton(HabiticaBase, ButtonEntity):
     entity_description: HabiticaButtonEntityDescription

homeassistant-stubs/components/habitica/calendar.pyi

@@ -1,6 +1,5 @@
 import abc
-from . import HabiticaConfigEntry as HabiticaConfigEntry
-from .coordinator import HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry, HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
 from .entity import HabiticaBase as HabiticaBase
 from .util import build_rrule as build_rrule, get_recurrence_rule as get_recurrence_rule
 from _typeshed import Incomplete
@@ -10,7 +9,7 @@ from dateutil.rrule import rrule as rrule
 from enum import StrEnum
 from homeassistant.components.calendar import CalendarEntity as CalendarEntity, CalendarEntityDescription as CalendarEntityDescription, CalendarEvent as CalendarEvent
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 
 PARALLEL_UPDATES: int
 
@@ -20,7 +19,7 @@ class HabiticaCalendar(StrEnum):
     TODO_REMINDERS = 'todo_reminders'
     DAILY_REMINDERS = 'daily_reminders'
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: HabiticaConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: HabiticaConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class HabiticaCalendarEntity(HabiticaBase, CalendarEntity, metaclass=abc.ABCMeta):
     def __init__(self, coordinator: HabiticaDataUpdateCoordinator) -> None: ...

homeassistant-stubs/components/habitica/config_flow.pyi

@@ -1,5 +1,5 @@
-from . import HabiticaConfigEntry as HabiticaConfigEntry
 from .const import CONF_API_USER as CONF_API_USER, DEFAULT_URL as DEFAULT_URL, DOMAIN as DOMAIN, FORGOT_PASSWORD_URL as FORGOT_PASSWORD_URL, HABITICANS_URL as HABITICANS_URL, SECTION_DANGER_ZONE as SECTION_DANGER_ZONE, SECTION_REAUTH_API_KEY as SECTION_REAUTH_API_KEY, SECTION_REAUTH_LOGIN as SECTION_REAUTH_LOGIN, SIGN_UP_URL as SIGN_UP_URL, SITE_DATA_URL as SITE_DATA_URL, X_CLIENT as X_CLIENT
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry
 from _typeshed import Incomplete
 from collections.abc import Mapping
 from habiticalib import LoginData as LoginData, UserData as UserData

homeassistant-stubs/components/habitica/const.pyi

@@ -26,6 +26,10 @@ ATTR_TYPE: str
 ATTR_PRIORITY: str
 ATTR_TAG: str
 ATTR_KEYWORD: str
+ATTR_REMOVE_TAG: str
+ATTR_ALIAS: str
+ATTR_COST: str
+ATTR_NOTES: str
 SERVICE_CAST_SKILL: str
 SERVICE_START_QUEST: str
 SERVICE_ACCEPT_QUEST: str
@@ -37,6 +41,7 @@ SERVICE_GET_TASKS: str
 SERVICE_SCORE_HABIT: str
 SERVICE_SCORE_REWARD: str
 SERVICE_TRANSFORMATION: str
+SERVICE_UPDATE_REWARD: str
 DEVELOPER_ID: str
 X_CLIENT: Incomplete
 SECTION_REAUTH_LOGIN: str

homeassistant-stubs/components/habitica/coordinator.pyi

@@ -17,12 +17,13 @@ _LOGGER: Incomplete
 class HabiticaData:
     user: UserData
     tasks: list[TaskData]
+type HabiticaConfigEntry = ConfigEntry[HabiticaDataUpdateCoordinator]
 
 class HabiticaDataUpdateCoordinator(DataUpdateCoordinator[HabiticaData]):
     config_entry: ConfigEntry
     habitica: Incomplete
     content: ContentData
-    def __init__(self, hass: HomeAssistant, habitica: Habitica) -> None: ...
+    def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry, habitica: Habitica) -> None: ...
     async def _async_setup(self) -> None: ...
     async def _async_update_data(self) -> HabiticaData: ...
     async def execute(self, func: Callable[[HabiticaDataUpdateCoordinator], Any]) -> None: ...

homeassistant-stubs/components/habitica/diagnostics.pyi

@@ -1,5 +1,5 @@
 from .const import CONF_API_USER as CONF_API_USER
-from .types import HabiticaConfigEntry as HabiticaConfigEntry
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry
 from homeassistant.const import CONF_URL as CONF_URL
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any

homeassistant-stubs/components/habitica/image.pyi

@@ -1,24 +1,23 @@
-from . import HabiticaConfigEntry as HabiticaConfigEntry
-from .coordinator import HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry, HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
 from .entity import HabiticaBase as HabiticaBase
 from _typeshed import Incomplete
 from enum import StrEnum
 from habiticalib import Avatar as Avatar
 from homeassistant.components.image import ImageEntity as ImageEntity, ImageEntityDescription as ImageEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 
 PARALLEL_UPDATES: int
 
 class HabiticaImageEntity(StrEnum):
     AVATAR = 'avatar'
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: HabiticaConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: HabiticaConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class HabiticaImage(HabiticaBase, ImageEntity):
     entity_description: Incomplete
     _attr_content_type: str
-    _current_appearance: Avatar | None
+    _avatar: Avatar | None
     _cache: bytes | None
     _attr_image_last_updated: Incomplete
     def __init__(self, hass: HomeAssistant, coordinator: HabiticaDataUpdateCoordinator) -> None: ...

homeassistant-stubs/components/habitica/sensor.pyi

@@ -1,7 +1,6 @@
 from .const import ASSETS_URL as ASSETS_URL, DOMAIN as DOMAIN
-from .coordinator import HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry, HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
 from .entity import HabiticaBase as HabiticaBase
-from .types import HabiticaConfigEntry as HabiticaConfigEntry
 from .util import get_attribute_points as get_attribute_points, get_attributes_total as get_attributes_total, inventory_list as inventory_list
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Mapping
@@ -12,7 +11,7 @@ from homeassistant.components.automation import automations_with_entity as autom
 from homeassistant.components.script import scripts_with_entity as scripts_with_entity
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue, async_delete_issue as async_delete_issue
 from homeassistant.helpers.typing import StateType as StateType
 from typing import Any
@@ -62,7 +61,7 @@ TASKS_MAP: Incomplete
 TASK_SENSOR_DESCRIPTION: tuple[HabiticaTaskSensorEntityDescription, ...]
 
 def entity_used_in(hass: HomeAssistant, entity_id: str) -> list[str]: ...
-async def async_setup_entry(hass: HomeAssistant, config_entry: HabiticaConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: HabiticaConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class HabiticaSensor(HabiticaBase, SensorEntity):
     entity_description: HabiticaSensorEntityDescription

homeassistant-stubs/components/habitica/services.pyi

@@ -1,7 +1,8 @@
-from .const import ATTR_ARGS as ATTR_ARGS, ATTR_CONFIG_ENTRY as ATTR_CONFIG_ENTRY, ATTR_DATA as ATTR_DATA, ATTR_DIRECTION as ATTR_DIRECTION, ATTR_ITEM as ATTR_ITEM, ATTR_KEYWORD as ATTR_KEYWORD, ATTR_PATH as ATTR_PATH, ATTR_PRIORITY as ATTR_PRIORITY, ATTR_SKILL as ATTR_SKILL, ATTR_TAG as ATTR_TAG, ATTR_TARGET as ATTR_TARGET, ATTR_TASK as ATTR_TASK, ATTR_TYPE as ATTR_TYPE, DOMAIN as DOMAIN, EVENT_API_CALL_SUCCESS as EVENT_API_CALL_SUCCESS, SERVICE_ABORT_QUEST as SERVICE_ABORT_QUEST, SERVICE_ACCEPT_QUEST as SERVICE_ACCEPT_QUEST, SERVICE_API_CALL as SERVICE_API_CALL, SERVICE_CANCEL_QUEST as SERVICE_CANCEL_QUEST, SERVICE_CAST_SKILL as SERVICE_CAST_SKILL, SERVICE_GET_TASKS as SERVICE_GET_TASKS, SERVICE_LEAVE_QUEST as SERVICE_LEAVE_QUEST, SERVICE_REJECT_QUEST as SERVICE_REJECT_QUEST, SERVICE_SCORE_HABIT as SERVICE_SCORE_HABIT, SERVICE_SCORE_REWARD as SERVICE_SCORE_REWARD, SERVICE_START_QUEST as SERVICE_START_QUEST, SERVICE_TRANSFORMATION as SERVICE_TRANSFORMATION
-from .types import HabiticaConfigEntry as HabiticaConfigEntry
+from .const import ATTR_ALIAS as ATTR_ALIAS, ATTR_ARGS as ATTR_ARGS, ATTR_CONFIG_ENTRY as ATTR_CONFIG_ENTRY, ATTR_COST as ATTR_COST, ATTR_DATA as ATTR_DATA, ATTR_DIRECTION as ATTR_DIRECTION, ATTR_ITEM as ATTR_ITEM, ATTR_KEYWORD as ATTR_KEYWORD, ATTR_NOTES as ATTR_NOTES, ATTR_PATH as ATTR_PATH, ATTR_PRIORITY as ATTR_PRIORITY, ATTR_REMOVE_TAG as ATTR_REMOVE_TAG, ATTR_SKILL as ATTR_SKILL, ATTR_TAG as ATTR_TAG, ATTR_TARGET as ATTR_TARGET, ATTR_TASK as ATTR_TASK, ATTR_TYPE as ATTR_TYPE, DOMAIN as DOMAIN, EVENT_API_CALL_SUCCESS as EVENT_API_CALL_SUCCESS, SERVICE_ABORT_QUEST as SERVICE_ABORT_QUEST, SERVICE_ACCEPT_QUEST as SERVICE_ACCEPT_QUEST, SERVICE_API_CALL as SERVICE_API_CALL, SERVICE_CANCEL_QUEST as SERVICE_CANCEL_QUEST, SERVICE_CAST_SKILL as SERVICE_CAST_SKILL, SERVICE_GET_TASKS as SERVICE_GET_TASKS, SERVICE_LEAVE_QUEST as SERVICE_LEAVE_QUEST, SERVICE_REJECT_QUEST as SERVICE_REJECT_QUEST, SERVICE_SCORE_HABIT as SERVICE_SCORE_HABIT, SERVICE_SCORE_REWARD as SERVICE_SCORE_REWARD, SERVICE_START_QUEST as SERVICE_START_QUEST, SERVICE_TRANSFORMATION as SERVICE_TRANSFORMATION, SERVICE_UPDATE_REWARD as SERVICE_UPDATE_REWARD
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry
 from _typeshed import Incomplete
 from habiticalib import TaskData as TaskData
+from homeassistant.components.todo import ATTR_RENAME as ATTR_RENAME
 from homeassistant.config_entries import ConfigEntryState as ConfigEntryState
 from homeassistant.const import ATTR_NAME as ATTR_NAME, CONF_NAME as CONF_NAME
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, ServiceResponse as ServiceResponse, SupportsResponse as SupportsResponse
@@ -15,6 +16,7 @@ SERVICE_CAST_SKILL_SCHEMA: Incomplete
 SERVICE_MANAGE_QUEST_SCHEMA: Incomplete
 SERVICE_SCORE_TASK_SCHEMA: Incomplete
 SERVICE_TRANSFORMATION_SCHEMA: Incomplete
+SERVICE_UPDATE_TASK_SCHEMA: Incomplete
 SERVICE_GET_TASKS_SCHEMA: Incomplete
 SKILL_MAP: Incomplete
 COST_MAP: Incomplete

homeassistant-stubs/components/habitica/switch.pyi

@@ -1,12 +1,11 @@
-from .coordinator import HabiticaData as HabiticaData, HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry, HabiticaData as HabiticaData, HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
 from .entity import HabiticaBase as HabiticaBase
-from .types import HabiticaConfigEntry as HabiticaConfigEntry
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from enum import StrEnum
 from homeassistant.components.switch import SwitchDeviceClass as SwitchDeviceClass, SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from typing import Any
 
 PARALLEL_UPDATES: int
@@ -22,7 +21,7 @@ class HabiticaSwitchEntity(StrEnum):
 
 SWTICH_DESCRIPTIONS: tuple[HabiticaSwitchEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: HabiticaConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: HabiticaConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class HabiticaSwitch(HabiticaBase, SwitchEntity):
     entity_description: HabiticaSwitchEntityDescription

homeassistant-stubs/components/habitica/todo.pyi

@@ -1,7 +1,6 @@
 from .const import ASSETS_URL as ASSETS_URL, DOMAIN as DOMAIN
-from .coordinator import HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry as HabiticaConfigEntry, HabiticaDataUpdateCoordinator as HabiticaDataUpdateCoordinator
 from .entity import HabiticaBase as HabiticaBase
-from .types import HabiticaConfigEntry as HabiticaConfigEntry
 from .util import next_due_date as next_due_date
 from _typeshed import Incomplete
 from enum import StrEnum
@@ -10,7 +9,7 @@ from homeassistant.components.todo import TodoItem as TodoItem, TodoItemStatus a
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, ServiceValidationError as ServiceValidationError
 from homeassistant.helpers.entity import EntityDescription as EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 
 _LOGGER: Incomplete
 PARALLEL_UPDATES: int
@@ -21,7 +20,7 @@ class HabiticaTodoList(StrEnum):
     TODOS = 'todos'
     REWARDS = 'rewards'
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: HabiticaConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: HabiticaConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class BaseHabiticaListEntity(HabiticaBase, TodoListEntity):
     def __init__(self, coordinator: HabiticaDataUpdateCoordinator) -> None: ...

homeassistant-stubs/components/heos/__init__.pyi

@@ -4,6 +4,7 @@ from .coordinator import HeosConfigEntry as HeosConfigEntry, HeosCoordinator as
 from _typeshed import Incomplete
 from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers import device_registry as dr
 from homeassistant.helpers.typing import ConfigType as ConfigType
 
 PLATFORMS: Incomplete
@@ -13,3 +14,4 @@ CONFIG_SCHEMA: Incomplete
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
 async def async_setup_entry(hass: HomeAssistant, entry: HeosConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: HeosConfigEntry) -> bool: ...
+async def async_remove_config_entry_device(hass: HomeAssistant, entry: HeosConfigEntry, device: dr.DeviceEntry) -> bool: ...

homeassistant-stubs/components/heos/config_flow.pyi

@@ -1,28 +1,30 @@
-from .const import DOMAIN as DOMAIN
+from .const import DOMAIN as DOMAIN, ENTRY_TITLE as ENTRY_TITLE
 from .coordinator import HeosConfigEntry as HeosConfigEntry
 from _typeshed import Incomplete
 from collections.abc import Mapping
-from homeassistant.config_entries import ConfigEntryState as ConfigEntryState, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import callback as callback
 from homeassistant.helpers import selector as selector
-from homeassistant.helpers.service_info.ssdp import ATTR_UPNP_FRIENDLY_NAME as ATTR_UPNP_FRIENDLY_NAME, SsdpServiceInfo as SsdpServiceInfo
-from pyheos import Heos
+from homeassistant.helpers.service_info.ssdp import SsdpServiceInfo as SsdpServiceInfo
 from typing import Any
 
 _LOGGER: Incomplete
 AUTH_SCHEMA: Incomplete
 
-def format_title(host: str) -> str: ...
 async def _validate_host(host: str, errors: dict[str, str]) -> bool: ...
-async def _validate_auth(user_input: dict[str, str], heos: Heos, errors: dict[str, str]) -> bool: ...
+async def _validate_auth(user_input: dict[str, str], entry: HeosConfigEntry, errors: dict[str, str]) -> bool: ...
+def _get_current_hosts(entry: HeosConfigEntry) -> set[str]: ...
 
 class HeosFlowHandler(ConfigFlow, domain=DOMAIN):
     VERSION: int
+    _discovered_host: str | None
+    def __init__(self) -> None: ...
     @staticmethod
     @callback
     def async_get_options_flow(config_entry: HeosConfigEntry) -> OptionsFlow: ...
     async def async_step_ssdp(self, discovery_info: SsdpServiceInfo) -> ConfigFlowResult: ...
+    async def async_step_confirm_discovery(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_reconfigure(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...

homeassistant-stubs/components/heos/const.pyi

@@ -1,5 +1,9 @@
 ATTR_PASSWORD: str
 ATTR_USERNAME: str
 DOMAIN: str
+ENTRY_TITLE: str
+SERVICE_GROUP_VOLUME_SET: str
+SERVICE_GROUP_VOLUME_DOWN: str
+SERVICE_GROUP_VOLUME_UP: str
 SERVICE_SIGN_IN: str
 SERVICE_SIGN_OUT: str

homeassistant-stubs/components/heos/coordinator.pyi

@@ -7,7 +7,7 @@ from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HassJob as HassJo
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers.event import async_call_later as async_call_later
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
-from pyheos import HeosNowPlayingMedia as HeosNowPlayingMedia, MediaItem as MediaItem, PlayerUpdateResult as PlayerUpdateResult
+from pyheos import HeosNowPlayingMedia as HeosNowPlayingMedia, HeosPlayer as HeosPlayer, MediaItem as MediaItem, PlayerUpdateResult as PlayerUpdateResult
 from typing import Any
 
 _LOGGER: Incomplete
@@ -16,6 +16,7 @@ type HeosConfigEntry = ConfigEntry[HeosCoordinator]
 class HeosCoordinator(DataUpdateCoordinator[None]):
     host: str
     heos: Incomplete
+    _platform_callbacks: list[Callable[[Sequence[HeosPlayer]], None]]
     _update_sources_pending: bool
     _source_list: list[str]
     _favorites: dict[int, MediaItem]
@@ -28,6 +29,8 @@ class HeosCoordinator(DataUpdateCoordinator[None]):
     async def async_setup(self) -> None: ...
     async def async_shutdown(self) -> None: ...
     def async_add_listener(self, update_callback: CALLBACK_TYPE, context: Any = None) -> Callable[[], None]: ...
+    def async_add_platform_callback(self, add_entities_callback: Callable[[Sequence[HeosPlayer]], None]) -> None: ...
+    def _async_handle_player_update_result(self, update_result: PlayerUpdateResult) -> None: ...
     async def _async_on_auth_failure(self) -> None: ...
     async def _async_on_disconnected(self) -> None: ...
     async def _async_on_reconnected(self) -> None: ...

homeassistant-stubs/components/heos/media_player.pyi

@@ -1,14 +1,16 @@
+from .const import SERVICE_GROUP_VOLUME_DOWN as SERVICE_GROUP_VOLUME_DOWN, SERVICE_GROUP_VOLUME_SET as SERVICE_GROUP_VOLUME_SET, SERVICE_GROUP_VOLUME_UP as SERVICE_GROUP_VOLUME_UP
 from .coordinator import HeosConfigEntry as HeosConfigEntry, HeosCoordinator as HeosCoordinator
 from _typeshed import Incomplete
 from collections.abc import Awaitable, Callable, Coroutine
 from datetime import datetime
 from homeassistant.components import media_source as media_source
-from homeassistant.components.media_player import ATTR_MEDIA_ENQUEUE as ATTR_MEDIA_ENQUEUE, BrowseMedia as BrowseMedia, MediaPlayerEnqueue as MediaPlayerEnqueue, MediaPlayerEntity as MediaPlayerEntity, MediaPlayerEntityFeature as MediaPlayerEntityFeature, MediaPlayerState as MediaPlayerState, MediaType as MediaType, RepeatMode as RepeatMode, async_process_play_media_url as async_process_play_media_url
+from homeassistant.components.media_player import ATTR_MEDIA_ENQUEUE as ATTR_MEDIA_ENQUEUE, ATTR_MEDIA_VOLUME_LEVEL as ATTR_MEDIA_VOLUME_LEVEL, BrowseMedia as BrowseMedia, MediaPlayerEnqueue as MediaPlayerEnqueue, MediaPlayerEntity as MediaPlayerEntity, MediaPlayerEntityFeature as MediaPlayerEntityFeature, MediaPlayerState as MediaPlayerState, MediaType as MediaType, RepeatMode as RepeatMode, async_process_play_media_url as async_process_play_media_url
 from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, ServiceValidationError as ServiceValidationError
+from homeassistant.helpers import entity_platform as entity_platform
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from homeassistant.util.dt import utcnow as utcnow
 from pyheos import HeosPlayer as HeosPlayer
@@ -22,7 +24,7 @@ HA_HEOS_ENQUEUE_MAP: Incomplete
 HEOS_HA_REPEAT_TYPE_MAP: Incomplete
 HA_HEOS_REPEAT_TYPE_MAP: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: HeosConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: HeosConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 type _FuncType[**_P] = Callable[_P, Awaitable[Any]]
 type _ReturnFuncType[**_P] = Callable[_P, Coroutine[Any, Any, None]]
 def catch_action_error[**_P](action: str) -> Callable[[_FuncType[_P]], _ReturnFuncType[_P]]: ...
@@ -62,6 +64,9 @@ class HeosMediaPlayer(CoordinatorEntity[HeosCoordinator], MediaPlayerEntity):
     async def async_set_repeat(self, repeat: RepeatMode) -> None: ...
     async def async_set_shuffle(self, shuffle: bool) -> None: ...
     async def async_set_volume_level(self, volume: float) -> None: ...
+    async def async_set_group_volume_level(self, volume_level: float) -> None: ...
+    async def async_group_volume_down(self) -> None: ...
+    async def async_group_volume_up(self) -> None: ...
     async def async_join_players(self, group_members: list[str]) -> None: ...
     async def async_unjoin_player(self) -> None: ...
     @property

homeassistant-stubs/components/here_travel_time/coordinator.pyi

@@ -2,6 +2,7 @@ from .const import DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL, DOMAIN as DOM
 from .model import HERETravelTimeConfig as HERETravelTimeConfig, HERETravelTimeData as HERETravelTimeData
 from _typeshed import Incomplete
 from datetime import datetime, time
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import UnitOfLength as UnitOfLength
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.location import find_coordinates as find_coordinates
@@ -13,17 +14,19 @@ BACKOFF_MULTIPLIER: float
 _LOGGER: Incomplete
 
 class HERERoutingDataUpdateCoordinator(DataUpdateCoordinator[HERETravelTimeData]):
+    config_entry: ConfigEntry
     _api: Incomplete
     config: Incomplete
-    def __init__(self, hass: HomeAssistant, api_key: str, config: HERETravelTimeConfig) -> None: ...
+    def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry, api_key: str, config: HERETravelTimeConfig) -> None: ...
     update_interval: Incomplete
     async def _async_update_data(self) -> HERETravelTimeData: ...
     def _parse_routing_response(self, response: dict[str, Any]) -> HERETravelTimeData: ...
 
 class HERETransitDataUpdateCoordinator(DataUpdateCoordinator[HERETravelTimeData | None]):
+    config_entry: ConfigEntry
     _api: Incomplete
     config: Incomplete
-    def __init__(self, hass: HomeAssistant, api_key: str, config: HERETravelTimeConfig) -> None: ...
+    def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry, api_key: str, config: HERETravelTimeConfig) -> None: ...
     update_interval: Incomplete
     async def _async_update_data(self) -> HERETravelTimeData | None: ...
     def _parse_transit_response(self, response: dict[str, Any]) -> HERETravelTimeData: ...

homeassistant-stubs/components/here_travel_time/sensor.pyi

@@ -7,14 +7,14 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_ATTRIBUTION as ATTR_ATTRIBUTION, ATTR_LATITUDE as ATTR_LATITUDE, ATTR_LONGITUDE as ATTR_LONGITUDE, CONF_MODE as CONF_MODE, CONF_NAME as CONF_NAME, UnitOfLength as UnitOfLength, UnitOfTime as UnitOfTime
 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.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from typing import Any
 
 SCAN_INTERVAL: Incomplete
 
 def sensor_descriptions(travel_mode: str) -> tuple[SensorEntityDescription, ...]: ...
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class HERETravelTimeSensor(CoordinatorEntity[HERERoutingDataUpdateCoordinator | HERETransitDataUpdateCoordinator], RestoreSensor):
     _attr_has_entity_name: bool