Add VoiceAssistantAudio message

aioesphomeapi/api.proto

@@ -1515,6 +1515,14 @@ message VoiceAssistantEventResponse {
   repeated VoiceAssistantEventData data = 2;
 }
 
+message VoiceAssistantAudio {
+  option (id) = 106;
+  option (source) = SOURCE_BOTH;
+  option (ifdef) = "USE_VOICE_ASSISTANT";
+
+  bytes data = 1;
+}
+
 // ==================== ALARM CONTROL PANEL ====================
 enum AlarmControlPanelState {
   ALARM_STATE_DISARMED = 0;

aioesphomeapi/client.py

@@ -71,6 +71,7 @@ from .api_pb2 import (  # type: ignore
     VoiceAssistantEventResponse,
     VoiceAssistantRequest,
     VoiceAssistantResponse,
+    VoiceAssistantAudio,
 )
 from .client_callbacks import (
     on_bluetooth_connections_free_response,
@@ -1296,6 +1297,31 @@ class APIClient:
 
         return unsub
 
+    def subscribe_voice_assistant_audio(
+        self,
+        handle_audio: Callable[
+            [VoiceAssistantAudio],
+            Coroutine[Any, Any, None],
+        ],
+    ) -> Callable[[], None]:
+        connection = self._get_connection()
+
+        audio_task: asyncio.Task[int | None] | None = None
+
+        def _on_voice_assistant_audio(msg: VoiceAssistantRequest) -> None:
+            nonlocal audio_task
+
+            audio = VoiceAssistantAudio.from_pb(msg)
+            self._create_background_task(handle_audio(audio))
+
+        connection.send_message(SubscribeVoiceAssistantRequest(subscribe=True))
+
+        remove_callback = connection.add_message_callback(
+            _on_voice_assistant_audio, (VoiceAssistantAudio,)
+        )
+
+        return remove_callback
+
     def _create_background_task(self, coro: Coroutine[Any, Any, None]) -> None:
         """Create a background task and add it to the background tasks set."""
         task = asyncio.create_task(coro)