Move code around to try and fix test

esphome · Sep 9, 2024 · 30a97ac · 30a97ac
1 parent 420a96e
commit 30a97ac
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/aioesphomeapi/client.py b/aioesphomeapi/client.py
@@ -1373,13 +1373,14 @@ def _on_voice_assistant_audio(msg: VoiceAssistantAudio) -> None:
             )
         )
 
-        def _on_voice_assistant_announcement_finished(
-            msg: VoiceAssistantAnnounceFinished,
-        ) -> None:
-            finished = VoiceAssistantAnnounceFinishedModel.from_pb(msg)
-            self._create_background_task(handle_announcement_finished(finished))
-
         if handle_announcement_finished is not None:
+
+            def _on_voice_assistant_announcement_finished(
+                msg: VoiceAssistantAnnounceFinished,
+            ) -> None:
+                finished = VoiceAssistantAnnounceFinishedModel.from_pb(msg)
+                self._create_background_task(handle_announcement_finished(finished))
+
             remove_callbacks.append(
                 connection.add_message_callback(
                     _on_voice_assistant_announcement_finished,