Use HassKey in cloud integration (#120322)

This commit is contained in:
epenet 2024-06-24 21:02:08 +02:00 committed by GitHub
parent b223cb7bb9
commit ea09d0cbed
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
17 changed files with 109 additions and 102 deletions

View file

@ -2,7 +2,6 @@
from __future__ import annotations
import asyncio
import logging
from typing import Any
@ -31,7 +30,7 @@ from homeassistant.setup import async_when_setup
from .assist_pipeline import async_migrate_cloud_pipeline_engine
from .client import CloudClient
from .const import DATA_PLATFORMS_SETUP, DOMAIN, TTS_ENTITY_UNIQUE_ID
from .const import DATA_CLOUD, DATA_PLATFORMS_SETUP, DOMAIN, TTS_ENTITY_UNIQUE_ID
from .prefs import CloudPreferences
ATTR_GENDER = "gender"
@ -97,7 +96,7 @@ async def async_get_engine(
discovery_info: DiscoveryInfoType | None = None,
) -> CloudProvider:
"""Set up Cloud speech component."""
cloud: Cloud[CloudClient] = hass.data[DOMAIN]
cloud = hass.data[DATA_CLOUD]
cloud_provider = CloudProvider(cloud)
if discovery_info is not None:
discovery_info["platform_loaded"].set()
@ -110,9 +109,9 @@ async def async_setup_entry(
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up Home Assistant Cloud text-to-speech platform."""
tts_platform_loaded: asyncio.Event = hass.data[DATA_PLATFORMS_SETUP][Platform.TTS]
tts_platform_loaded = hass.data[DATA_PLATFORMS_SETUP][Platform.TTS]
tts_platform_loaded.set()
cloud: Cloud[CloudClient] = hass.data[DOMAIN]
cloud = hass.data[DATA_CLOUD]
async_add_entities([CloudTTSEntity(cloud)])