Refactor RimeNonJsonTTSService to extend WebsocketTTSService, enhancing WebSocket functionality and improving code clarity

This commit is contained in:
Gokul Js
2025-12-08 20:30:46 +05:30
parent 12093fcffc
commit 5bfea84bd5

View File

@@ -33,8 +33,8 @@ 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 ( from pipecat.services.tts_service import (
AudioContextWordTTSService, AudioContextWordTTSService,
InterruptibleTTSService,
TTSService, TTSService,
WebsocketTTSService,
) )
from pipecat.transcriptions import language from pipecat.transcriptions import language
from pipecat.transcriptions.language import Language, resolve_language from pipecat.transcriptions.language import Language, resolve_language
@@ -581,7 +581,7 @@ class RimeHttpTTSService(TTSService):
yield TTSStoppedFrame() yield TTSStoppedFrame()
class RimeNonJsonTTSService(InterruptibleTTSService): class RimeNonJsonTTSService(WebsocketTTSService):
"""Pipecat TTS service for Rime's non-JSON WebSocket API. """Pipecat TTS service for Rime's non-JSON WebSocket API.
This service enables Text-to-Speech synthesis over WebSocket endpoints This service enables Text-to-Speech synthesis over WebSocket endpoints
@@ -770,7 +770,7 @@ class RimeNonJsonTTSService(InterruptibleTTSService):
# Send EOS command to gracefully close # Send EOS command to gracefully close
await self._websocket.send("<EOS>") await self._websocket.send("<EOS>")
await self._websocket.close() await self._websocket.close()
logger.debug("Disconnected from Rime None json websocket") logger.debug("Disconnected from Rime non json websocket")
except Exception as e: except Exception as e:
logger.error(f"{self} exception: {e}") logger.error(f"{self} exception: {e}")
await self.push_error(ErrorFrame(error=f"{self} error: {e}")) await self.push_error(ErrorFrame(error=f"{self} error: {e}"))