Improve conversation typing (#88440)

This commit is contained in:
Marc Mueller 2023-02-20 03:27:30 +01:00 committed by GitHub
parent c2b770bcb9
commit f3c6558db4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,7 +3,7 @@ from __future__ import annotations
import asyncio import asyncio
from collections import defaultdict from collections import defaultdict
from collections.abc import Iterable, Mapping from collections.abc import Iterable
from dataclasses import dataclass from dataclasses import dataclass
import logging import logging
from pathlib import Path from pathlib import Path
@ -83,7 +83,7 @@ class DefaultAgent(AbstractConversationAgent):
# intent -> [sentences] # intent -> [sentences]
self._config_intents: dict[str, Any] = {} self._config_intents: dict[str, Any] = {}
self._slot_lists: dict[str, TextSlotList] | None = None self._slot_lists: dict[str, SlotList] | None = None
async def async_initialize(self, config_intents): async def async_initialize(self, config_intents):
"""Initialize the default agent.""" """Initialize the default agent."""
@ -133,7 +133,7 @@ class DefaultAgent(AbstractConversationAgent):
conversation_id, conversation_id,
) )
slot_lists: Mapping[str, SlotList] = self._make_slot_lists() slot_lists = self._make_slot_lists()
result = await self.hass.async_add_executor_job( result = await self.hass.async_add_executor_job(
self._recognize, self._recognize,
@ -437,7 +437,7 @@ class DefaultAgent(AbstractConversationAgent):
return return
self._slot_lists = None self._slot_lists = None
def _make_slot_lists(self) -> Mapping[str, SlotList]: def _make_slot_lists(self) -> dict[str, SlotList]:
"""Create slot lists with areas and entity names/aliases.""" """Create slot lists with areas and entity names/aliases."""
if self._slot_lists is not None: if self._slot_lists is not None:
return self._slot_lists return self._slot_lists