elevenlabs: fix send_task not closing properly (#596)

livekit · Aug 7, 2024 · 58c9e3e · 58c9e3e
1 parent e167424
commit 58c9e3e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/.changeset/hip-comics-bow.md b/.changeset/hip-comics-bow.md
@@ -0,0 +1,5 @@
+---
+"livekit-plugins-elevenlabs": patch
+---
+
+elevenlabs: fix send_task not closing properly
diff --git a/livekit-plugins/livekit-plugins-elevenlabs/livekit/plugins/elevenlabs/tts.py b/livekit-plugins/livekit-plugins-elevenlabs/livekit/plugins/elevenlabs/tts.py
@@ -353,7 +353,15 @@ async def recv_task():
                     segment_id=segment_id,
                 )
 
-        await asyncio.gather(send_task(), recv_task())
+        tasks = [
+            asyncio.create_task(send_task()),
+            asyncio.create_task(recv_task()),
+        ]
+
+        try:
+            await asyncio.gather(*tasks)
+        finally:
+            await utils.aio.gracefully_cancel(*tasks)
 
     def _process_stream_event(
         self, *, data: dict, request_id: str, segment_id: str