diff --git a/homeassistant/components/conversation/__init__.py b/homeassistant/components/conversation/__init__.py index 9fd6d1ad3e2..deab740909e 100644 --- a/homeassistant/components/conversation/__init__.py +++ b/homeassistant/components/conversation/__init__.py @@ -4,6 +4,7 @@ from __future__ import annotations from http import HTTPStatus import logging import re +from typing import Any import voluptuous as vol @@ -84,7 +85,11 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: {"type": "conversation/process", "text": str, vol.Optional("conversation_id"): str} ) @websocket_api.async_response -async def websocket_process(hass, connection, msg): +async def websocket_process( + hass: HomeAssistant, + connection: websocket_api.ActiveConnection, + msg: dict[str, Any], +) -> None: """Process text.""" connection.send_result( msg["id"], @@ -96,7 +101,11 @@ async def websocket_process(hass, connection, msg): @websocket_api.websocket_command({"type": "conversation/agent/info"}) @websocket_api.async_response -async def websocket_get_agent_info(hass, connection, msg): +async def websocket_get_agent_info( + hass: HomeAssistant, + connection: websocket_api.ActiveConnection, + msg: dict[str, Any], +) -> None: """Do we need onboarding.""" agent = await _get_agent(hass) @@ -111,7 +120,11 @@ async def websocket_get_agent_info(hass, connection, msg): @websocket_api.websocket_command({"type": "conversation/onboarding/set", "shown": bool}) @websocket_api.async_response -async def websocket_set_onboarding(hass, connection, msg): +async def websocket_set_onboarding( + hass: HomeAssistant, + connection: websocket_api.ActiveConnection, + msg: dict[str, Any], +) -> None: """Set onboarding status.""" agent = await _get_agent(hass) @@ -120,7 +133,7 @@ async def websocket_set_onboarding(hass, connection, msg): if success: connection.send_result(msg["id"]) else: - connection.send_error(msg["id"]) + connection.send_error(msg["id"], "error", "Failed to set onboarding") class ConversationProcessView(http.HomeAssistantView): @@ -165,7 +178,10 @@ async def _get_agent(hass: core.HomeAssistant) -> AbstractConversationAgent: async def _async_converse( - hass: core.HomeAssistant, text: str, conversation_id: str, context: core.Context + hass: core.HomeAssistant, + text: str, + conversation_id: str | None, + context: core.Context, ) -> intent.IntentResponse: """Process text and get intent.""" agent = await _get_agent(hass)