From 7f2e848a5c4a754805d1945521821e545175175a Mon Sep 17 00:00:00 2001 From: Martin Schweiger Date: Fri, 30 May 2025 09:40:46 +0800 Subject: [PATCH] use FrameProcessor class methods --- src/pipecat/services/assemblyai/stt.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/src/pipecat/services/assemblyai/stt.py b/src/pipecat/services/assemblyai/stt.py index ed4d9d55b..8f6569ffe 100644 --- a/src/pipecat/services/assemblyai/stt.py +++ b/src/pipecat/services/assemblyai/stt.py @@ -132,7 +132,7 @@ class AssemblyAISTTService(STTService): extra_headers=headers, ) self._connected = True - self._receive_task = asyncio.create_task(self._receive_task_handler()) + self._receive_task = self.create_task(self._receive_task_handler()) except Exception as e: logger.error(f"Failed to connect to AssemblyAI: {e}") self._connected = False @@ -166,11 +166,7 @@ class AssemblyAISTTService(STTService): logger.warning(f"Error during termination handshake: {e}") if self._receive_task: - self._receive_task.cancel() - try: - await self._receive_task - except asyncio.CancelledError: - pass + await self.cancel_task(self._receive_task) await self._websocket.close()