homeassistant-stubs 2024.5.5__py3-none-any.whl → 2024.6.0__py3-none-any.whl

homeassistant-stubs/helpers/intent.pyi

@@ -13,6 +13,7 @@ from homeassistant.const import ATTR_DEVICE_CLASS as ATTR_DEVICE_CLASS, ATTR_ENT
 from homeassistant.core import Context as Context, HomeAssistant as HomeAssistant, State as State, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.loader import bind_hass as bind_hass
+from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Any
 
 _LOGGER: Incomplete
@@ -23,46 +24,115 @@ INTENT_TOGGLE: str
 INTENT_GET_STATE: str
 INTENT_NEVERMIND: str
 INTENT_SET_POSITION: str
+INTENT_START_TIMER: str
+INTENT_CANCEL_TIMER: str
+INTENT_INCREASE_TIMER: str
+INTENT_DECREASE_TIMER: str
+INTENT_PAUSE_TIMER: str
+INTENT_UNPAUSE_TIMER: str
+INTENT_TIMER_STATUS: str
 SLOT_SCHEMA: Incomplete
-DATA_KEY: str
+DATA_KEY: HassKey[dict[str, IntentHandler]]
 SPEECH_TYPE_PLAIN: str
 SPEECH_TYPE_SSML: str
 
 def async_register(hass: HomeAssistant, handler: IntentHandler) -> None: ...
 def async_remove(hass: HomeAssistant, intent_type: str) -> None: ...
-async def async_handle(hass: HomeAssistant, platform: str, intent_type: str, slots: _SlotsType | None = None, text_input: str | None = None, context: Context | None = None, language: str | None = None, assistant: str | None = None) -> IntentResponse: ...
+def async_get(hass: HomeAssistant) -> Iterable[IntentHandler]: ...
+async def async_handle(hass: HomeAssistant, platform: str, intent_type: str, slots: _SlotsType | None = None, text_input: str | None = None, context: Context | None = None, language: str | None = None, assistant: str | None = None, device_id: str | None = None, conversation_agent_id: str | None = None) -> IntentResponse: ...
 
 class IntentError(HomeAssistantError): ...
 class UnknownIntent(IntentError): ...
 class InvalidSlotInfo(IntentError): ...
-class IntentHandleError(IntentError): ...
+
+class IntentHandleError(IntentError):
+    response_key: Incomplete
+    def __init__(self, message: str = '', response_key: str | None = None) -> None: ...
+
 class IntentUnexpectedError(IntentError): ...
 
-class NoStatesMatchedError(IntentError):
-    name: Incomplete
-    area: Incomplete
-    floor: Incomplete
-    domains: Incomplete
-    device_classes: Incomplete
-    def __init__(self, name: str | None = None, area: str | None = None, floor: str | None = None, domains: set[str] | None = None, device_classes: set[str] | None = None) -> None: ...
-
-class DuplicateNamesMatchedError(IntentError):
-    name: Incomplete
-    area: Incomplete
-    def __init__(self, name: str, area: str | None) -> None: ...
-
-def _is_device_class(state: State, entity: entity_registry.RegistryEntry | None, device_classes: Collection[str]) -> bool: ...
-def _has_name(state: State, entity: entity_registry.RegistryEntry | None, name: str) -> bool: ...
-def _find_area(id_or_name: str, areas: area_registry.AreaRegistry) -> area_registry.AreaEntry | None: ...
-def _find_floor(id_or_name: str, floors: floor_registry.FloorRegistry) -> floor_registry.FloorEntry | None: ...
-def _filter_by_areas(states_and_entities: list[tuple[State, entity_registry.RegistryEntry | None]], areas: Iterable[area_registry.AreaEntry], devices: device_registry.DeviceRegistry) -> Iterable[tuple[State, entity_registry.RegistryEntry | None]]: ...
-def async_match_states(hass: HomeAssistant, name: str | None = None, area_name: str | None = None, area: area_registry.AreaEntry | None = None, floor_name: str | None = None, floor: floor_registry.FloorEntry | None = None, domains: Collection[str] | None = None, device_classes: Collection[str] | None = None, states: Iterable[State] | None = None, entities: entity_registry.EntityRegistry | None = None, areas: area_registry.AreaRegistry | None = None, floors: floor_registry.FloorRegistry | None = None, devices: device_registry.DeviceRegistry | None = None, assistant: str | None = None) -> Iterable[State]: ...
+class MatchFailedReason(Enum):
+    NAME: Incomplete
+    AREA: Incomplete
+    FLOOR: Incomplete
+    DOMAIN: Incomplete
+    DEVICE_CLASS: Incomplete
+    FEATURE: Incomplete
+    STATE: Incomplete
+    ASSISTANT: Incomplete
+    INVALID_AREA: Incomplete
+    INVALID_FLOOR: Incomplete
+    DUPLICATE_NAME: Incomplete
+    def is_no_entities_reason(self) -> bool: ...
+
+@dataclass
+class MatchTargetsConstraints:
+    name: str | None = ...
+    area_name: str | None = ...
+    floor_name: str | None = ...
+    domains: Collection[str] | None = ...
+    device_classes: Collection[str] | None = ...
+    features: int | None = ...
+    states: Collection[str] | None = ...
+    assistant: str | None = ...
+    allow_duplicate_names: bool = ...
+    @property
+    def has_constraints(self) -> bool: ...
+    def __init__(self, name, area_name, floor_name, domains, device_classes, features, states, assistant, allow_duplicate_names) -> None: ...
+
+@dataclass
+class MatchTargetsPreferences:
+    area_id: str | None = ...
+    floor_id: str | None = ...
+    def __init__(self, area_id, floor_id) -> None: ...
+
+@dataclass
+class MatchTargetsResult:
+    is_match: bool
+    no_match_reason: MatchFailedReason | None = ...
+    states: list[State] = ...
+    no_match_name: str | None = ...
+    areas: list[area_registry.AreaEntry] = ...
+    floors: list[floor_registry.FloorEntry] = ...
+    def __init__(self, is_match, no_match_reason, states, no_match_name, areas, floors) -> None: ...
+
+class MatchFailedError(IntentError):
+    result: Incomplete
+    constraints: Incomplete
+    preferences: Incomplete
+    def __init__(self, result: MatchTargetsResult, constraints: MatchTargetsConstraints, preferences: MatchTargetsPreferences | None = None) -> None: ...
+    def __str__(self) -> str: ...
+
+class NoStatesMatchedError(MatchFailedError):
+    def __init__(self, reason: MatchFailedReason, name: str | None = None, area: str | None = None, floor: str | None = None, domains: set[str] | None = None, device_classes: set[str] | None = None) -> None: ...
+
+@dataclass
+class MatchTargetsCandidate:
+    state: State
+    entity: entity_registry.RegistryEntry | None = ...
+    area: area_registry.AreaEntry | None = ...
+    floor: floor_registry.FloorEntry | None = ...
+    device: device_registry.DeviceEntry | None = ...
+    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 _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]: ...
+def _filter_by_device_classes(device_classes: Iterable[str], candidates: Iterable[MatchTargetsCandidate]) -> Iterable[MatchTargetsCandidate]: ...
+def _add_areas(areas: area_registry.AreaRegistry, devices: device_registry.DeviceRegistry, candidates: Iterable[MatchTargetsCandidate]) -> None: ...
+def async_match_targets(hass: HomeAssistant, constraints: MatchTargetsConstraints, preferences: MatchTargetsPreferences | None = None, states: list[State] | None = None) -> MatchTargetsResult: ...
+def async_match_states(hass: HomeAssistant, name: str | None = None, area_name: str | None = None, floor_name: str | None = None, domains: Collection[str] | None = None, device_classes: Collection[str] | None = None, states: list[State] | None = None) -> Iterable[State]: ...
 def async_test_feature(state: State, feature: int, feature_name: str) -> None: ...
 
 class IntentHandler:
-    intent_type: str | None
-    slot_schema: vol.Schema | None
-    platforms: Iterable[str] | None
+    intent_type: str
+    platforms: set[str] | None
+    description: str | None
+    @property
+    def slot_schema(self) -> dict | None: ...
     def async_can_handle(self, intent_obj: Intent) -> bool: ...
     def async_validate_slots(self, slots: _SlotsType) -> _SlotsType: ...
     @cached_property
@@ -70,26 +140,33 @@ class IntentHandler:
     async def async_handle(self, intent_obj: Intent) -> IntentResponse: ...
     def __repr__(self) -> str: ...
 
+def non_empty_string(value: Any) -> str: ...
+
 class DynamicServiceIntentHandler(IntentHandler, metaclass=abc.ABCMeta):
-    slot_schema: Incomplete
     service_timeout: float
     intent_type: Incomplete
     speech: Incomplete
-    extra_slots: Incomplete
-    def __init__(self, intent_type: str, speech: str | None = None, extra_slots: dict[str, vol.Schema] | None = None) -> None: ...
+    required_domains: Incomplete
+    required_features: Incomplete
+    required_states: Incomplete
+    description: Incomplete
+    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: ...
     @cached_property
-    def _slot_schema(self) -> vol.Schema: ...
+    def slot_schema(self) -> dict: ...
     @abstractmethod
     def get_domain_and_service(self, intent_obj: Intent, state: State) -> tuple[str, str]: ...
     async def async_handle(self, intent_obj: Intent) -> IntentResponse: ...
-    async def async_handle_states(self, intent_obj: Intent, states: list[State], area: area_registry.AreaEntry | None = None) -> IntentResponse: ...
+    async def async_handle_states(self, intent_obj: Intent, match_result: MatchTargetsResult, match_constraints: MatchTargetsConstraints, match_preferences: MatchTargetsPreferences | None = None) -> IntentResponse: ...
     async def async_call_service(self, domain: str, service: str, intent_obj: Intent, state: State) -> None: ...
     async def _run_then_background(self, task: asyncio.Task[Any]) -> None: ...
 
 class ServiceIntentHandler(DynamicServiceIntentHandler):
     domain: Incomplete
     service: Incomplete
-    def __init__(self, intent_type: str, domain: str, service: str, speech: str | None = None, extra_slots: dict[str, vol.Schema] | None = None) -> None: ...
+    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 get_domain_and_service(self, intent_obj: Intent, state: State) -> tuple[str, str]: ...
 
 class IntentCategory(Enum):
@@ -107,7 +184,9 @@ class Intent:
     language: Incomplete
     category: Incomplete
     assistant: Incomplete
-    def __init__(self, hass: HomeAssistant, platform: str, intent_type: str, slots: _SlotsType, text_input: str | None, context: Context, language: str, category: IntentCategory | None = None, assistant: str | None = None) -> None: ...
+    device_id: Incomplete
+    conversation_agent_id: Incomplete
+    def __init__(self, hass: HomeAssistant, platform: str, intent_type: str, slots: _SlotsType, text_input: str | None, context: Context, language: str, category: IntentCategory | None = None, assistant: str | None = None, device_id: str | None = None, conversation_agent_id: str | None = None) -> None: ...
     def create_response(self) -> IntentResponse: ...
 
 class IntentResponseType(Enum):
@@ -124,6 +203,7 @@ class IntentResponseErrorCode(str, Enum):
 
 class IntentResponseTargetType(str, Enum):
     AREA: str
+    FLOOR: str
     DEVICE: str
     ENTITY: str
     DOMAIN: str
@@ -149,6 +229,7 @@ class IntentResponse:
     failed_results: Incomplete
     matched_states: Incomplete
     unmatched_states: Incomplete
+    speech_slots: Incomplete
     response_type: Incomplete
     def __init__(self, language: str, intent: Intent | None = None) -> None: ...
     def async_set_speech(self, speech: str, speech_type: str = 'plain', extra_data: Any | None = None) -> None: ...
@@ -158,4 +239,5 @@ class IntentResponse:
     def async_set_targets(self, intent_targets: list[IntentResponseTarget]) -> None: ...
     def async_set_results(self, success_results: list[IntentResponseTarget], failed_results: list[IntentResponseTarget] | None = None) -> None: ...
     def async_set_states(self, matched_states: list[State], unmatched_states: list[State] | None = None) -> None: ...
+    def async_set_speech_slots(self, speech_slots: dict[str, Any]) -> None: ...
     def as_dict(self) -> dict[str, Any]: ...

homeassistant-stubs/helpers/issue_registry.pyi

@@ -1,19 +1,27 @@
 import dataclasses
 from .registry import BaseRegistry as BaseRegistry
+from .singleton import singleton as singleton
 from .storage import Store as Store
 from _typeshed import Incomplete
 from datetime import datetime
 from enum import StrEnum
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.util.async_ import run_callback_threadsafe as run_callback_threadsafe
-from typing import Any
+from homeassistant.util.event_type import EventType as EventType
+from homeassistant.util.hass_dict import HassKey as HassKey
+from typing import Any, Literal, TypedDict
 
-DATA_REGISTRY: str
-EVENT_REPAIRS_ISSUE_REGISTRY_UPDATED: str
+DATA_REGISTRY: HassKey[IssueRegistry]
+EVENT_REPAIRS_ISSUE_REGISTRY_UPDATED: EventType[EventIssueRegistryUpdatedData]
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
 STORAGE_VERSION_MINOR: int
 
+class EventIssueRegistryUpdatedData(TypedDict):
+    action: Literal['create', 'remove', 'update']
+    domain: str
+    issue_id: str
+
 class IssueSeverity(StrEnum):
     CRITICAL: str
     ERROR: str
@@ -44,13 +52,13 @@ class IssueRegistryStore(Store[dict[str, list[dict[str, Any]]]]):
 class IssueRegistry(BaseRegistry):
     hass: Incomplete
     issues: Incomplete
-    _read_only: Incomplete
     _store: Incomplete
-    def __init__(self, hass: HomeAssistant, *, read_only: bool = False) -> None: ...
+    def __init__(self, hass: HomeAssistant) -> None: ...
     def async_get_issue(self, domain: str, issue_id: str) -> IssueEntry | None: ...
     def async_get_or_create(self, domain: str, issue_id: str, *, breaks_in_ha_version: str | None = None, data: dict[str, str | int | float | None] | None = None, is_fixable: bool, is_persistent: bool, issue_domain: str | None = None, learn_more_url: str | None = None, severity: IssueSeverity, translation_key: str, translation_placeholders: dict[str, str] | None = None) -> IssueEntry: ...
     def async_delete(self, domain: str, issue_id: str) -> None: ...
     def async_ignore(self, domain: str, issue_id: str, ignore: bool) -> IssueEntry: ...
+    def make_read_only(self) -> None: ...
     async def async_load(self) -> None: ...
     def _data_to_save(self) -> dict[str, list[dict[str, str | None]]]: ...
 

homeassistant-stubs/helpers/label_registry.pyi

@@ -1,5 +1,6 @@
 from .normalized_name_base_registry import NormalizedNameBaseRegistryEntry as NormalizedNameBaseRegistryEntry, NormalizedNameBaseRegistryItems as NormalizedNameBaseRegistryItems, normalize_name as normalize_name
 from .registry import BaseRegistry as BaseRegistry
+from .singleton import singleton as singleton
 from .storage import Store as Store
 from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
@@ -8,9 +9,10 @@ from dataclasses import dataclass
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.util import slugify as slugify
 from homeassistant.util.event_type import EventType as EventType
+from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Literal, TypedDict
 
-DATA_REGISTRY: str
+DATA_REGISTRY: HassKey[LabelRegistry]
 EVENT_LABEL_REGISTRY_UPDATED: EventType[EventLabelRegistryUpdatedData]
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int

homeassistant-stubs/helpers/llm.pyi

@@ -0,0 +1,87 @@
+import abc
+import voluptuous as vol
+from . import intent as intent, service as service
+from .singleton import singleton as singleton
+from _typeshed import Incomplete
+from abc import ABC, abstractmethod
+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.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.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.util import yaml as yaml
+from homeassistant.util.json import JsonObjectType as JsonObjectType
+from typing import Any
+
+LLM_API_ASSIST: str
+BASE_PROMPT: str
+DEFAULT_INSTRUCTIONS_PROMPT: str
+
+def async_render_no_api_prompt(hass: HomeAssistant) -> str: ...
+def _async_get_apis(hass: HomeAssistant) -> dict[str, API]: ...
+def async_register_api(hass: HomeAssistant, api: API) -> None: ...
+async def async_get_api(hass: HomeAssistant, api_id: str, llm_context: LLMContext) -> APIInstance: ...
+def async_get_apis(hass: HomeAssistant) -> list[API]: ...
+
+@dataclass(slots=True)
+class LLMContext:
+    platform: str
+    context: Context | None
+    user_prompt: str | None
+    language: str | None
+    assistant: str | None
+    device_id: str | None
+    def __init__(self, platform, context, user_prompt, language, assistant, device_id) -> None: ...
+
+@dataclass(slots=True)
+class ToolInput:
+    tool_name: str
+    tool_args: dict[str, Any]
+    def __init__(self, tool_name, tool_args) -> None: ...
+
+class Tool(metaclass=abc.ABCMeta):
+    name: str
+    description: str | None
+    parameters: vol.Schema
+    @abstractmethod
+    async def async_call(self, hass: HomeAssistant, tool_input: ToolInput, llm_context: LLMContext) -> JsonObjectType: ...
+    def __repr__(self) -> str: ...
+
+@dataclass
+class APIInstance:
+    api: API
+    api_prompt: str
+    llm_context: LLMContext
+    tools: list[Tool]
+    async def async_call_tool(self, tool_input: ToolInput) -> JsonObjectType: ...
+    def __init__(self, api, api_prompt, llm_context, tools) -> None: ...
+
+@dataclass(slots=True, kw_only=True)
+class API(ABC, metaclass=abc.ABCMeta):
+    hass: HomeAssistant
+    id: str
+    name: str
+    @abstractmethod
+    async def async_get_api_instance(self, llm_context: LLMContext) -> APIInstance: ...
+    def __init__(self, *, hass, id, name) -> None: ...
+
+class IntentTool(Tool):
+    name: Incomplete
+    description: Incomplete
+    extra_slots: Incomplete
+    parameters: Incomplete
+    def __init__(self, intent_handler: intent.IntentHandler) -> None: ...
+    async def async_call(self, hass: HomeAssistant, tool_input: ToolInput, llm_context: LLMContext) -> JsonObjectType: ...
+
+class AssistAPI(API):
+    IGNORE_INTENTS: Incomplete
+    def __init__(self, hass: HomeAssistant) -> None: ...
+    async def async_get_api_instance(self, llm_context: LLMContext) -> APIInstance: ...
+    def _async_get_api_prompt(self, llm_context: LLMContext, exposed_entities: dict | None) -> str: ...
+    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]]: ...

homeassistant-stubs/helpers/normalized_name_base_registry.pyi

@@ -1,14 +1,12 @@
 from .registry import BaseRegistryItems as BaseRegistryItems
 from _typeshed import Incomplete
 from dataclasses import dataclass
-from typing import TypeVar
 
 @dataclass(slots=True, frozen=True, kw_only=True)
 class NormalizedNameBaseRegistryEntry:
     name: str
     normalized_name: str
     def __init__(self, *, name, normalized_name) -> None: ...
-_VT = TypeVar('_VT', bound=NormalizedNameBaseRegistryEntry)
 
 def normalize_name(name: str) -> str: ...
 

homeassistant-stubs/helpers/ratelimit.pyi

@@ -1,9 +1,7 @@
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Hashable
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from typing import TypeVarTuple
 
-_Ts = TypeVarTuple('_Ts')
 _LOGGER: Incomplete
 
 class KeyedRateLimit:

homeassistant-stubs/helpers/recorder.pyi

@@ -1,14 +1,15 @@
 import asyncio
 from dataclasses import dataclass
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Any
 
-DOMAIN: str
+DOMAIN: HassKey[RecorderData]
 
 @dataclass(slots=True)
 class RecorderData:
     recorder_platforms: dict[str, Any] = ...
-    db_connected: asyncio.Future = ...
+    db_connected: asyncio.Future[bool] = ...
     def __init__(self, recorder_platforms, db_connected) -> None: ...
 
 def async_migration_in_progress(hass: HomeAssistant) -> bool: ...

homeassistant-stubs/helpers/redact.pyi

@@ -1,10 +1,8 @@
 from collections.abc import Callable as Callable, Iterable, Mapping
 from homeassistant.core import callback as callback
-from typing import Any, TypeVar, overload
+from typing import Any, overload
 
 REDACTED: str
-_T = TypeVar('_T')
-_ValueT = TypeVar('_ValueT')
 
 def partial_redact(x: str | Any, unmasked_prefix: int = 4, unmasked_suffix: int = 4) -> str: ...
 @overload

homeassistant-stubs/helpers/registry.pyi

@@ -1,15 +1,14 @@
 import abc
 from .storage import Store as Store
+from _typeshed import Incomplete
 from abc import ABC, abstractmethod
 from collections import UserDict
-from collections.abc import Mapping, Sequence, ValuesView
+from collections.abc import Mapping as Mapping, Sequence as Sequence, ValuesView
 from homeassistant.core import CoreState as CoreState, HomeAssistant as HomeAssistant, callback as callback
-from typing import Any, Generic, Literal, TypeVar
 
 SAVE_DELAY: int
 SAVE_DELAY_LONG: int
-_DataT = TypeVar('_DataT')
-_StoreDataT = TypeVar('_StoreDataT', bound=Mapping[str, Any] | Sequence[Any])
+RegistryIndexType: Incomplete
 
 class BaseRegistryItems(UserDict[str, _DataT], ABC, metaclass=abc.ABCMeta):
     data: dict[str, _DataT]
@@ -19,10 +18,10 @@ class BaseRegistryItems(UserDict[str, _DataT], ABC, metaclass=abc.ABCMeta):
     @abstractmethod
     def _unindex_entry(self, key: str, replacement_entry: _DataT | None = None) -> None: ...
     def __setitem__(self, key: str, entry: _DataT) -> None: ...
-    def _unindex_entry_value(self, key: str, value: str, index: dict[str, dict[str, Literal[True]]]) -> None: ...
+    def _unindex_entry_value(self, key: str, value: str, index: RegistryIndexType) -> None: ...
     def __delitem__(self, key: str) -> None: ...
 
-class BaseRegistry(ABC, Generic[_StoreDataT], metaclass=abc.ABCMeta):
+class BaseRegistry(ABC, metaclass=abc.ABCMeta):
     hass: HomeAssistant
     _store: Store[_StoreDataT]
     def async_schedule_save(self) -> None: ...

homeassistant-stubs/helpers/restore_state.pyi

@@ -4,6 +4,7 @@ from .entity import Entity as Entity
 from .event import async_track_time_interval as async_track_time_interval
 from .frame import report as report
 from .json import JSONEncoder as JSONEncoder
+from .singleton import singleton as singleton
 from .storage import Store as Store
 from _typeshed import Incomplete
 from abc import ABC, abstractmethod
@@ -11,10 +12,11 @@ from datetime import datetime
 from homeassistant.const import ATTR_RESTORED as ATTR_RESTORED, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import HomeAssistant as HomeAssistant, State as State, callback as callback, valid_entity_id as valid_entity_id
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.util.hass_dict import HassKey as HassKey
 from homeassistant.util.json import json_loads as json_loads
 from typing import Any, Self
 
-DATA_RESTORE_STATE: str
+DATA_RESTORE_STATE: HassKey[RestoreStateData]
 _LOGGER: Incomplete
 STORAGE_KEY: str
 STORAGE_VERSION: int

homeassistant-stubs/helpers/script.pyi

@@ -3,9 +3,10 @@ import logging
 import voluptuous as vol
 from . import condition as condition, service as service, template as template
 from .condition import ConditionCheckerType as ConditionCheckerType, trace_condition_function as trace_condition_function
-from .dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
+from .dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send_internal as async_dispatcher_send_internal
 from .event import async_call_later as async_call_later, async_track_template as async_track_template
 from .script_variables import ScriptVariables as ScriptVariables
+from .template import Template as Template
 from .trace import TraceElement as TraceElement, async_trace_path as async_trace_path, script_execution_set as script_execution_set, trace_append_element as trace_append_element, trace_id_get as trace_id_get, trace_path as trace_path, trace_path_get as trace_path_get, trace_path_stack_cv as trace_path_stack_cv, trace_set_result as trace_set_result, 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, trace_update_result as trace_update_result
 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
@@ -23,10 +24,10 @@ from homeassistant.core import Context as Context, Event as Event, HassJob as Ha
 from homeassistant.util import slugify as slugify
 from homeassistant.util.async_ import create_eager_task as create_eager_task
 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, TypeVar, TypedDict
+from typing import Any, Literal, TypedDict
 
-_T = TypeVar('_T')
 SCRIPT_MODE_PARALLEL: str
 SCRIPT_MODE_QUEUED: str
 SCRIPT_MODE_RESTART: str
@@ -40,9 +41,9 @@ _MAX_EXCEEDED_CHOICES: Incomplete
 DEFAULT_MAX_EXCEEDED: str
 ATTR_CUR: str
 ATTR_MAX: str
-DATA_SCRIPTS: str
-DATA_SCRIPT_BREAKPOINTS: str
-DATA_NEW_SCRIPT_RUNS_NOT_ALLOWED: str
+DATA_SCRIPTS: HassKey[list[ScriptData]]
+DATA_SCRIPT_BREAKPOINTS: HassKey[dict[str, dict[str, set[str]]]]
+DATA_NEW_SCRIPT_RUNS_NOT_ALLOWED: HassKey[None]
 RUN_ID_ANY: str
 NODE_ANY: str
 _LOGGER: Incomplete
@@ -53,7 +54,11 @@ ACTION_TRACE_NODE_MAX_LEN: int
 SCRIPT_BREAKPOINT_HIT: Incomplete
 SCRIPT_DEBUG_CONTINUE_STOP: SignalTypeFormat[Literal['continue', 'stop']]
 SCRIPT_DEBUG_CONTINUE_ALL: str
-script_stack_cv: ContextVar[list[int] | None]
+script_stack_cv: ContextVar[list[str] | None]
+
+class ScriptData(TypedDict):
+    instance: Script
+    started_before_shutdown: bool
 
 class ScriptStoppedError(Exception): ...
 
@@ -122,6 +127,7 @@ class _ScriptRun:
     async def _async_variables_step(self) -> None: ...
     async def _async_set_conversation_response_step(self) -> None: ...
     async def _async_stop_step(self) -> None: ...
+    async def _async_sequence_step(self) -> None: ...
     async def _async_parallel_step(self) -> None: ...
     async def _async_run_script(self, script: Script) -> None: ...
 
@@ -159,6 +165,7 @@ class Script:
     _hass: Incomplete
     sequence: Incomplete
     name: Incomplete
+    unique_id: Incomplete
     domain: Incomplete
     running_description: Incomplete
     _change_listener: Incomplete
@@ -176,14 +183,15 @@ class Script:
     _choose_data: Incomplete
     _if_data: Incomplete
     _parallel_scripts: Incomplete
+    _sequence_scripts: Incomplete
     variables: Incomplete
     _variables_dynamic: Incomplete
     _copy_variables_on_run: Incomplete
-    def __init__(self, hass: HomeAssistant, sequence: Sequence[dict[str, Any]], name: str, domain: str, *, change_listener: Callable[..., Any] | None = None, copy_variables: bool = False, log_exceptions: bool = True, logger: logging.Logger | None = None, max_exceeded: str = ..., max_runs: int = ..., running_description: str | None = None, script_mode: str = ..., top_level: bool = True, variables: ScriptVariables | None = None) -> None: ...
+    def __init__(self, hass: HomeAssistant, sequence: Sequence[dict[str, Any]], name: str, domain: str, *, change_listener: Callable[[], Any] | None = None, copy_variables: bool = False, log_exceptions: bool = True, logger: logging.Logger | None = None, max_exceeded: str = ..., max_runs: int = ..., running_description: str | None = None, script_mode: str = ..., top_level: bool = True, variables: ScriptVariables | None = None) -> None: ...
     @property
     def change_listener(self) -> Callable[..., Any] | None: ...
     @change_listener.setter
-    def change_listener(self, change_listener: Callable[..., Any]) -> None: ...
+    def change_listener(self, change_listener: Callable[[], Any]) -> None: ...
     _logger: Incomplete
     def _set_logger(self, logger: logging.Logger | None = None) -> None: ...
     def update_logger(self, logger: logging.Logger | None = None) -> None: ...
@@ -224,6 +232,8 @@ class Script:
     async def _async_get_if_data(self, step: int) -> _IfData: ...
     async def _async_prep_parallel_scripts(self, step: int) -> list[Script]: ...
     async def _async_get_parallel_scripts(self, step: int) -> list[Script]: ...
+    async def _async_prep_sequence_script(self, step: int) -> Script: ...
+    async def _async_get_sequence_script(self, step: int) -> Script: ...
     def _log(self, msg: str, *args: Any, level: int = ..., **kwargs: Any) -> None: ...
 
 def breakpoint_clear(hass: HomeAssistant, key: str, run_id: str | None, node: str) -> None: ...

homeassistant-stubs/helpers/selector.pyi

@@ -1,28 +1,27 @@
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Mapping, Sequence
-from enum import IntFlag, StrEnum
+from enum import StrEnum
 from homeassistant.const import CONF_MODE as CONF_MODE, CONF_UNIT_OF_MEASUREMENT as CONF_UNIT_OF_MEASUREMENT
 from homeassistant.core import split_entity_id as split_entity_id, valid_entity_id as valid_entity_id
 from homeassistant.generated.countries import COUNTRIES as COUNTRIES
 from homeassistant.util import decorator as decorator
 from homeassistant.util.yaml import dumper as dumper
-from typing import Any, Generic, Literal, Required, TypeVar, TypedDict
+from typing import Any, Literal, Required, TypedDict
 
 SELECTORS: decorator.Registry[str, type[Selector]]
-_T = TypeVar('_T', bound=Mapping[str, Any])
 
 def _get_selector_class(config: Any) -> type[Selector]: ...
 def selector(config: Any) -> Selector: ...
 def validate_selector(config: Any) -> dict: ...
 
-class Selector(Generic[_T]):
+class Selector:
     CONFIG_SCHEMA: Callable
     config: _T
     selector_type: str
     def __init__(self, config: Mapping[str, Any] | None = None) -> None: ...
     def serialize(self) -> dict[str, dict[str, _T]]: ...
 
-def _entity_features() -> dict[str, type[IntFlag]]: ...
+def _entity_feature_flag(domain: str, enum_name: str, feature_name: str) -> int: ...
 def _validate_supported_feature(supported_feature: str) -> int: ...
 def _validate_supported_features(supported_features: int | list[str]) -> int: ...