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

homeassistant-stubs/exceptions.pyi

@@ -1,9 +1,10 @@
 from .core import Context as Context
 from .util.event_type import EventType as EventType
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Generator, Sequence
+from collections.abc import Callable as Callable, Sequence
 from dataclasses import dataclass
 from typing import Any
+from typing_extensions import Generator
 
 _function_cache: dict[str, Callable[[str, str, dict[str, str] | None], str]]
 
@@ -34,14 +35,14 @@ class ConditionError(HomeAssistantError):
     type: str
     @staticmethod
     def _indent(indent: int, message: str) -> str: ...
-    def output(self, indent: int) -> Generator[str, None, None]: ...
+    def output(self, indent: int) -> Generator[str]: ...
     def __str__(self) -> str: ...
     def __init__(self, type) -> None: ...
 
 @dataclass(slots=True)
 class ConditionErrorMessage(ConditionError):
     message: str
-    def output(self, indent: int) -> Generator[str, None, None]: ...
+    def output(self, indent: int) -> Generator[str]: ...
     def __init__(self, type, message) -> None: ...
 
 @dataclass(slots=True)
@@ -49,13 +50,13 @@ class ConditionErrorIndex(ConditionError):
     index: int
     total: int
     error: ConditionError
-    def output(self, indent: int) -> Generator[str, None, None]: ...
+    def output(self, indent: int) -> Generator[str]: ...
     def __init__(self, type, index, total, error) -> None: ...
 
 @dataclass(slots=True)
 class ConditionErrorContainer(ConditionError):
     errors: Sequence[ConditionError]
-    def output(self, indent: int) -> Generator[str, None, None]: ...
+    def output(self, indent: int) -> Generator[str]: ...
     def __init__(self, type, errors) -> None: ...
 
 class IntegrationError(HomeAssistantError):

homeassistant-stubs/helpers/__init__.pyi

@@ -1,5 +0,0 @@
-from .typing import ConfigType as ConfigType
-from collections.abc import Iterable, Sequence
-
-def config_per_platform(config: ConfigType, domain: str) -> Iterable[tuple[str | None, ConfigType]]: ...
-def extract_domain_configs(config: ConfigType, domain: str) -> Sequence[str]: ...

homeassistant-stubs/helpers/collection.pyi

@@ -4,14 +4,14 @@ from . import entity_registry as entity_registry
 from .entity import Entity as Entity
 from .entity_component import EntityComponent as EntityComponent
 from .storage import Store as Store
-from .typing import ConfigType as ConfigType
+from .typing import ConfigType as ConfigType, VolDictType as VolDictType
 from _typeshed import Incomplete
 from abc import ABC, abstractmethod
 from collections.abc import Awaitable, Callable, Iterable
 from dataclasses import dataclass
 from homeassistant.components import websocket_api as websocket_api
 from homeassistant.const import CONF_ID as CONF_ID
-from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.util import slugify as slugify
 from typing import Any, Generic, TypedDict
@@ -25,13 +25,13 @@ CHANGE_REMOVED: str
 _EntityT = TypeVar('_EntityT', bound=Entity, default=Entity)
 
 @dataclass(slots=True)
-class CollectionChangeSet:
+class CollectionChange:
     change_type: str
     item_id: str
     item: Any
     def __init__(self, change_type, item_id, item) -> None: ...
 ChangeListener = Callable[[str, str, dict], Awaitable[None]]
-ChangeSetListener = Callable[[Iterable[CollectionChangeSet]], Awaitable[None]]
+ChangeSetListener = Callable[[Iterable[CollectionChange]], Awaitable[None]]
 
 class CollectionError(HomeAssistantError): ...
 
@@ -65,7 +65,7 @@ class ObservableCollection(ABC):
     def async_items(self) -> list[_ItemT]: ...
     def async_add_listener(self, listener: ChangeListener) -> Callable[[], None]: ...
     def async_add_change_set_listener(self, listener: ChangeSetListener) -> Callable[[], None]: ...
-    async def notify_changes(self, change_sets: Iterable[CollectionChangeSet]) -> None: ...
+    async def notify_changes(self, change_set: Iterable[CollectionChange]) -> None: ...
 
 class YamlCollection(ObservableCollection[dict]):
     logger: Incomplete
@@ -129,10 +129,10 @@ class _CollectionLifeCycle(Generic[_EntityT]):
     entities: dict[str, CollectionEntity]
     def async_setup(self) -> None: ...
     def _entity_removed(self, item_id: str) -> None: ...
-    def _add_entity(self, change_set: CollectionChangeSet) -> CollectionEntity: ...
-    async def _remove_entity(self, change_set: CollectionChangeSet) -> None: ...
-    async def _update_entity(self, change_set: CollectionChangeSet) -> None: ...
-    async def _collection_changed(self, change_sets: Iterable[CollectionChangeSet]) -> None: ...
+    def _add_entity(self, change_set: CollectionChange) -> CollectionEntity: ...
+    async def _remove_entity(self, change_set: CollectionChange) -> None: ...
+    async def _update_entity(self, change_set: CollectionChange) -> None: ...
+    async def _collection_changed(self, change_set: Iterable[CollectionChange]) -> None: ...
     def __init__(self, domain, platform, entity_component, collection, entity_class, ent_reg, entities) -> None: ...
 
 def sync_entity_lifecycle(hass: HomeAssistant, domain: str, platform: str, entity_component: EntityComponent[_EntityT], collection: StorageCollection | YamlCollection, entity_class: type[CollectionEntity]) -> None: ...
@@ -143,12 +143,15 @@ class StorageCollectionWebsocket:
     model_name: Incomplete
     create_schema: Incomplete
     update_schema: Incomplete
-    def __init__(self, storage_collection: _StorageCollectionT, api_prefix: str, model_name: str, create_schema: dict, update_schema: dict) -> None: ...
+    _remove_subscription: Incomplete
+    _subscribers: Incomplete
+    def __init__(self, storage_collection: _StorageCollectionT, api_prefix: str, model_name: str, create_schema: VolDictType, update_schema: VolDictType) -> None: ...
     @property
     def item_id_key(self) -> str: ...
-    def async_setup(self, hass: HomeAssistant, *, create_list: bool = True, create_create: bool = True) -> None: ...
+    def async_setup(self, hass: HomeAssistant) -> None: ...
     def ws_list_item(self, hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict) -> None: ...
     async def ws_create_item(self, hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict) -> None: ...
+    def _ws_subscribe(self, hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict) -> None: ...
     async def ws_update_item(self, hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict) -> None: ...
     async def ws_delete_item(self, hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict) -> None: ...
 

homeassistant-stubs/helpers/condition.pyi

@@ -4,7 +4,7 @@ from .template import Template as Template, render_complex as render_complex
 from .trace import TraceElement as TraceElement, trace_append_element as trace_append_element, trace_path as trace_path, trace_path_get as trace_path_get, trace_stack_cv as trace_stack_cv, trace_stack_pop as trace_stack_pop, trace_stack_push as trace_stack_push, trace_stack_top as trace_stack_top
 from .typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsType
 from _typeshed import Incomplete
-from collections.abc import Container, Generator
+from collections.abc import Container
 from datetime import time as dt_time, timedelta
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass
 from homeassistant.const import ATTR_DEVICE_CLASS as ATTR_DEVICE_CLASS, ATTR_GPS_ACCURACY as ATTR_GPS_ACCURACY, ATTR_LATITUDE as ATTR_LATITUDE, ATTR_LONGITUDE as ATTR_LONGITUDE, CONF_ABOVE as CONF_ABOVE, CONF_AFTER as CONF_AFTER, CONF_ATTRIBUTE as CONF_ATTRIBUTE, CONF_BEFORE as CONF_BEFORE, CONF_BELOW as CONF_BELOW, CONF_CONDITION as CONF_CONDITION, CONF_DEVICE_ID as CONF_DEVICE_ID, CONF_ENABLED as CONF_ENABLED, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_FOR as CONF_FOR, CONF_ID as CONF_ID, CONF_MATCH as CONF_MATCH, CONF_STATE as CONF_STATE, CONF_VALUE_TEMPLATE as CONF_VALUE_TEMPLATE, CONF_WEEKDAY as CONF_WEEKDAY, CONF_ZONE as CONF_ZONE, ENTITY_MATCH_ALL as ENTITY_MATCH_ALL, ENTITY_MATCH_ANY as ENTITY_MATCH_ANY, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN, SUN_EVENT_SUNRISE as SUN_EVENT_SUNRISE, SUN_EVENT_SUNSET as SUN_EVENT_SUNSET, WEEKDAYS as WEEKDAYS
@@ -13,6 +13,7 @@ from homeassistant.exceptions import ConditionError as ConditionError, Condition
 from homeassistant.loader import IntegrationNotFound as IntegrationNotFound, async_get_integration as async_get_integration
 from homeassistant.util.async_ import run_callback_threadsafe as run_callback_threadsafe
 from typing import Any, Protocol
+from typing_extensions import Generator
 
 ASYNC_FROM_CONFIG_FORMAT: str
 FROM_CONFIG_FORMAT: str
@@ -30,7 +31,7 @@ ConditionCheckerType: Incomplete
 def condition_trace_append(variables: TemplateVarsType, path: str) -> TraceElement: ...
 def condition_trace_set_result(result: bool, **kwargs: Any) -> None: ...
 def condition_trace_update_result(**kwargs: Any) -> None: ...
-def trace_condition(variables: TemplateVarsType) -> Generator[TraceElement, None, None]: ...
+def trace_condition(variables: TemplateVarsType) -> Generator[TraceElement]: ...
 def trace_condition_function(condition: ConditionCheckerType) -> ConditionCheckerType: ...
 async def _async_get_condition_platform(hass: HomeAssistant, config: ConfigType) -> ConditionProtocol | None: ...
 async def async_from_config(hass: HomeAssistant, config: ConfigType) -> ConditionCheckerType: ...

homeassistant-stubs/helpers/config_validation.pyi

@@ -2,6 +2,7 @@ import re
 import voluptuous as vol
 from . import template as template_helper
 from .frame import get_integration_logger as get_integration_logger
+from .typing import VolDictType as VolDictType, VolSchemaType as VolSchemaType
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Hashable
 from datetime import date as date_sys, datetime as datetime_sys, time as time_sys, timedelta
@@ -126,8 +127,8 @@ class multi_select:
 def _deprecated_or_removed(key: str, replacement_key: str | None, default: Any | None, raise_if_present: bool, option_removed: bool) -> Callable[[dict], dict]: ...
 def deprecated(key: str, replacement_key: str | None = None, default: Any | None = None, raise_if_present: bool | None = False) -> Callable[[dict], dict]: ...
 def removed(key: str, default: Any | None = None, raise_if_present: bool | None = True) -> Callable[[dict], dict]: ...
-def key_value_schemas(key: str, value_schemas: dict[Hashable, vol.Schema], default_schema: vol.Schema | None = None, default_description: str | None = None) -> Callable[[Any], dict[Hashable, Any]]: ...
-def key_dependency(key: Hashable, dependency: Hashable) -> Callable[[dict[Hashable, Any]], dict[Hashable, Any]]: ...
+def key_value_schemas(key: str, value_schemas: dict[Hashable, VolSchemaType | Callable[[Any], dict[str, Any]]], default_schema: VolSchemaType | None = None, default_description: str | None = None) -> Callable[[Any], dict[Hashable, Any]]: ...
+def key_dependency(key: Hashable, dependency: Hashable) -> Callable[[dict[_KT, _VT]], dict[_KT, _VT]]: ...
 def custom_serializer(schema: Any) -> Any: ...
 def expand_condition_shorthand(value: Any | None) -> Any: ...
 def empty_config_schema(domain: str) -> Callable[[dict], dict]: ...
@@ -157,13 +158,13 @@ SCRIPT_ACTION_BASE_SCHEMA: Incomplete
 EVENT_SCHEMA: Incomplete
 SERVICE_SCHEMA: Incomplete
 NUMERIC_STATE_THRESHOLD_SCHEMA: Incomplete
-CONDITION_BASE_SCHEMA: Incomplete
+CONDITION_BASE_SCHEMA: VolDictType
 NUMERIC_STATE_CONDITION_SCHEMA: Incomplete
 STATE_CONDITION_BASE_SCHEMA: Incomplete
 STATE_CONDITION_STATE_SCHEMA: Incomplete
 STATE_CONDITION_ATTRIBUTE_SCHEMA: Incomplete
 
-def STATE_CONDITION_SCHEMA(value: Any) -> dict: ...
+def STATE_CONDITION_SCHEMA(value: Any) -> dict[str, Any]: ...
 
 SUN_CONDITION_SCHEMA: Incomplete
 TEMPLATE_CONDITION_SCHEMA: Incomplete

homeassistant-stubs/helpers/deprecation.pyi

@@ -1,3 +1,5 @@
+import functools
+from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from enum import Enum
 from typing import Any, NamedTuple
@@ -23,6 +25,14 @@ class DeprecatedAlias(NamedTuple):
     replacement: str
     breaks_in_ha_version: str | None
 
+class DeferredDeprecatedAlias:
+    breaks_in_ha_version: Incomplete
+    replacement: Incomplete
+    _value_fn: Incomplete
+    def __init__(self, value_fn: Callable[[], Any], replacement: str, breaks_in_ha_version: str | None) -> None: ...
+    @functools.cached_property
+    def value(self) -> Any: ...
+
 _PREFIX_DEPRECATED: str
 
 def check_if_deprecated_constant(name: str, module_globals: dict[str, Any]) -> Any: ...

homeassistant-stubs/helpers/device.pyi

@@ -0,0 +1,8 @@
+from . import device_registry as dr
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+
+def async_entity_id_to_device_id(hass: HomeAssistant, entity_id_or_uuid: str) -> str | None: ...
+def async_device_info_to_link_from_entity(hass: HomeAssistant, entity_id_or_uuid: str) -> dr.DeviceInfo | None: ...
+def async_device_info_to_link_from_device_id(hass: HomeAssistant, device_id: str | None) -> dr.DeviceInfo | None: ...
+def async_remove_stale_devices_links_keep_entity_device(hass: HomeAssistant, entry_id: str, source_entity_id_or_uuid: str) -> None: ...
+def async_remove_stale_devices_links_keep_current_device(hass: HomeAssistant, entry_id: str, current_device_id: str | None) -> None: ...

homeassistant-stubs/helpers/device_registry.pyi

@@ -66,6 +66,7 @@ class DeviceInfo(TypedDict, total=False):
 
 DEVICE_INFO_TYPES: Incomplete
 DEVICE_INFO_KEYS: Incomplete
+LOW_PRIO_CONFIG_ENTRY_DOMAINS: Incomplete
 
 class _EventDeviceRegistryUpdatedData_CreateRemove(TypedDict):
     action: Literal['create', 'remove']
@@ -86,6 +87,14 @@ class DeviceInfoError(HomeAssistantError):
     domain: Incomplete
     def __init__(self, domain: str, device_info: DeviceInfo, message: str) -> None: ...
 
+class DeviceCollisionError(HomeAssistantError): ...
+
+class DeviceIdentifierCollisionError(DeviceCollisionError):
+    def __init__(self, identifiers: set[tuple[str, str]], existing_device: DeviceEntry) -> None: ...
+
+class DeviceConnectionCollisionError(DeviceCollisionError):
+    def __init__(self, normalized_connections: set[tuple[str, str]], existing_device: DeviceEntry) -> None: ...
+
 def _validate_device_info(config_entry: ConfigEntry, device_info: DeviceInfo) -> str: ...
 
 _cached_parse_url: Incomplete
@@ -107,6 +116,7 @@ class DeviceEntry:
     model: str | None
     name_by_user: str | None
     name: str | None
+    primary_config_entry: str | None
     serial_number: str | None
     suggested_area: str | None
     sw_version: str | None
@@ -120,7 +130,7 @@ class DeviceEntry:
     def json_repr(self) -> bytes | None: ...
     @cached_property
     def as_storage_fragment(self) -> json_fragment: ...
-    def __init__(self, area_id, config_entries, configuration_url, connections, disabled_by, entry_type, hw_version, id, identifiers, labels, manufacturer, model, name_by_user, name, serial_number, suggested_area, sw_version, via_device_id, is_new) -> None: ...
+    def __init__(self, area_id, config_entries, configuration_url, connections, disabled_by, entry_type, hw_version, id, identifiers, labels, manufacturer, model, name_by_user, name, primary_config_entry, serial_number, suggested_area, sw_version, via_device_id, is_new) -> None: ...
     def __lt__(self, other): ...
     def __le__(self, other): ...
     def __gt__(self, other): ...
@@ -177,7 +187,9 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
     def _async_get_deleted_device(self, identifiers: set[tuple[str, str]], connections: set[tuple[str, str]]) -> DeletedDeviceEntry | None: ...
     def _substitute_name_placeholders(self, domain: str, name: str, translation_placeholders: Mapping[str, str]) -> str: ...
     def async_get_or_create(self, *, config_entry_id: str, configuration_url: str | URL | None | UndefinedType = ..., connections: set[tuple[str, str]] | None | UndefinedType = ..., default_manufacturer: str | None | UndefinedType = ..., default_model: str | None | UndefinedType = ..., default_name: str | None | UndefinedType = ..., disabled_by: DeviceEntryDisabler | None | UndefinedType = ..., entry_type: DeviceEntryType | None | UndefinedType = ..., hw_version: str | None | UndefinedType = ..., identifiers: set[tuple[str, str]] | None | UndefinedType = ..., manufacturer: str | None | UndefinedType = ..., model: str | None | UndefinedType = ..., name: str | None | UndefinedType = ..., serial_number: str | None | UndefinedType = ..., suggested_area: str | None | UndefinedType = ..., sw_version: str | None | UndefinedType = ..., translation_key: str | None = None, translation_placeholders: Mapping[str, str] | None = None, via_device: tuple[str, str] | None | UndefinedType = ...) -> DeviceEntry: ...
-    def async_update_device(self, device_id: str, *, add_config_entry_id: str | UndefinedType = ..., area_id: str | None | UndefinedType = ..., configuration_url: str | URL | None | UndefinedType = ..., disabled_by: DeviceEntryDisabler | None | UndefinedType = ..., entry_type: DeviceEntryType | None | UndefinedType = ..., hw_version: str | None | UndefinedType = ..., labels: set[str] | UndefinedType = ..., manufacturer: str | None | UndefinedType = ..., merge_connections: set[tuple[str, str]] | UndefinedType = ..., merge_identifiers: set[tuple[str, str]] | UndefinedType = ..., model: str | None | UndefinedType = ..., name_by_user: str | None | UndefinedType = ..., name: str | None | UndefinedType = ..., new_connections: set[tuple[str, str]] | UndefinedType = ..., new_identifiers: set[tuple[str, str]] | UndefinedType = ..., remove_config_entry_id: str | UndefinedType = ..., serial_number: str | None | UndefinedType = ..., suggested_area: str | None | UndefinedType = ..., sw_version: str | None | UndefinedType = ..., via_device_id: str | None | UndefinedType = ...) -> DeviceEntry | None: ...
+    def async_update_device(self, device_id: str, *, add_config_entry: ConfigEntry | UndefinedType = ..., add_config_entry_id: str | UndefinedType = ..., allow_collisions: bool = False, area_id: str | None | UndefinedType = ..., configuration_url: str | URL | None | UndefinedType = ..., device_info_type: str | UndefinedType = ..., disabled_by: DeviceEntryDisabler | None | UndefinedType = ..., entry_type: DeviceEntryType | None | UndefinedType = ..., hw_version: str | None | UndefinedType = ..., labels: set[str] | UndefinedType = ..., manufacturer: str | None | UndefinedType = ..., merge_connections: set[tuple[str, str]] | UndefinedType = ..., merge_identifiers: set[tuple[str, str]] | UndefinedType = ..., model: str | None | UndefinedType = ..., name_by_user: str | None | UndefinedType = ..., name: str | None | UndefinedType = ..., new_connections: set[tuple[str, str]] | UndefinedType = ..., new_identifiers: set[tuple[str, str]] | UndefinedType = ..., remove_config_entry_id: str | UndefinedType = ..., serial_number: str | None | UndefinedType = ..., suggested_area: str | None | UndefinedType = ..., sw_version: str | None | UndefinedType = ..., via_device_id: str | None | UndefinedType = ...) -> DeviceEntry | None: ...
+    def _validate_connections(self, device_id: str, connections: set[tuple[str, str]], allow_collisions: bool) -> set[tuple[str, str]]: ...
+    def _validate_identifiers(self, device_id: str, identifiers: set[tuple[str, str]], allow_collisions: bool) -> set[tuple[str, str]]: ...
     def async_remove_device(self, device_id: str) -> None: ...
     async def async_load(self) -> None: ...
     def _data_to_save(self) -> dict[str, Any]: ...

homeassistant-stubs/helpers/entity.pyi

@@ -5,6 +5,7 @@ from . import device_registry as dr, entity_registry as er, singleton as singlet
 from .device_registry import DeviceInfo as DeviceInfo, EventDeviceRegistryUpdatedData as EventDeviceRegistryUpdatedData
 from .entity_platform import EntityPlatform as EntityPlatform
 from .event import async_track_device_registry_updated_event as async_track_device_registry_updated_event, async_track_entity_registry_updated_event as async_track_entity_registry_updated_event
+from .frame import report_non_thread_safe_operation as report_non_thread_safe_operation
 from .typing import StateType as StateType, UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
 from abc import ABCMeta
@@ -98,6 +99,7 @@ class Entity(cached_properties=CACHED_PROPERTIES_WITH_ATTR_, metaclass=ABCCached
     _no_platform_reported: bool
     _name_translation_placeholders_reported: bool
     _update_staged: bool
+    _verified_state_writable: bool
     parallel_updates: asyncio.Semaphore | None
     registry_entry: er.RegistryEntry | None
     _removed_from_registry: bool
@@ -112,7 +114,6 @@ class Entity(cached_properties=CACHED_PROPERTIES_WITH_ATTR_, metaclass=ABCCached
     __combined_unrecorded_attributes: frozenset[str]
     _job_types: dict[str, HassJobType] | None
     _state_info: StateInfo
-    _is_custom_component: bool
     __capabilities_updated_at: deque[float]
     __capabilities_updated_at_reported: bool
     __remove_future: asyncio.Future[None] | None
@@ -144,7 +145,7 @@ class Entity(cached_properties=CACHED_PROPERTIES_WITH_ATTR_, metaclass=ABCCached
     def should_poll(self) -> bool: ...
     @cached_property
     def unique_id(self) -> str | None: ...
-    @property
+    @cached_property
     def use_device_name(self) -> bool: ...
     @cached_property
     def has_entity_name(self) -> bool: ...

homeassistant-stubs/helpers/entity_component.pyi

@@ -1,8 +1,7 @@
 import logging
-import voluptuous as vol
 from . import discovery as discovery, entity as entity, service as service
 from .entity_platform import EntityPlatform as EntityPlatform
-from .typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
+from .typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType, VolDictType as VolDictType, VolSchemaType as VolSchemaType
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Iterable
 from datetime import timedelta
@@ -43,8 +42,8 @@ class EntityComponent(Generic[_EntityT]):
     async def async_setup_entry(self, config_entry: ConfigEntry) -> bool: ...
     async def async_unload_entry(self, config_entry: ConfigEntry) -> bool: ...
     async def async_extract_from_service(self, service_call: ServiceCall, expand_group: bool = True) -> list[_EntityT]: ...
-    def async_register_legacy_entity_service(self, name: str, schema: dict[str | vol.Marker, Any] | vol.Schema, func: str | Callable[..., Any], required_features: list[int] | None = None, supports_response: SupportsResponse = ...) -> None: ...
-    def async_register_entity_service(self, name: str, schema: dict[str | vol.Marker, Any] | vol.Schema, func: str | Callable[..., Any], required_features: list[int] | None = None, supports_response: SupportsResponse = ...) -> None: ...
+    def async_register_legacy_entity_service(self, name: str, schema: VolDictType | VolSchemaType, func: str | Callable[..., Any], required_features: list[int] | None = None, supports_response: SupportsResponse = ...) -> None: ...
+    def async_register_entity_service(self, name: str, schema: VolDictType | VolSchemaType, func: str | Callable[..., Any], required_features: list[int] | None = None, supports_response: SupportsResponse = ...) -> None: ...
     async def async_setup_platform(self, platform_type: str, platform_config: ConfigType, discovery_info: DiscoveryInfoType | None = None) -> None: ...
     async def _async_reset(self) -> None: ...
     async def async_remove_entity(self, entity_id: str) -> None: ...

homeassistant-stubs/helpers/entity_platform.pyi

@@ -1,11 +1,10 @@
 import asyncio
-import voluptuous as vol
 from . import service as service, translation as translation
 from .entity import Entity as Entity
 from .entity_registry import EntityRegistry as EntityRegistry, RegistryEntryDisabler as RegistryEntryDisabler, RegistryEntryHider as RegistryEntryHider
 from .event import async_call_later as async_call_later
 from .issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue
-from .typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType, UNDEFINED as UNDEFINED
+from .typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType, UNDEFINED as UNDEFINED, VolDictType as VolDictType, VolSchemaType as VolSchemaType
 from _typeshed import Incomplete
 from collections.abc import Awaitable, Callable as Callable, Coroutine, Iterable
 from contextvars import ContextVar
@@ -91,7 +90,7 @@ class EntityPlatform:
     async def async_destroy(self) -> None: ...
     async def async_remove_entity(self, entity_id: str) -> None: ...
     async def async_extract_from_service(self, service_call: ServiceCall, expand_group: bool = True) -> list[Entity]: ...
-    def async_register_entity_service(self, name: str, schema: dict[str | vol.Marker, Any] | vol.Schema, func: str | Callable[..., Any], required_features: Iterable[int] | None = None, supports_response: SupportsResponse = ...) -> None: ...
+    def async_register_entity_service(self, name: str, schema: VolDictType | VolSchemaType | None, func: str | Callable[..., Any], required_features: Iterable[int] | None = None, supports_response: SupportsResponse = ...) -> None: ...
     async def _async_update_entity_states(self) -> None: ...
 
 current_platform: ContextVar[EntityPlatform | None]

homeassistant-stubs/helpers/event.pyi

@@ -12,8 +12,8 @@ from collections import defaultdict
 from collections.abc import Callable, Coroutine, Iterable, Mapping, Sequence
 from dataclasses import dataclass
 from datetime import datetime, timedelta
-from homeassistant.const import EVENT_CORE_CONFIG_UPDATE as EVENT_CORE_CONFIG_UPDATE, EVENT_STATE_CHANGED as EVENT_STATE_CHANGED, MATCH_ALL as MATCH_ALL, SUN_EVENT_SUNRISE as SUN_EVENT_SUNRISE, SUN_EVENT_SUNSET as SUN_EVENT_SUNSET
-from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, EventStateChangedData as EventStateChangedData, HassJob as HassJob, HassJobType as HassJobType, HomeAssistant as HomeAssistant, State as State, callback as callback, split_entity_id as split_entity_id
+from homeassistant.const import EVENT_CORE_CONFIG_UPDATE as EVENT_CORE_CONFIG_UPDATE, EVENT_STATE_CHANGED as EVENT_STATE_CHANGED, EVENT_STATE_REPORTED as EVENT_STATE_REPORTED, MATCH_ALL as MATCH_ALL, SUN_EVENT_SUNRISE as SUN_EVENT_SUNRISE, SUN_EVENT_SUNSET as SUN_EVENT_SUNSET
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, EventStateChangedData as EventStateChangedData, EventStateEventData as EventStateEventData, EventStateReportedData as EventStateReportedData, HassJob as HassJob, HassJobType as HassJobType, HomeAssistant as HomeAssistant, State as State, callback as callback, split_entity_id as split_entity_id
 from homeassistant.exceptions import TemplateError as TemplateError
 from homeassistant.loader import bind_hass as bind_hass
 from homeassistant.util.async_ import run_callback_threadsafe as run_callback_threadsafe
@@ -22,6 +22,7 @@ from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Any, Concatenate, Generic, TypeVar
 
 _TRACK_STATE_CHANGE_DATA: HassKey[_KeyedEventData[EventStateChangedData]]
+_TRACK_STATE_REPORT_DATA: HassKey[_KeyedEventData[EventStateReportedData]]
 _TRACK_STATE_ADDED_DOMAIN_DATA: HassKey[_KeyedEventData[EventStateChangedData]]
 _TRACK_STATE_REMOVED_DOMAIN_DATA: HassKey[_KeyedEventData[EventStateChangedData]]
 _TRACK_ENTITY_REGISTRY_UPDATED_DATA: HassKey[_KeyedEventData[EventEntityRegistryUpdatedData]]
@@ -33,6 +34,7 @@ _LOGGER: Incomplete
 RANDOM_MICROSECOND_MIN: int
 RANDOM_MICROSECOND_MAX: int
 _TypedDictT = TypeVar('_TypedDictT', bound=Mapping[str, Any])
+_StateEventDataT = TypeVar('_StateEventDataT', bound=EventStateEventData)
 
 @dataclass(slots=True, frozen=True)
 class _KeyedEventTracker(Generic[_TypedDictT]):
@@ -75,12 +77,16 @@ def async_track_state_change(hass: HomeAssistant, entity_ids: str | Iterable[str
 track_state_change: Incomplete
 
 def async_track_state_change_event(hass: HomeAssistant, entity_ids: str | Iterable[str], action: Callable[[Event[EventStateChangedData]], Any], job_type: HassJobType | None = None) -> CALLBACK_TYPE: ...
-def _async_dispatch_entity_id_event(hass: HomeAssistant, callbacks: dict[str, list[HassJob[[Event[EventStateChangedData]], Any]]], event: Event[EventStateChangedData]) -> None: ...
-def _async_state_change_filter(hass: HomeAssistant, callbacks: dict[str, list[HassJob[[Event[EventStateChangedData]], Any]]], event_data: EventStateChangedData) -> bool: ...
+def _async_dispatch_entity_id_event(hass: HomeAssistant, callbacks: dict[str, list[HassJob[[Event[_StateEventDataT]], Any]]], event: Event[_StateEventDataT]) -> None: ...
+def _async_state_filter(hass: HomeAssistant, callbacks: dict[str, list[HassJob[[Event[_StateEventDataT]], Any]]], event_data: _StateEventDataT) -> bool: ...
 
 _KEYED_TRACK_STATE_CHANGE: Incomplete
 
 def _async_track_state_change_event(hass: HomeAssistant, entity_ids: str | Iterable[str], action: Callable[[Event[EventStateChangedData]], Any], job_type: HassJobType | None) -> CALLBACK_TYPE: ...
+
+_KEYED_TRACK_STATE_REPORT: Incomplete
+
+def async_track_state_report_event(hass: HomeAssistant, entity_ids: str | Iterable[str], action: Callable[[Event[EventStateReportedData]], Any], job_type: HassJobType | None = None) -> CALLBACK_TYPE: ...
 def _remove_empty_listener() -> None: ...
 def _remove_listener(hass: HomeAssistant, tracker: _KeyedEventTracker[_TypedDictT], keys: Iterable[str], job: HassJob[[Event[_TypedDictT]], Any], callbacks: dict[str, list[HassJob[[Event[_TypedDictT]], Any]]]) -> None: ...
 def _async_track_event(tracker: _KeyedEventTracker[_TypedDictT], hass: HomeAssistant, keys: str | Iterable[str], action: Callable[[Event[_TypedDictT]], None], job_type: HassJobType | None) -> CALLBACK_TYPE: ...

homeassistant-stubs/helpers/frame.pyi

@@ -35,3 +35,4 @@ class MissingIntegrationFrame(HomeAssistantError): ...
 def report(what: str, exclude_integrations: set | None = None, error_if_core: bool = True, level: int = ..., log_custom_component_only: bool = False, error_if_integration: bool = False) -> None: ...
 def _report_integration(what: str, integration_frame: IntegrationFrame, level: int = ..., error: bool = False) -> None: ...
 def warn_use(func: _CallableT, what: str) -> _CallableT: ...
+def report_non_thread_safe_operation(what: str) -> None: ...

homeassistant-stubs/helpers/http.pyi

@@ -16,7 +16,7 @@ _LOGGER: Incomplete
 AllowCorsType: Incomplete
 KEY_AUTHENTICATED: Final[str]
 KEY_ALLOW_ALL_CORS: Incomplete
-KEY_ALLOW_CONFIGRED_CORS: Incomplete
+KEY_ALLOW_CONFIGURED_CORS: Incomplete
 KEY_HASS: AppKey[HomeAssistant]
 current_request: ContextVar[Request | None]
 

homeassistant-stubs/helpers/intent.pyi

@@ -2,6 +2,7 @@ import abc
 import asyncio
 import voluptuous as vol
 from . import area_registry as area_registry, device_registry as device_registry, entity_registry as entity_registry, floor_registry as floor_registry
+from .typing import VolSchemaType as VolSchemaType
 from _typeshed import Incomplete
 from abc import abstractmethod
 from collections.abc import Collection, Iterable
@@ -18,6 +19,7 @@ from typing import Any
 
 _LOGGER: Incomplete
 _SlotsType = dict[str, Any]
+_IntentSlotsType: Incomplete
 INTENT_TURN_OFF: str
 INTENT_TURN_ON: str
 INTENT_TOGGLE: str
@@ -116,8 +118,8 @@ class MatchTargetsCandidate:
     matched_name: str | None = ...
     def __init__(self, state, entity, area, floor, device, matched_name) -> None: ...
 
-def _find_areas(name: str, areas: area_registry.AreaRegistry) -> Iterable[area_registry.AreaEntry]: ...
-def _find_floors(name: str, floors: floor_registry.FloorRegistry) -> Iterable[floor_registry.FloorEntry]: ...
+def find_areas(name: str, areas: area_registry.AreaRegistry) -> Iterable[area_registry.AreaEntry]: ...
+def find_floors(name: str, floors: floor_registry.FloorRegistry) -> Iterable[floor_registry.FloorEntry]: ...
 def _normalize_name(name: str) -> str: ...
 def _filter_by_name(name: str, candidates: Iterable[MatchTargetsCandidate]) -> Iterable[MatchTargetsCandidate]: ...
 def _filter_by_features(features: int, candidates: Iterable[MatchTargetsCandidate]) -> Iterable[MatchTargetsCandidate]: ...
@@ -153,7 +155,7 @@ class DynamicServiceIntentHandler(IntentHandler, metaclass=abc.ABCMeta):
     platforms: Incomplete
     required_slots: Incomplete
     optional_slots: Incomplete
-    def __init__(self, intent_type: str, speech: str | None = None, required_slots: dict[str | tuple[str, str], vol.Schema] | None = None, optional_slots: dict[str | tuple[str, str], vol.Schema] | None = None, required_domains: set[str] | None = None, required_features: int | None = None, required_states: set[str] | None = None, description: str | None = None, platforms: set[str] | None = None) -> None: ...
+    def __init__(self, intent_type: str, speech: str | None = None, required_slots: _IntentSlotsType | None = None, optional_slots: _IntentSlotsType | None = None, required_domains: set[str] | None = None, required_features: int | None = None, required_states: set[str] | None = None, description: str | None = None, platforms: set[str] | None = None) -> None: ...
     @cached_property
     def slot_schema(self) -> dict: ...
     @abstractmethod
@@ -166,7 +168,7 @@ class DynamicServiceIntentHandler(IntentHandler, metaclass=abc.ABCMeta):
 class ServiceIntentHandler(DynamicServiceIntentHandler):
     domain: Incomplete
     service: Incomplete
-    def __init__(self, intent_type: str, domain: str, service: str, speech: str | None = None, required_slots: dict[str | tuple[str, str], vol.Schema] | None = None, optional_slots: dict[str | tuple[str, str], vol.Schema] | None = None, required_domains: set[str] | None = None, required_features: int | None = None, required_states: set[str] | None = None, description: str | None = None, platforms: set[str] | None = None) -> None: ...
+    def __init__(self, intent_type: str, domain: str, service: str, speech: str | None = None, required_slots: _IntentSlotsType | None = None, optional_slots: _IntentSlotsType | None = None, required_domains: set[str] | None = None, required_features: int | None = None, required_states: set[str] | None = None, description: str | None = None, platforms: set[str] | None = None) -> None: ...
     def get_domain_and_service(self, intent_obj: Intent, state: State) -> tuple[str, str]: ...
 
 class IntentCategory(Enum):

homeassistant-stubs/helpers/llm.pyi

@@ -1,22 +1,27 @@
 import abc
 import voluptuous as vol
-from . import intent as intent, service as service
+from . import intent as intent, selector as selector, service as service
 from .singleton import singleton as singleton
 from _typeshed import Incomplete
 from abc import ABC, abstractmethod
+from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.climate.intent import INTENT_GET_TEMPERATURE as INTENT_GET_TEMPERATURE
 from homeassistant.components.conversation.trace import ConversationTraceEventType as ConversationTraceEventType, async_conversation_trace_append as async_conversation_trace_append
 from homeassistant.components.cover.intent import INTENT_CLOSE_COVER as INTENT_CLOSE_COVER, INTENT_OPEN_COVER as INTENT_OPEN_COVER
 from homeassistant.components.homeassistant.exposed_entities import async_should_expose as async_should_expose
 from homeassistant.components.intent import async_device_supports_timers as async_device_supports_timers
+from homeassistant.components.script import ATTR_VARIABLES as ATTR_VARIABLES
 from homeassistant.components.weather.intent import INTENT_GET_WEATHER as INTENT_GET_WEATHER
-from homeassistant.core import Context as Context, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.const import ATTR_DOMAIN as ATTR_DOMAIN, ATTR_ENTITY_ID as ATTR_ENTITY_ID, ATTR_SERVICE as ATTR_SERVICE, EVENT_HOMEASSISTANT_CLOSE as EVENT_HOMEASSISTANT_CLOSE, EVENT_SERVICE_REMOVED as EVENT_SERVICE_REMOVED, SERVICE_TURN_ON as SERVICE_TURN_ON
+from homeassistant.core import Context as Context, Event as Event, HomeAssistant as HomeAssistant, callback as callback, split_entity_id as split_entity_id
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.util import yaml as yaml
+from homeassistant.util.hass_dict import HassKey as HassKey
 from homeassistant.util.json import JsonObjectType as JsonObjectType
 from typing import Any
 
+SCRIPT_PARAMETERS_CACHE: HassKey[dict[str, tuple[str | None, vol.Schema]]]
 LLM_API_ASSIST: str
 BASE_PROMPT: str
 DEFAULT_INSTRUCTIONS_PROMPT: str
@@ -57,8 +62,9 @@ class APIInstance:
     api_prompt: str
     llm_context: LLMContext
     tools: list[Tool]
+    custom_serializer: Callable[[Any], Any] | None = ...
     async def async_call_tool(self, tool_input: ToolInput) -> JsonObjectType: ...
-    def __init__(self, api, api_prompt, llm_context, tools) -> None: ...
+    def __init__(self, api, api_prompt, llm_context, tools, custom_serializer) -> None: ...
 
 @dataclass(slots=True, kw_only=True)
 class API(ABC, metaclass=abc.ABCMeta):
@@ -86,3 +92,11 @@ class AssistAPI(API):
     def _async_get_tools(self, llm_context: LLMContext, exposed_entities: dict | None) -> list[Tool]: ...
 
 def _get_exposed_entities(hass: HomeAssistant, assistant: str) -> dict[str, dict[str, Any]]: ...
+def _selector_serializer(schema: Any) -> Any: ...
+
+class ScriptTool(Tool):
+    name: Incomplete
+    parameters: Incomplete
+    description: Incomplete
+    def __init__(self, hass: HomeAssistant, script_entity_id: str) -> None: ...
+    async def async_call(self, hass: HomeAssistant, tool_input: ToolInput, llm_context: LLMContext) -> JsonObjectType: ...

homeassistant-stubs/helpers/schema_config_entry_flow.pyi

@@ -4,7 +4,7 @@ from . import selector as selector
 from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
 from abc import ABC, abstractmethod
-from collections.abc import Callable as Callable, Container, Coroutine, Mapping
+from collections.abc import Callable, Container, Coroutine, Mapping
 from dataclasses import dataclass
 from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow, OptionsFlowWithConfigEntry as OptionsFlowWithConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback, split_entity_id as split_entity_id
@@ -78,4 +78,4 @@ class SchemaOptionsFlowHandler(OptionsFlowWithConfigEntry):
     def async_create_entry(self, data: Mapping[str, Any], **kwargs: Any) -> ConfigFlowResult: ...
 
 def wrapped_entity_config_entry_title(hass: HomeAssistant, entity_id_or_uuid: str) -> str: ...
-def entity_selector_without_own_entities(handler: SchemaOptionsFlowHandler, entity_selector_config: selector.EntitySelectorConfig) -> vol.Schema: ...
+def entity_selector_without_own_entities(handler: SchemaOptionsFlowHandler, entity_selector_config: selector.EntitySelectorConfig) -> selector.EntitySelector: ...

homeassistant-stubs/helpers/script.pyi

@@ -11,7 +11,7 @@ from .trace import TraceElement as TraceElement, async_trace_path as async_trace
 from .trigger import async_initialize_triggers as async_initialize_triggers, async_validate_trigger_config as async_validate_trigger_config
 from .typing import ConfigType as ConfigType, UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
-from collections.abc import AsyncGenerator, Callable as Callable, Mapping, Sequence
+from collections.abc import Callable as Callable, Mapping, Sequence
 from contextvars import ContextVar
 from dataclasses import dataclass
 from datetime import datetime, timedelta
@@ -27,6 +27,7 @@ from homeassistant.util.dt import utcnow as utcnow
 from homeassistant.util.hass_dict import HassKey as HassKey
 from homeassistant.util.signal_type import SignalType as SignalType, SignalTypeFormat as SignalTypeFormat
 from typing import Any, Literal, TypedDict
+from typing_extensions import AsyncGenerator
 
 SCRIPT_MODE_PARALLEL: str
 SCRIPT_MODE_QUEUED: str
@@ -64,7 +65,7 @@ class ScriptStoppedError(Exception): ...
 
 def _set_result_unless_done(future: asyncio.Future[None]) -> None: ...
 def action_trace_append(variables, path): ...
-async def trace_action(hass: HomeAssistant, script_run: _ScriptRun, stop: asyncio.Future[None], variables: dict[str, Any]) -> AsyncGenerator[TraceElement, None]: ...
+async def trace_action(hass: HomeAssistant, script_run: _ScriptRun, stop: asyncio.Future[None], variables: dict[str, Any]) -> AsyncGenerator[TraceElement]: ...
 def make_script_schema(schema: Mapping[Any, Any], default_script_mode: str, extra: int = ...) -> vol.Schema: ...
 
 STATIC_VALIDATION_ACTION_TYPES: Incomplete