home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-03 04:35:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

Improve intent schema typing (#120442)

Browse Source
This commit is contained in:
Marc Mueller
2024-06-25 19:02:04 +02:00
committed by GitHub
parent b393024acd
commit 185e79fa1b
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
homeassistant/helpers/intent.py
View File

@@ -33,6 +33,7 @@ from . import (
entity_registry,
floor_registry,
)
from .typing import VolSchemaType
_LOGGER = logging.getLogger(__name__)
type _SlotsType = dict[str, Any]
@@ -807,8 +808,8 @@ class DynamicServiceIntentHandler(IntentHandler):
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_slots: dict[str | tuple[str, str], VolSchemaType] | None = None,
optional_slots: dict[str | tuple[str, str], VolSchemaType] | None = None,
required_domains: set[str] | None = None,
required_features: int | None = None,
required_states: set[str] | None = None,
@@ -824,7 +825,7 @@ class DynamicServiceIntentHandler(IntentHandler):
self.description = description
self.platforms = platforms
self.required_slots: dict[tuple[str, str], vol.Schema] = {}
self.required_slots: dict[tuple[str, str], VolSchemaType] = {}
if required_slots:
for key, value_schema in required_slots.items():
if isinstance(key, str):
@@ -833,7 +834,7 @@ class DynamicServiceIntentHandler(IntentHandler):
self.required_slots[key] = value_schema
self.optional_slots: dict[tuple[str, str], vol.Schema] = {}
self.optional_slots: dict[tuple[str, str], VolSchemaType] = {}
if optional_slots:
for key, value_schema in optional_slots.items():
if isinstance(key, str):
@@ -1107,8 +1108,8 @@ class ServiceIntentHandler(DynamicServiceIntentHandler):
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_slots: dict[str | tuple[str, str], VolSchemaType] | None = None,
optional_slots: dict[str | tuple[str, str], VolSchemaType] | None = None,
required_domains: set[str] | None = None,
required_features: int | None = None,
required_states: set[str] | None = None,