wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Removing watchdog from SarvamTTSService.

Browse Source
This commit is contained in:
Filipi Fuchter
2025-08-26 18:29:10 -03:00
committed by Filipi da Silva Fuchter
parent c9aaa463b7
commit 76eef837b6
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/pipecat/services/sarvam/tts.py
View File

@@ -31,7 +31,6 @@ from pipecat.frames.frames import (
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.tts_service import InterruptibleTTSService, TTSService from pipecat.services.tts_service import InterruptibleTTSService, TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language
from pipecat.utils.asyncio.watchdog_async_iterator import WatchdogAsyncIterator
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
try: try:
@@ -478,7 +477,6 @@ class SarvamTTSService(InterruptibleTTSService):
if self._websocket and not self._keepalive_task: if self._websocket and not self._keepalive_task:
self._keepalive_task = self.create_task( self._keepalive_task = self.create_task(
self._keepalive_task_handler(), self._keepalive_task_handler(),
watchdog_timeout_secs=25,
) )
async def _disconnect(self): async def _disconnect(self):
@@ -561,9 +559,7 @@ class SarvamTTSService(InterruptibleTTSService):
async def _receive_messages(self): async def _receive_messages(self):
"""Receive and process messages from Sarvam WebSocket.""" """Receive and process messages from Sarvam WebSocket."""
async for message in WatchdogAsyncIterator( async for message in self._get_websocket():
self._get_websocket(), manager=self.task_manager
):
if isinstance(message, str): if isinstance(message, str):
msg = json.loads(message) msg = json.loads(message)
if msg.get("type") == "audio": if msg.get("type") == "audio":
@@ -586,7 +582,6 @@ class SarvamTTSService(InterruptibleTTSService):
"""Handle keepalive messages to maintain WebSocket connection.""" """Handle keepalive messages to maintain WebSocket connection."""
KEEPALIVE_SLEEP = 20 KEEPALIVE_SLEEP = 20
while True: while True:
self.reset_watchdog()
await asyncio.sleep(KEEPALIVE_SLEEP) await asyncio.sleep(KEEPALIVE_SLEEP)
await self._send_keepalive() await self._send_keepalive()