Add websocket type hints in conversation (#80535)

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)