From 713dcb7a4dedcc52163408e794bea404660722f8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aleix=20Conchillo=20Flaqu=C3=A9?= <aleix@daily.co>
Date: Tue, 15 Oct 2024 17:47:12 -0700
Subject: [PATCH] transports(daily): cancel messages task when canceling

---
 src/pipecat/transports/services/daily.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/pipecat/transports/services/daily.py b/src/pipecat/transports/services/daily.py
index 7c124e376..7c537bd53 100644
--- a/src/pipecat/transports/services/daily.py
+++ b/src/pipecat/transports/services/daily.py
@@ -740,13 +740,17 @@ class DailyOutputTransport(BaseOutputTransport):
             self._messages_task.cancel()
             await self._messages_task
             self._messages_task = None
-        self._messages_task = None
         # Leave the room.
         await self._client.leave()
 
     async def cancel(self, frame: CancelFrame):
         # Parent stop.
         await super().cancel(frame)
+        # Cancel messages task
+        if self._messages_task:
+            self._messages_task.cancel()
+            await self._messages_task
+            self._messages_task = None
         # Leave the room.
         await self._client.leave()