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

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/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

homeassistant-stubs/helpers/selector.pyi

@@ -19,6 +19,7 @@ class Selector:
     config: _T
     selector_type: str
     def __init__(self, config: Mapping[str, Any] | None = None) -> None: ...
+    def __eq__(self, other: object) -> bool: ...
     def serialize(self) -> dict[str, dict[str, _T]]: ...
 
 def _entity_feature_flag(domain: str, enum_name: str, feature_name: str) -> int: ...
@@ -74,7 +75,7 @@ class AssistPipelineSelectorConfig(TypedDict, total=False): ...
 class AssistPipelineSelector(Selector[AssistPipelineSelectorConfig]):
     selector_type: str
     CONFIG_SCHEMA: Incomplete
-    def __init__(self, config: AssistPipelineSelectorConfig) -> None: ...
+    def __init__(self, config: AssistPipelineSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> str: ...
 
 class AttributeSelectorConfig(TypedDict, total=False):
@@ -153,7 +154,7 @@ class ConstantSelectorConfig(TypedDict, total=False):
 class ConstantSelector(Selector[ConstantSelectorConfig]):
     selector_type: str
     CONFIG_SCHEMA: Incomplete
-    def __init__(self, config: ConstantSelectorConfig | None = None) -> None: ...
+    def __init__(self, config: ConstantSelectorConfig) -> None: ...
     def __call__(self, data: Any) -> Any: ...
 
 class QrErrorCorrectionLevel(StrEnum):
@@ -170,7 +171,7 @@ class QrCodeSelectorConfig(TypedDict, total=False):
 class QrCodeSelector(Selector[QrCodeSelectorConfig]):
     selector_type: str
     CONFIG_SCHEMA: Incomplete
-    def __init__(self, config: QrCodeSelectorConfig | None = None) -> None: ...
+    def __init__(self, config: QrCodeSelectorConfig) -> None: ...
     def __call__(self, data: Any) -> Any: ...
 
 class ConversationAgentSelectorConfig(TypedDict, total=False):
@@ -179,7 +180,7 @@ class ConversationAgentSelectorConfig(TypedDict, total=False):
 class ConversationAgentSelector(Selector[ConversationAgentSelectorConfig]):
     selector_type: str
     CONFIG_SCHEMA: Incomplete
-    def __init__(self, config: ConversationAgentSelectorConfig) -> None: ...
+    def __init__(self, config: ConversationAgentSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> str: ...
 
 class CountrySelectorConfig(TypedDict, total=False):
@@ -221,6 +222,7 @@ class DeviceSelector(Selector[DeviceSelectorConfig]):
 
 class DurationSelectorConfig(TypedDict, total=False):
     enable_day: bool
+    allow_negative: bool
 
 class DurationSelector(Selector[DurationSelectorConfig]):
     selector_type: str
@@ -245,7 +247,7 @@ class FloorSelectorConfig(TypedDict, total=False):
     device: DeviceFilterSelectorConfig | list[DeviceFilterSelectorConfig]
     multiple: bool
 
-class FloorSelector(Selector[AreaSelectorConfig]):
+class FloorSelector(Selector[FloorSelectorConfig]):
     selector_type: str
     CONFIG_SCHEMA: Incomplete
     def __init__(self, config: FloorSelectorConfig | None = None) -> None: ...
@@ -277,7 +279,7 @@ class LanguageSelectorConfig(TypedDict, total=False):
 class LanguageSelector(Selector[LanguageSelectorConfig]):
     selector_type: str
     CONFIG_SCHEMA: Incomplete
-    def __init__(self, config: LanguageSelectorConfig) -> None: ...
+    def __init__(self, config: LanguageSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> str: ...
 
 class LocationSelectorConfig(TypedDict, total=False):
@@ -348,7 +350,7 @@ class SelectSelectorConfig(TypedDict, total=False):
 class SelectSelector(Selector[SelectSelectorConfig]):
     selector_type: str
     CONFIG_SCHEMA: Incomplete
-    def __init__(self, config: SelectSelectorConfig | None = None) -> None: ...
+    def __init__(self, config: SelectSelectorConfig) -> None: ...
     def __call__(self, data: Any) -> Any: ...
 
 class TargetSelectorConfig(TypedDict, total=False):
@@ -438,5 +440,5 @@ class FileSelectorConfig(TypedDict):
 class FileSelector(Selector[FileSelectorConfig]):
     selector_type: str
     CONFIG_SCHEMA: Incomplete
-    def __init__(self, config: FileSelectorConfig | None = None) -> None: ...
+    def __init__(self, config: FileSelectorConfig) -> None: ...
     def __call__(self, data: Any) -> str: ...

homeassistant-stubs/helpers/service.pyi

@@ -1,10 +1,9 @@
 import dataclasses
-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, label_registry as label_registry, template as template, translation as translation
 from .entity import Entity as Entity
 from .group import expand_entity_ids as expand_entity_ids
 from .selector import TargetSelector as TargetSelector
-from .typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsType
+from .typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsType, VolSchemaType as VolSchemaType
 from _typeshed import Incomplete
 from collections.abc import Awaitable, Callable as Callable, Coroutine, Iterable
 from homeassistant.auth.permissions.const import CAT_ENTITIES as CAT_ENTITIES, POLICY_CONTROL as POLICY_CONTROL
@@ -30,7 +29,11 @@ def validate_attribute_option(attribute_option: str) -> Any: ...
 def validate_supported_feature(supported_feature: str) -> Any: ...
 
 _FIELD_SCHEMA: Incomplete
+_SECTION_SCHEMA: Incomplete
 _SERVICE_SCHEMA: Incomplete
+
+def starts_with_dot(key: str) -> str: ...
+
 _SERVICES_SCHEMA: Incomplete
 
 class ServiceParams(TypedDict):
@@ -82,7 +85,7 @@ def _get_permissible_entity_candidates(call: ServiceCall, entities: dict[str, En
 async def entity_service_call(hass: HomeAssistant, registered_entities: dict[str, Entity], func: str | HassJob, call: ServiceCall, required_features: Iterable[int] | None = None) -> EntityServiceResponse | None: ...
 async def _handle_entity_call(hass: HomeAssistant, entity: Entity, func: str | HassJob, data: dict | ServiceCall, context: Context) -> ServiceResponse: ...
 async def _async_admin_handler(hass: HomeAssistant, service_job: HassJob[[ServiceCall], Awaitable[None] | None], call: ServiceCall) -> None: ...
-def async_register_admin_service(hass: HomeAssistant, domain: str, service: str, service_func: Callable[[ServiceCall], Awaitable[None] | None], schema: vol.Schema = ...) -> None: ...
+def async_register_admin_service(hass: HomeAssistant, domain: str, service: str, service_func: Callable[[ServiceCall], Awaitable[None] | None], schema: VolSchemaType = ...) -> None: ...
 def verify_domain_control(hass: HomeAssistant, domain: str) -> Callable[[Callable[[ServiceCall], Any]], Callable[[ServiceCall], Any]]: ...
 
 class ReloadServiceHelper:

homeassistant-stubs/helpers/template.pyi

@@ -4,7 +4,7 @@ from .singleton import singleton as singleton
 from .translation import async_translate_state as async_translate_state
 from .typing import TemplateVarsType as TemplateVarsType
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Generator, Iterable
+from collections.abc import Callable as Callable, Iterable
 from contextlib import AbstractContextManager
 from contextvars import ContextVar
 from datetime import datetime, timedelta
@@ -22,6 +22,7 @@ from jinja2.sandbox import ImmutableSandboxedEnvironment
 from lru import LRU
 from types import CodeType, TracebackType
 from typing import Any, Literal, NoReturn, Self, overload
+from typing_extensions import Generator
 
 _LOGGER: Incomplete
 _SENTINEL: Incomplete
@@ -139,7 +140,7 @@ class AllStates:
     __getitem__ = __getattr__
     def _collect_all(self) -> None: ...
     def _collect_all_lifecycle(self) -> None: ...
-    def __iter__(self) -> Generator[TemplateState, None, None]: ...
+    def __iter__(self) -> Generator[TemplateState]: ...
     def __len__(self) -> int: ...
     def __call__(self, entity_id: str, rounded: bool | object = ..., with_unit: bool = False) -> str: ...
     def __repr__(self) -> str: ...
@@ -161,7 +162,7 @@ class DomainStates:
     __getitem__ = __getattr__
     def _collect_domain(self) -> None: ...
     def _collect_domain_lifecycle(self) -> None: ...
-    def __iter__(self) -> Generator[TemplateState, None, None]: ...
+    def __iter__(self) -> Generator[TemplateState]: ...
     def __len__(self) -> int: ...
     def __repr__(self) -> str: ...
 
@@ -215,7 +216,7 @@ class TemplateStateFromEntityId(TemplateStateBase):
 _create_template_state_no_collect: Incomplete
 
 def _collect_state(hass: HomeAssistant, entity_id: str) -> None: ...
-def _state_generator(hass: HomeAssistant, domain: str | None) -> Generator[TemplateState, None, None]: ...
+def _state_generator(hass: HomeAssistant, domain: str | None) -> Generator[TemplateState]: ...
 def _get_state_if_valid(hass: HomeAssistant, entity_id: str) -> TemplateState | None: ...
 def _get_state(hass: HomeAssistant, entity_id: str) -> TemplateState | None: ...
 def _get_template_state_from_state(hass: HomeAssistant, entity_id: str, state: State | None) -> TemplateState | None: ...
@@ -231,6 +232,7 @@ def integration_entities(hass: HomeAssistant, entry_name: str) -> Iterable[str]:
 def config_entry_id(hass: HomeAssistant, entity_id: str) -> str | None: ...
 def device_id(hass: HomeAssistant, entity_id_or_device_name: str) -> str | None: ...
 def device_attr(hass: HomeAssistant, device_or_entity_id: str, attr_name: str) -> Any: ...
+def config_entry_attr(hass: HomeAssistant, config_entry_id_: str, attr_name: str) -> Any: ...
 def is_device_attr(hass: HomeAssistant, device_or_entity_id: str, attr_name: str, attr_value: Any) -> bool: ...
 def issues(hass: HomeAssistant) -> dict[tuple[str, str], dict[str, Any]]: ...
 def issue(hass: HomeAssistant, domain: str, issue_id: str) -> dict[str, Any] | None: ...
@@ -312,8 +314,8 @@ def bitwise_or(first_value, second_value): ...
 def bitwise_xor(first_value, second_value): ...
 def struct_pack(value: Any | None, format_string: str) -> bytes | None: ...
 def struct_unpack(value: bytes, format_string: str, offset: int = 0) -> Any | None: ...
-def base64_encode(value): ...
-def base64_decode(value): ...
+def base64_encode(value: str) -> str: ...
+def base64_decode(value: str, encoding: str | None = 'utf-8') -> str | bytes: ...
 def ordinal(value): ...
 def from_json(value): ...
 def _to_json_default(obj: Any) -> None: ...

homeassistant-stubs/helpers/trace.pyi

@@ -1,10 +1,11 @@
 from .typing import TemplateVarsType as TemplateVarsType
 from _typeshed import Incomplete
 from collections import deque
-from collections.abc import Callable as Callable, Coroutine, Generator
+from collections.abc import Callable as Callable, Coroutine
 from contextvars import ContextVar
 from homeassistant.core import ServiceResponse as ServiceResponse
 from typing import Any
+from typing_extensions import Generator
 
 class TraceElement:
     __slots__: Incomplete
@@ -54,5 +55,5 @@ class StopReason:
 
 def script_execution_set(reason: str, response: ServiceResponse = None) -> None: ...
 def script_execution_get() -> str | None: ...
-def trace_path(suffix: str | list[str]) -> Generator[None, None, None]: ...
+def trace_path(suffix: str | list[str]) -> Generator[None]: ...
 def async_trace_path(suffix: str | list[str]) -> Callable[[Callable[[Unpack[_Ts]], Coroutine[Any, Any, None]]], Callable[[Unpack[_Ts]], Coroutine[Any, Any, None]]]: ...

homeassistant-stubs/helpers/typing.pyi

@@ -1,4 +1,4 @@
-from .deprecation import DeprecatedAlias as DeprecatedAlias, 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 .deprecation import DeferredDeprecatedAlias as DeferredDeprecatedAlias, 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 _typeshed import Incomplete
 from collections.abc import Mapping
 from enum import Enum
@@ -11,12 +11,17 @@ ServiceDataType = dict[str, Any]
 StateType: Incomplete
 TemplateVarsType: Incomplete
 NoEventData = Mapping[str, Never]
+VolSchemaType: Incomplete
+VolDictType: Incomplete
 QueryType = Any
 
 class UndefinedType(Enum):
     _singleton: int
 
 UNDEFINED: Incomplete
+
+def _deprecated_typing_helper(attr: str) -> DeferredDeprecatedAlias: ...
+
 _DEPRECATED_ContextType: Incomplete
 _DEPRECATED_EventType: Incomplete
 _DEPRECATED_HomeAssistantType: Incomplete

homeassistant-stubs/helpers/update_coordinator.pyi

@@ -4,7 +4,7 @@ from . import entity as entity, event as event
 from .debounce import Debouncer as Debouncer
 from _typeshed import Incomplete
 from abc import abstractmethod
-from collections.abc import Awaitable, Callable as Callable, Coroutine, Generator
+from collections.abc import Awaitable, Callable as Callable, Coroutine
 from datetime import datetime, timedelta
 from homeassistant import config_entries as config_entries
 from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
@@ -12,7 +12,7 @@ from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, H
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryError as ConfigEntryError, ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.util.dt import utcnow as utcnow
 from typing import Any, Generic, Protocol
-from typing_extensions import TypeVar
+from typing_extensions import Generator, TypeVar
 
 REQUEST_REFRESH_DEFAULT_COOLDOWN: int
 REQUEST_REFRESH_DEFAULT_IMMEDIATE: bool
@@ -48,7 +48,7 @@ class DataUpdateCoordinator(BaseDataUpdateCoordinatorProtocol, Generic[_DataT]):
     def async_update_listeners(self) -> None: ...
     async def async_shutdown(self) -> None: ...
     def _unschedule_refresh(self) -> None: ...
-    def async_contexts(self) -> Generator[Any, None, None]: ...
+    def async_contexts(self) -> Generator[Any]: ...
     def _async_unsub_refresh(self) -> None: ...
     def _async_unsub_shutdown(self) -> None: ...
     @property

homeassistant-stubs/loader.pyi

@@ -15,7 +15,7 @@ from .generated.usb import USB as USB
 from .generated.zeroconf import HOMEKIT as HOMEKIT, ZEROCONF as ZEROCONF
 from .helpers import device_registry as dr
 from .helpers.json import json_bytes as json_bytes, json_fragment as json_fragment
-from .helpers.typing import ConfigType as ConfigType
+from .helpers.typing import ConfigType as ConfigType, UNDEFINED as UNDEFINED
 from .util.hass_dict import HassKey as HassKey
 from .util.json import JSON_DECODE_EXCEPTIONS as JSON_DECODE_EXCEPTIONS, json_loads as json_loads
 from _typeshed import Incomplete
@@ -45,7 +45,6 @@ PACKAGE_CUSTOM_COMPONENTS: str
 PACKAGE_BUILTIN: str
 CUSTOM_WARNING: str
 IMPORT_EVENT_LOOP_WARNING: str
-_UNDEF: Incomplete
 MOVED_ZEROCONF_PROPS: Incomplete
 
 class DHCPMatcherRequired(TypedDict, total=True):