From 8fcca8c88b64d6b9a05490a55f593136bcb77aa6 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Mon, 10 Jan 2022 17:02:54 +0100 Subject: [PATCH] Allow None agent in conversation (#63761) --- homeassistant/components/almond/__init__.py | 2 +- homeassistant/components/conversation/__init__.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/almond/__init__.py b/homeassistant/components/almond/__init__.py index 0b5f7ec56b5..15c280d9c1e 100644 --- a/homeassistant/components/almond/__init__.py +++ b/homeassistant/components/almond/__init__.py @@ -221,7 +221,7 @@ async def _configure_almond_for_ha( await hass.auth.async_remove_refresh_token(token) -async def async_unload_entry(hass, entry): +async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Unload Almond.""" conversation.async_set_agent(hass, None) return True diff --git a/homeassistant/components/conversation/__init__.py b/homeassistant/components/conversation/__init__.py index a962c039d56..3263e2e77e0 100644 --- a/homeassistant/components/conversation/__init__.py +++ b/homeassistant/components/conversation/__init__.py @@ -1,4 +1,6 @@ """Support for functionality to have conversations with Home Assistant.""" +from __future__ import annotations + from http import HTTPStatus import logging import re @@ -48,7 +50,7 @@ async_register = bind_hass(async_register) @core.callback @bind_hass -def async_set_agent(hass: core.HomeAssistant, agent: AbstractConversationAgent): +def async_set_agent(hass: core.HomeAssistant, agent: AbstractConversationAgent | None): """Set the agent to handle the conversations.""" hass.data[DATA_AGENT] = agent