jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-11 23:37:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adjust "Assist in progress" sensor in ESPHome (#126928)

Browse Source 
Adjust sensor
This commit is contained in:
Michael Hansen 2024-09-27 10:10:50 -05:00 committed by Franck Nijhof
parent 02e15a4ce7
commit c4f189863c
No known key found for this signature in database
GPG Key ID: D62583BA8AB11CA3
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
homeassistant/components/esphome/assist_satellite.py
View File

@ -315,6 +315,10 @@ class EsphomeAssistSatellite(
"code": event.data["code"], "code": event.data["code"],
"message": event.data["message"], "message": event.data["message"],
} }
elif event_type == VoiceAssistantEventType.VOICE_ASSISTANT_RUN_END:
if self._tts_streaming_task is None:
# No TTS
self.entry_data.async_set_assist_pipeline_state(False)
self.cli.send_voice_assistant_event(event_type, data_to_send) self.cli.send_voice_assistant_event(event_type, data_to_send)
@ -413,7 +417,6 @@ class EsphomeAssistSatellite(
# Run the pipeline # Run the pipeline
_LOGGER.debug("Running pipeline from %s to %s", start_stage, end_stage) _LOGGER.debug("Running pipeline from %s to %s", start_stage, end_stage)
self.entry_data.async_set_assist_pipeline_state(True)
self._pipeline_task = self.config_entry.async_create_background_task( self._pipeline_task = self.config_entry.async_create_background_task(
self.hass, self.hass,
self.async_accept_pipeline_from_satellite( self.async_accept_pipeline_from_satellite(
@ -443,7 +446,6 @@ class EsphomeAssistSatellite(
def handle_pipeline_finished(self) -> None: def handle_pipeline_finished(self) -> None:
"""Handle when pipeline has finished running.""" """Handle when pipeline has finished running."""
self.entry_data.async_set_assist_pipeline_state(False)
self._stop_udp_server() self._stop_udp_server()
_LOGGER.debug("Pipeline finished") _LOGGER.debug("Pipeline finished")
@ -561,6 +563,7 @@ class EsphomeAssistSatellite(
# State change # State change
self.tts_response_finished() self.tts_response_finished()
self.entry_data.async_set_assist_pipeline_state(False)
async def _wrap_audio_stream(self) -> AsyncIterable[bytes]: async def _wrap_audio_stream(self) -> AsyncIterable[bytes]:
"""Yield audio chunks from the queue until None.""" """Yield audio chunks from the queue until None."""