homeassistant-stubs 2025.4.0b9__py3-none-any.whl → 2026.1.0b0__py3-none-any.whl

homeassistant-stubs/components/onkyo/const.pyi

@@ -1,81 +1,21 @@
 from _typeshed import Incomplete
-from enum import Enum
-from typing import Literal, Self
+from aioonkyo import HDMIOutputParam, InputSourceParam, ListeningModeParam
+from typing import Literal
 
 DOMAIN: str
 DEVICE_INTERVIEW_TIMEOUT: int
 DEVICE_DISCOVERY_TIMEOUT: int
-CONF_SOURCES: str
-CONF_RECEIVER_MAX_VOLUME: str
 type VolumeResolution = Literal[50, 80, 100, 200]
 OPTION_VOLUME_RESOLUTION: str
 OPTION_VOLUME_RESOLUTION_DEFAULT: VolumeResolution
 VOLUME_RESOLUTION_ALLOWED: tuple[VolumeResolution, ...]
 OPTION_MAX_VOLUME: str
 OPTION_MAX_VOLUME_DEFAULT: float
-
-class EnumWithMeaning(Enum):
-    value_meaning: str
-    def __new__(cls, value: str) -> Self: ...
-    @staticmethod
-    def _get_meanings() -> dict[str, str]: ...
-
 OPTION_INPUT_SOURCES: str
 OPTION_LISTENING_MODES: str
-_INPUT_SOURCE_MEANINGS: Incomplete
-
-class InputSource(EnumWithMeaning):
-    DVR = '00'
-    CBL = '01'
-    GAME = '02'
-    AUX = '03'
-    GAME2 = '04'
-    PC = '05'
-    VIDEO7 = '06'
-    EXTRA1 = '07'
-    EXTRA2 = '08'
-    EXTRA3 = '09'
-    DVD = '10'
-    STRM_BOX = '11'
-    TV = '12'
-    TAPE = '20'
-    TAPE2 = '21'
-    PHONO = '22'
-    CD = '23'
-    FM = '24'
-    AM = '25'
-    TUNER = '26'
-    MUSIC_SERVER = '27'
-    INTERNET_RADIO = '28'
-    USB = '29'
-    USB_REAR = '2A'
-    NETWORK = '2B'
-    AIRPLAY = '2D'
-    BLUETOOTH = '2E'
-    USB_DAC_IN = '2F'
-    MULTI_CH = '30'
-    XM = '31'
-    SIRIUS = '32'
-    DAB = '33'
-    UNIVERSAL_PORT = '40'
-    LINE = '41'
-    LINE2 = '42'
-    OPTICAL = '44'
-    COAXIAL = '45'
-    HDMI_5 = '55'
-    HDMI_6 = '56'
-    HDMI_7 = '57'
-    MAIN_SOURCE = '80'
-    @staticmethod
-    def _get_meanings() -> dict[str, str]: ...
-
-_LISTENING_MODE_MEANINGS: Incomplete
-
-class ListeningMode(EnumWithMeaning):
-    _ignore_ = 'ListeningMode _k _v _meaning'
-    ListeningMode = ...
-    @staticmethod
-    def _get_meanings() -> dict[str, str]: ...
-
+InputSource = InputSourceParam
+ListeningMode = ListeningModeParam
+HDMIOutput = HDMIOutputParam
 ZONES: Incomplete
-PYEISCP_COMMANDS: Incomplete
+LEGACY_HDMI_OUTPUT_MAPPING: Incomplete
+LEGACY_REV_HDMI_OUTPUT_MAPPING: Incomplete

homeassistant-stubs/components/onkyo/media_player.pyi

@@ -1,28 +1,18 @@
 import asyncio
 from . import OnkyoConfigEntry as OnkyoConfigEntry
-from .const import CONF_RECEIVER_MAX_VOLUME as CONF_RECEIVER_MAX_VOLUME, CONF_SOURCES as CONF_SOURCES, DOMAIN as DOMAIN, InputSource as InputSource, ListeningMode as ListeningMode, OPTION_MAX_VOLUME as OPTION_MAX_VOLUME, OPTION_VOLUME_RESOLUTION as OPTION_VOLUME_RESOLUTION, PYEISCP_COMMANDS as PYEISCP_COMMANDS, VolumeResolution as VolumeResolution, ZONES as ZONES
-from .receiver import Receiver as Receiver, async_discover as async_discover
+from .const import DOMAIN as DOMAIN, InputSource as InputSource, LEGACY_HDMI_OUTPUT_MAPPING as LEGACY_HDMI_OUTPUT_MAPPING, LEGACY_REV_HDMI_OUTPUT_MAPPING as LEGACY_REV_HDMI_OUTPUT_MAPPING, ListeningMode as ListeningMode, OPTION_MAX_VOLUME as OPTION_MAX_VOLUME, OPTION_VOLUME_RESOLUTION as OPTION_VOLUME_RESOLUTION, VolumeResolution as VolumeResolution, ZONES as ZONES
+from .receiver import ReceiverManager as ReceiverManager
 from .services import DATA_MP_ENTITIES as DATA_MP_ENTITIES
+from .util import get_meaning as get_meaning
 from _typeshed import Incomplete
-from enum import Enum as Enum
-from functools import cache
+from aioonkyo import Status as Status, Zone, status
 from homeassistant.components.media_player import MediaPlayerEntity as MediaPlayerEntity, MediaPlayerEntityFeature as MediaPlayerEntityFeature, MediaPlayerState as MediaPlayerState, MediaType as MediaType
-from homeassistant.config_entries import SOURCE_IMPORT as SOURCE_IMPORT
-from homeassistant.const import CONF_HOST as CONF_HOST, CONF_NAME as CONF_NAME
-from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.data_entry_flow import FlowResultType as FlowResultType
+from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ServiceValidationError as ServiceValidationError
-from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback, AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue
-from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
-from typing import Any, Literal
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from typing import Any
 
 _LOGGER: Incomplete
-CONF_MAX_VOLUME_DEFAULT: int
-CONF_RECEIVER_MAX_VOLUME_DEFAULT: int
-CONF_SOURCES_DEFAULT: Incomplete
-ISSUE_URL_PLACEHOLDER: str
-PLATFORM_SCHEMA: Incomplete
 SUPPORTED_FEATURES_BASE: Incomplete
 SUPPORTED_FEATURES_VOLUME: Incomplete
 PLAYABLE_SOURCES: Incomplete
@@ -30,58 +20,45 @@ ATTR_PRESET: str
 ATTR_AUDIO_INFORMATION: str
 ATTR_VIDEO_INFORMATION: str
 ATTR_VIDEO_OUT: str
-AUDIO_VIDEO_INFORMATION_UPDATE_WAIT_TIME: int
+QUERY_STATE_DELAY: int
+QUERY_AV_INFO_DELAY: int
 AUDIO_INFORMATION_MAPPING: Incomplete
 VIDEO_INFORMATION_MAPPING: Incomplete
-type LibValue = str | tuple[str, ...]
 
-def _get_single_lib_value(value: LibValue) -> str: ...
-def _get_lib_mapping[T: Enum](cmds: Any, cls: type[T]) -> dict[T, LibValue]: ...
-@cache
-def _input_source_lib_mappings(zone: str) -> dict[InputSource, LibValue]: ...
-@cache
-def _rev_input_source_lib_mappings(zone: str) -> dict[LibValue, InputSource]: ...
-@cache
-def _listening_mode_lib_mappings(zone: str) -> dict[ListeningMode, LibValue]: ...
-@cache
-def _rev_listening_mode_lib_mappings(zone: str) -> dict[LibValue, ListeningMode]: ...
-async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
 async def async_setup_entry(hass: HomeAssistant, entry: OnkyoConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class OnkyoMediaPlayer(MediaPlayerEntity):
     _attr_should_poll: bool
+    _attr_has_entity_name: bool
     _supports_volume: bool
-    _supports_sound_mode: bool
+    _supports_sound_mode: bool | None
     _supports_audio_info: bool
     _supports_video_info: bool
-    _query_timer: asyncio.TimerHandle | None
-    _receiver: Incomplete
+    _query_state_task: asyncio.Task | None
+    _query_av_info_task: asyncio.Task | None
+    _manager: Incomplete
+    _zone: Incomplete
     _attr_name: Incomplete
     _attr_unique_id: Incomplete
-    _zone: Incomplete
     _volume_resolution: Incomplete
     _max_volume: Incomplete
-    _options_sources: Incomplete
-    _source_lib_mapping: Incomplete
-    _rev_source_lib_mapping: Incomplete
     _source_mapping: Incomplete
     _rev_source_mapping: Incomplete
-    _options_sound_modes: Incomplete
-    _sound_mode_lib_mapping: Incomplete
-    _rev_sound_mode_lib_mapping: Incomplete
     _sound_mode_mapping: Incomplete
     _rev_sound_mode_mapping: Incomplete
+    _hdmi_output_mapping: Incomplete
+    _rev_hdmi_output_mapping: Incomplete
     _attr_source_list: Incomplete
     _attr_sound_mode_list: Incomplete
     _attr_supported_features: Incomplete
     _attr_extra_state_attributes: Incomplete
-    def __init__(self, receiver: Receiver, zone: str, *, volume_resolution: VolumeResolution, max_volume: float, sources: dict[InputSource, str], sound_modes: dict[ListeningMode, str]) -> None: ...
+    def __init__(self, manager: ReceiverManager, zone: Zone, *, volume_resolution: VolumeResolution, max_volume: float, sources: dict[InputSource, str], sound_modes: dict[ListeningMode, str]) -> None: ...
     async def async_added_to_hass(self) -> None: ...
     async def async_will_remove_from_hass(self) -> None: ...
-    @callback
-    def _update_receiver(self, propname: str, value: Any) -> None: ...
-    @callback
-    def _query_receiver(self, propname: str) -> None: ...
+    @property
+    def available(self) -> bool: ...
+    async def query_state(self) -> None: ...
+    def cancel_tasks(self) -> None: ...
     async def async_turn_on(self) -> None: ...
     async def async_turn_off(self) -> None: ...
     async def async_set_volume_level(self, volume: float) -> None: ...
@@ -92,21 +69,11 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
     async def async_select_sound_mode(self, sound_mode: str) -> None: ...
     async def async_select_output(self, hdmi_output: str) -> None: ...
     async def async_play_media(self, media_type: MediaType | str, media_id: str, **kwargs: Any) -> None: ...
-    @callback
-    def backfill_state(self) -> None: ...
     _attr_state: Incomplete
     _attr_volume_level: Incomplete
     _attr_is_volume_muted: Incomplete
-    @callback
-    def process_update(self, update: tuple[str, str, Any]) -> None: ...
     _attr_source: Incomplete
-    @callback
-    def _parse_source(self, source_lib: LibValue) -> None: ...
     _attr_sound_mode: Incomplete
-    @callback
-    def _parse_sound_mode(self, mode_lib: LibValue) -> None: ...
-    @callback
-    def _parse_audio_information(self, audio_information: tuple[str] | Literal['N/A']) -> None: ...
-    @callback
-    def _parse_video_information(self, video_information: tuple[str] | Literal['N/A']) -> None: ...
+    def process_update(self, message: status.Known) -> None: ...
+    def _query_state_delayed(self) -> None: ...
     def _query_av_info_delayed(self) -> None: ...

homeassistant-stubs/components/onkyo/receiver.pyi

@@ -1,36 +1,38 @@
-import pyeiscp
+import asyncio
+from . import OnkyoConfigEntry as OnkyoConfigEntry
 from .const import DEVICE_DISCOVERY_TIMEOUT as DEVICE_DISCOVERY_TIMEOUT, DEVICE_INTERVIEW_TIMEOUT as DEVICE_INTERVIEW_TIMEOUT, ZONES as ZONES
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Iterable
+from aioonkyo import Instruction as Instruction, Receiver as Receiver, ReceiverInfo as ReceiverInfo, Status as Status
+from collections.abc import Awaitable, Callable as Callable, Iterable
 from dataclasses import dataclass, field
-from typing import Any
+from homeassistant.components import network as network
+from homeassistant.core import HomeAssistant as HomeAssistant
 
 _LOGGER: Incomplete
 
 @dataclass
 class Callbacks:
-    connect: list[Callable[[Receiver], None]] = field(default_factory=list)
-    update: list[Callable[[Receiver, tuple[str, str, Any]], None]] = field(default_factory=list)
+    connect: list[Callable[[bool], Awaitable[None]]] = field(default_factory=list)
+    disconnect: list[Callable[[], Awaitable[None]]] = field(default_factory=list)
+    update: list[Callable[[Status], Awaitable[None]]] = field(default_factory=list)
+    def clear(self) -> None: ...
 
-@dataclass
-class Receiver:
-    conn: pyeiscp.Connection
-    model_name: str
-    identifier: str
-    host: str
-    first_connect: bool = ...
-    callbacks: Callbacks = field(default_factory=Callbacks)
-    @classmethod
-    async def async_create(cls, info: ReceiverInfo) -> Receiver: ...
-    def on_connect(self) -> None: ...
-    def on_update(self, message: tuple[str, str, Any]) -> None: ...
-
-@dataclass
-class ReceiverInfo:
-    host: str
-    port: int
-    model_name: str
-    identifier: str
+class ReceiverManager:
+    hass: HomeAssistant
+    entry: OnkyoConfigEntry
+    info: ReceiverInfo
+    receiver: Receiver | None
+    connected: bool
+    callbacks: Callbacks
+    _started: asyncio.Event
+    def __init__(self, hass: HomeAssistant, entry: OnkyoConfigEntry, info: ReceiverInfo) -> None: ...
+    async def start(self) -> Awaitable[None] | None: ...
+    async def _run(self) -> None: ...
+    async def on_connect(self, reconnect: bool) -> None: ...
+    async def on_disconnect(self) -> None: ...
+    async def on_update(self, message: Status) -> None: ...
+    async def write(self, message: Instruction) -> None: ...
+    def start_unloading(self) -> None: ...
 
-async def async_interview(host: str) -> ReceiverInfo | None: ...
-async def async_discover() -> Iterable[ReceiverInfo]: ...
+async def async_interview(host: str) -> ReceiverInfo: ...
+async def async_discover(hass: HomeAssistant) -> Iterable[ReceiverInfo]: ...

homeassistant-stubs/components/onkyo/services.pyi

@@ -1,14 +1,15 @@
-from .const import DOMAIN as DOMAIN
+from .const import DOMAIN as DOMAIN, LEGACY_REV_HDMI_OUTPUT_MAPPING as LEGACY_REV_HDMI_OUTPUT_MAPPING
 from .media_player import OnkyoMediaPlayer as OnkyoMediaPlayer
 from _typeshed import Incomplete
+from aioonkyo import Zone as Zone
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID
-from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall
+from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
 from homeassistant.util.hass_dict import HassKey as HassKey
 
-DATA_MP_ENTITIES: HassKey[dict[str, dict[str, OnkyoMediaPlayer]]]
+DATA_MP_ENTITIES: HassKey[dict[str, dict[Zone, OnkyoMediaPlayer]]]
 ATTR_HDMI_OUTPUT: str
-ACCEPTED_VALUES: Incomplete
 ONKYO_SELECT_OUTPUT_SCHEMA: Incomplete
 SERVICE_SELECT_HDMI_OUTPUT: str
 
-async def async_register_services(hass: HomeAssistant) -> None: ...
+@callback
+def async_setup_services(hass: HomeAssistant) -> None: ...

homeassistant-stubs/components/onkyo/util.pyi

@@ -0,0 +1,3 @@
+from .const import InputSource as InputSource, ListeningMode as ListeningMode
+
+def get_meaning(param: InputSource | ListeningMode) -> str: ...

homeassistant-stubs/components/open_router/__init__.pyi

@@ -0,0 +1,15 @@
+from .const import LOGGER as LOGGER
+from _typeshed import Incomplete
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import CONF_API_KEY as CONF_API_KEY, Platform as Platform
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import ConfigEntryError as ConfigEntryError, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.helpers.httpx_client import get_async_client as get_async_client
+from openai import AsyncOpenAI
+
+PLATFORMS: Incomplete
+type OpenRouterConfigEntry = ConfigEntry[AsyncOpenAI]
+
+async def async_setup_entry(hass: HomeAssistant, entry: OpenRouterConfigEntry) -> bool: ...
+async def _async_update_listener(hass: HomeAssistant, entry: OpenRouterConfigEntry) -> None: ...
+async def async_unload_entry(hass: HomeAssistant, entry: OpenRouterConfigEntry) -> bool: ...

homeassistant-stubs/components/open_router/ai_task.pyi

@@ -0,0 +1,17 @@
+from . import OpenRouterConfigEntry as OpenRouterConfigEntry
+from .entity import OpenRouterEntity as OpenRouterEntity
+from _typeshed import Incomplete
+from homeassistant.components import ai_task as ai_task, conversation as conversation
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from homeassistant.util.json import json_loads as json_loads
+
+_LOGGER: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: OpenRouterConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class OpenRouterAITaskEntity(ai_task.AITaskEntity, OpenRouterEntity):
+    _attr_name: Incomplete
+    _attr_supported_features: Incomplete
+    async def _async_generate_data(self, task: ai_task.GenDataTask, chat_log: conversation.ChatLog) -> ai_task.GenDataTaskResult: ...

homeassistant-stubs/components/open_router/config_flow.pyi

@@ -0,0 +1,42 @@
+from .const import CONF_PROMPT as CONF_PROMPT, DOMAIN as DOMAIN, RECOMMENDED_CONVERSATION_OPTIONS as RECOMMENDED_CONVERSATION_OPTIONS
+from _typeshed import Incomplete
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, ConfigSubentryFlow as ConfigSubentryFlow, SOURCE_USER as SOURCE_USER, SubentryFlowResult as SubentryFlowResult
+from homeassistant.const import CONF_API_KEY as CONF_API_KEY, CONF_LLM_HASS_API as CONF_LLM_HASS_API, CONF_MODEL as CONF_MODEL
+from homeassistant.core import callback as callback
+from homeassistant.helpers import llm as llm
+from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
+from homeassistant.helpers.selector import SelectOptionDict as SelectOptionDict, SelectSelector as SelectSelector, SelectSelectorConfig as SelectSelectorConfig, SelectSelectorMode as SelectSelectorMode, TemplateSelector as TemplateSelector
+from python_open_router import Model as Model
+from typing import Any
+
+_LOGGER: Incomplete
+
+class OpenRouterConfigFlow(ConfigFlow, domain=DOMAIN):
+    VERSION: int
+    @classmethod
+    @callback
+    def async_get_supported_subentry_types(cls, config_entry: ConfigEntry) -> dict[str, type[ConfigSubentryFlow]]: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+
+class OpenRouterSubentryFlowHandler(ConfigSubentryFlow):
+    models: dict[str, Model]
+    def __init__(self) -> None: ...
+    async def _get_models(self) -> None: ...
+
+class ConversationFlowHandler(OpenRouterSubentryFlowHandler):
+    options: dict[str, Any]
+    def __init__(self) -> None: ...
+    @property
+    def _is_new(self) -> bool: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...
+    async def async_step_reconfigure(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...
+    async def async_step_init(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...
+
+class AITaskDataFlowHandler(OpenRouterSubentryFlowHandler):
+    options: dict[str, Any]
+    def __init__(self) -> None: ...
+    @property
+    def _is_new(self) -> bool: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...
+    async def async_step_reconfigure(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...
+    async def async_step_init(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...

homeassistant-stubs/components/open_router/const.pyi

@@ -0,0 +1,8 @@
+from _typeshed import Incomplete
+from homeassistant.const import CONF_LLM_HASS_API as CONF_LLM_HASS_API, CONF_PROMPT as CONF_PROMPT
+from homeassistant.helpers import llm as llm
+
+DOMAIN: str
+LOGGER: Incomplete
+CONF_RECOMMENDED: str
+RECOMMENDED_CONVERSATION_OPTIONS: Incomplete

homeassistant-stubs/components/open_router/conversation.pyi

@@ -0,0 +1,20 @@
+from . import OpenRouterConfigEntry as OpenRouterConfigEntry
+from .const import DOMAIN as DOMAIN
+from .entity import OpenRouterEntity as OpenRouterEntity
+from _typeshed import Incomplete
+from homeassistant.components import conversation as conversation
+from homeassistant.config_entries import ConfigSubentry as ConfigSubentry
+from homeassistant.const import CONF_LLM_HASS_API as CONF_LLM_HASS_API, CONF_PROMPT as CONF_PROMPT, MATCH_ALL as MATCH_ALL
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from typing import Literal
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: OpenRouterConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class OpenRouterConversationEntity(OpenRouterEntity, conversation.ConversationEntity):
+    _attr_name: Incomplete
+    _attr_supported_features: Incomplete
+    def __init__(self, entry: OpenRouterConfigEntry, subentry: ConfigSubentry) -> None: ...
+    @property
+    def supported_languages(self) -> list[str] | Literal['*']: ...
+    async def _async_handle_message(self, user_input: conversation.ConversationInput, chat_log: conversation.ChatLog) -> conversation.ConversationResult: ...

homeassistant-stubs/components/open_router/entity.pyi

@@ -0,0 +1,36 @@
+import voluptuous as vol
+from . import OpenRouterConfigEntry as OpenRouterConfigEntry
+from .const import DOMAIN as DOMAIN, LOGGER as LOGGER
+from _typeshed import Incomplete
+from collections.abc import AsyncGenerator, Callable as Callable
+from homeassistant.components import conversation as conversation
+from homeassistant.config_entries import ConfigSubentry as ConfigSubentry
+from homeassistant.const import CONF_MODEL as CONF_MODEL
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers import llm as llm
+from homeassistant.helpers.entity import Entity as Entity
+from openai.types.chat import ChatCompletionContentPartImageParam as ChatCompletionContentPartImageParam, ChatCompletionFunctionToolParam, ChatCompletionMessage as ChatCompletionMessage, ChatCompletionMessageParam as ChatCompletionMessageParam
+from openai.types.shared_params.response_format_json_schema import JSONSchema as JSONSchema
+from pathlib import Path
+from typing import Any
+
+MAX_TOOL_ITERATIONS: int
+
+def _adjust_schema(schema: dict[str, Any]) -> None: ...
+def _format_structured_output(name: str, schema: vol.Schema, llm_api: llm.APIInstance | None) -> JSONSchema: ...
+def _format_tool(tool: llm.Tool, custom_serializer: Callable[[Any], Any] | None) -> ChatCompletionFunctionToolParam: ...
+def _convert_content_to_chat_message(content: conversation.Content) -> ChatCompletionMessageParam | None: ...
+def _decode_tool_arguments(arguments: str) -> Any: ...
+async def _transform_response(message: ChatCompletionMessage) -> AsyncGenerator[conversation.AssistantContentDeltaDict]: ...
+async def async_prepare_files_for_prompt(hass: HomeAssistant, files: list[tuple[Path, str | None]]) -> list[ChatCompletionContentPartImageParam]: ...
+
+class OpenRouterEntity(Entity):
+    _attr_has_entity_name: bool
+    entry: Incomplete
+    subentry: Incomplete
+    model: Incomplete
+    _attr_unique_id: Incomplete
+    _attr_device_info: Incomplete
+    def __init__(self, entry: OpenRouterConfigEntry, subentry: ConfigSubentry) -> None: ...
+    async def _async_handle_chat_log(self, chat_log: conversation.ChatLog, structure_name: str | None = None, structure: vol.Schema | None = None) -> None: ...

homeassistant-stubs/components/openai_conversation/__init__.pyi

@@ -1,7 +1,8 @@
 import openai
-from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_FILENAMES as CONF_FILENAMES, CONF_MAX_TOKENS as CONF_MAX_TOKENS, CONF_PROMPT as CONF_PROMPT, CONF_REASONING_EFFORT as CONF_REASONING_EFFORT, CONF_TEMPERATURE as CONF_TEMPERATURE, CONF_TOP_P as CONF_TOP_P, DOMAIN as DOMAIN, LOGGER as LOGGER, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_MAX_TOKENS as RECOMMENDED_MAX_TOKENS, RECOMMENDED_REASONING_EFFORT as RECOMMENDED_REASONING_EFFORT, RECOMMENDED_TEMPERATURE as RECOMMENDED_TEMPERATURE, RECOMMENDED_TOP_P as RECOMMENDED_TOP_P
+from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_FILENAMES as CONF_FILENAMES, CONF_MAX_TOKENS as CONF_MAX_TOKENS, CONF_PROMPT as CONF_PROMPT, CONF_REASONING_EFFORT as CONF_REASONING_EFFORT, CONF_TEMPERATURE as CONF_TEMPERATURE, CONF_TOP_P as CONF_TOP_P, DEFAULT_AI_TASK_NAME as DEFAULT_AI_TASK_NAME, DEFAULT_NAME as DEFAULT_NAME, DOMAIN as DOMAIN, LOGGER as LOGGER, RECOMMENDED_AI_TASK_OPTIONS as RECOMMENDED_AI_TASK_OPTIONS, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_MAX_TOKENS as RECOMMENDED_MAX_TOKENS, RECOMMENDED_REASONING_EFFORT as RECOMMENDED_REASONING_EFFORT, RECOMMENDED_TEMPERATURE as RECOMMENDED_TEMPERATURE, RECOMMENDED_TOP_P as RECOMMENDED_TOP_P
+from .entity import async_prepare_files_for_prompt as async_prepare_files_for_prompt
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigSubentry as ConfigSubentry
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, ServiceResponse as ServiceResponse, SupportsResponse as SupportsResponse
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady, HomeAssistantError as HomeAssistantError, ServiceValidationError as ServiceValidationError
@@ -17,7 +18,10 @@ PLATFORMS: Incomplete
 CONFIG_SCHEMA: Incomplete
 type OpenAIConfigEntry = ConfigEntry[openai.AsyncClient]
 
-def encode_file(file_path: str) -> tuple[str, str]: ...
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
 async def async_setup_entry(hass: HomeAssistant, entry: OpenAIConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_update_options(hass: HomeAssistant, entry: OpenAIConfigEntry) -> None: ...
+async def async_migrate_integration(hass: HomeAssistant) -> None: ...
+async def async_migrate_entry(hass: HomeAssistant, entry: OpenAIConfigEntry) -> bool: ...
+def _add_ai_task_subentry(hass: HomeAssistant, entry: OpenAIConfigEntry) -> None: ...

homeassistant-stubs/components/openai_conversation/ai_task.pyi

@@ -0,0 +1,20 @@
+from . import OpenAIConfigEntry as OpenAIConfigEntry
+from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_IMAGE_MODEL as CONF_IMAGE_MODEL, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_IMAGE_MODEL as RECOMMENDED_IMAGE_MODEL, UNSUPPORTED_IMAGE_MODELS as UNSUPPORTED_IMAGE_MODELS
+from .entity import OpenAIBaseLLMEntity as OpenAIBaseLLMEntity
+from _typeshed import Incomplete
+from homeassistant.components import ai_task as ai_task, conversation as conversation
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigSubentry as ConfigSubentry
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from homeassistant.util.json import json_loads as json_loads
+
+_LOGGER: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class OpenAITaskEntity(ai_task.AITaskEntity, OpenAIBaseLLMEntity):
+    _attr_supported_features: Incomplete
+    def __init__(self, entry: OpenAIConfigEntry, subentry: ConfigSubentry) -> None: ...
+    async def _async_generate_data(self, task: ai_task.GenDataTask, chat_log: conversation.ChatLog) -> ai_task.GenDataTaskResult: ...
+    async def _async_generate_image(self, task: ai_task.GenImageTask, chat_log: conversation.ChatLog) -> ai_task.GenImageTaskResult: ...

homeassistant-stubs/components/openai_conversation/config_flow.pyi

@@ -1,32 +1,37 @@
-from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_MAX_TOKENS as CONF_MAX_TOKENS, CONF_PROMPT as CONF_PROMPT, CONF_REASONING_EFFORT as CONF_REASONING_EFFORT, CONF_RECOMMENDED as CONF_RECOMMENDED, CONF_TEMPERATURE as CONF_TEMPERATURE, CONF_TOP_P as CONF_TOP_P, CONF_WEB_SEARCH as CONF_WEB_SEARCH, CONF_WEB_SEARCH_CITY as CONF_WEB_SEARCH_CITY, CONF_WEB_SEARCH_CONTEXT_SIZE as CONF_WEB_SEARCH_CONTEXT_SIZE, CONF_WEB_SEARCH_COUNTRY as CONF_WEB_SEARCH_COUNTRY, CONF_WEB_SEARCH_REGION as CONF_WEB_SEARCH_REGION, CONF_WEB_SEARCH_TIMEZONE as CONF_WEB_SEARCH_TIMEZONE, CONF_WEB_SEARCH_USER_LOCATION as CONF_WEB_SEARCH_USER_LOCATION, DOMAIN as DOMAIN, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_MAX_TOKENS as RECOMMENDED_MAX_TOKENS, RECOMMENDED_REASONING_EFFORT as RECOMMENDED_REASONING_EFFORT, RECOMMENDED_TEMPERATURE as RECOMMENDED_TEMPERATURE, RECOMMENDED_TOP_P as RECOMMENDED_TOP_P, RECOMMENDED_WEB_SEARCH as RECOMMENDED_WEB_SEARCH, RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE as RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE, RECOMMENDED_WEB_SEARCH_USER_LOCATION as RECOMMENDED_WEB_SEARCH_USER_LOCATION, UNSUPPORTED_MODELS as UNSUPPORTED_MODELS
+from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_CODE_INTERPRETER as CONF_CODE_INTERPRETER, CONF_IMAGE_MODEL as CONF_IMAGE_MODEL, CONF_MAX_TOKENS as CONF_MAX_TOKENS, CONF_PROMPT as CONF_PROMPT, CONF_REASONING_EFFORT as CONF_REASONING_EFFORT, CONF_RECOMMENDED as CONF_RECOMMENDED, CONF_TEMPERATURE as CONF_TEMPERATURE, CONF_TOP_P as CONF_TOP_P, CONF_VERBOSITY as CONF_VERBOSITY, CONF_WEB_SEARCH as CONF_WEB_SEARCH, CONF_WEB_SEARCH_CITY as CONF_WEB_SEARCH_CITY, CONF_WEB_SEARCH_CONTEXT_SIZE as CONF_WEB_SEARCH_CONTEXT_SIZE, CONF_WEB_SEARCH_COUNTRY as CONF_WEB_SEARCH_COUNTRY, CONF_WEB_SEARCH_INLINE_CITATIONS as CONF_WEB_SEARCH_INLINE_CITATIONS, CONF_WEB_SEARCH_REGION as CONF_WEB_SEARCH_REGION, CONF_WEB_SEARCH_TIMEZONE as CONF_WEB_SEARCH_TIMEZONE, CONF_WEB_SEARCH_USER_LOCATION as CONF_WEB_SEARCH_USER_LOCATION, DEFAULT_AI_TASK_NAME as DEFAULT_AI_TASK_NAME, DEFAULT_CONVERSATION_NAME as DEFAULT_CONVERSATION_NAME, DOMAIN as DOMAIN, RECOMMENDED_AI_TASK_OPTIONS as RECOMMENDED_AI_TASK_OPTIONS, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_CODE_INTERPRETER as RECOMMENDED_CODE_INTERPRETER, RECOMMENDED_CONVERSATION_OPTIONS as RECOMMENDED_CONVERSATION_OPTIONS, RECOMMENDED_IMAGE_MODEL as RECOMMENDED_IMAGE_MODEL, RECOMMENDED_MAX_TOKENS as RECOMMENDED_MAX_TOKENS, RECOMMENDED_REASONING_EFFORT as RECOMMENDED_REASONING_EFFORT, RECOMMENDED_TEMPERATURE as RECOMMENDED_TEMPERATURE, RECOMMENDED_TOP_P as RECOMMENDED_TOP_P, RECOMMENDED_VERBOSITY as RECOMMENDED_VERBOSITY, RECOMMENDED_WEB_SEARCH as RECOMMENDED_WEB_SEARCH, RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE as RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE, RECOMMENDED_WEB_SEARCH_INLINE_CITATIONS as RECOMMENDED_WEB_SEARCH_INLINE_CITATIONS, RECOMMENDED_WEB_SEARCH_USER_LOCATION as RECOMMENDED_WEB_SEARCH_USER_LOCATION, UNSUPPORTED_CODE_INTERPRETER_MODELS as UNSUPPORTED_CODE_INTERPRETER_MODELS, UNSUPPORTED_IMAGE_MODELS as UNSUPPORTED_IMAGE_MODELS, UNSUPPORTED_MODELS as UNSUPPORTED_MODELS, UNSUPPORTED_WEB_SEARCH_MODELS as UNSUPPORTED_WEB_SEARCH_MODELS
 from _typeshed import Incomplete
 from homeassistant.components.zone import ENTITY_ID_HOME as ENTITY_ID_HOME
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
-from homeassistant.const import ATTR_LATITUDE as ATTR_LATITUDE, ATTR_LONGITUDE as ATTR_LONGITUDE, CONF_API_KEY as CONF_API_KEY, CONF_LLM_HASS_API as CONF_LLM_HASS_API
-from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, ConfigSubentryFlow as ConfigSubentryFlow, SubentryFlowResult as SubentryFlowResult
+from homeassistant.const import ATTR_LATITUDE as ATTR_LATITUDE, ATTR_LONGITUDE as ATTR_LONGITUDE, CONF_API_KEY as CONF_API_KEY, CONF_LLM_HASS_API as CONF_LLM_HASS_API, CONF_NAME as CONF_NAME
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers import llm as llm
 from homeassistant.helpers.httpx_client import get_async_client as get_async_client
 from homeassistant.helpers.selector import NumberSelector as NumberSelector, NumberSelectorConfig as NumberSelectorConfig, SelectOptionDict as SelectOptionDict, SelectSelector as SelectSelector, SelectSelectorConfig as SelectSelectorConfig, SelectSelectorMode as SelectSelectorMode, TemplateSelector as TemplateSelector
 from homeassistant.helpers.typing import VolDictType as VolDictType
-from types import MappingProxyType
 from typing import Any
 
 _LOGGER: Incomplete
 STEP_USER_DATA_SCHEMA: Incomplete
-RECOMMENDED_OPTIONS: Incomplete
 
 async def validate_input(hass: HomeAssistant, data: dict[str, Any]) -> None: ...
 
 class OpenAIConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
+    MINOR_VERSION: int
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
-    @staticmethod
-    def async_get_options_flow(config_entry: ConfigEntry) -> OptionsFlow: ...
+    @classmethod
+    @callback
+    def async_get_supported_subentry_types(cls, config_entry: ConfigEntry) -> dict[str, type[ConfigSubentryFlow]]: ...
 
-class OpenAIOptionsFlow(OptionsFlow):
-    last_rendered_recommended: Incomplete
-    def __init__(self, config_entry: ConfigEntry) -> None: ...
-    async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
-    async def get_location_data(self) -> dict[str, str]: ...
-
-def openai_config_option_schema(hass: HomeAssistant, options: dict[str, Any] | MappingProxyType[str, Any]) -> VolDictType: ...
+class OpenAISubentryFlowHandler(ConfigSubentryFlow):
+    last_rendered_recommended: bool
+    options: dict[str, Any]
+    @property
+    def _is_new(self) -> bool: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...
+    async def async_step_reconfigure(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...
+    async def async_step_init(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...
+    async def async_step_advanced(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...
+    async def async_step_model(self, user_input: dict[str, Any] | None = None) -> SubentryFlowResult: ...
+    def _get_reasoning_options(self, model: str) -> list[str]: ...
+    async def _get_location_data(self) -> dict[str, str]: ...

homeassistant-stubs/components/openai_conversation/const.pyi

@@ -1,8 +1,16 @@
 import logging
+from _typeshed import Incomplete
+from homeassistant.const import CONF_LLM_HASS_API as CONF_LLM_HASS_API
+from homeassistant.helpers import llm as llm
 
 DOMAIN: str
 LOGGER: logging.Logger
+DEFAULT_CONVERSATION_NAME: str
+DEFAULT_AI_TASK_NAME: str
+DEFAULT_NAME: str
 CONF_CHAT_MODEL: str
+CONF_IMAGE_MODEL: str
+CONF_CODE_INTERPRETER: str
 CONF_FILENAMES: str
 CONF_MAX_TOKENS: str
 CONF_PROMPT: str
@@ -10,6 +18,7 @@ CONF_REASONING_EFFORT: str
 CONF_RECOMMENDED: str
 CONF_TEMPERATURE: str
 CONF_TOP_P: str
+CONF_VERBOSITY: str
 CONF_WEB_SEARCH: str
 CONF_WEB_SEARCH_USER_LOCATION: str
 CONF_WEB_SEARCH_CONTEXT_SIZE: str
@@ -17,12 +26,23 @@ CONF_WEB_SEARCH_CITY: str
 CONF_WEB_SEARCH_REGION: str
 CONF_WEB_SEARCH_COUNTRY: str
 CONF_WEB_SEARCH_TIMEZONE: str
+CONF_WEB_SEARCH_INLINE_CITATIONS: str
+RECOMMENDED_CODE_INTERPRETER: bool
 RECOMMENDED_CHAT_MODEL: str
+RECOMMENDED_IMAGE_MODEL: str
 RECOMMENDED_MAX_TOKENS: int
 RECOMMENDED_REASONING_EFFORT: str
 RECOMMENDED_TEMPERATURE: float
 RECOMMENDED_TOP_P: float
+RECOMMENDED_VERBOSITY: str
 RECOMMENDED_WEB_SEARCH: bool
 RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE: str
 RECOMMENDED_WEB_SEARCH_USER_LOCATION: bool
+RECOMMENDED_WEB_SEARCH_INLINE_CITATIONS: bool
 UNSUPPORTED_MODELS: list[str]
+UNSUPPORTED_WEB_SEARCH_MODELS: list[str]
+UNSUPPORTED_IMAGE_MODELS: list[str]
+UNSUPPORTED_CODE_INTERPRETER_MODELS: list[str]
+UNSUPPORTED_EXTENDED_CACHE_RETENTION_MODELS: list[str]
+RECOMMENDED_CONVERSATION_OPTIONS: Incomplete
+RECOMMENDED_AI_TASK_OPTIONS: Incomplete