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

services(azure): close the audio stream on exit

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-07-02 10:11:35 -07:00
parent 5bcece56f3
commit b9b0bcdcbd
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/pipecat/services/azure.py
View File

@@ -19,12 +19,11 @@ from pipecat.frames.frames import (
ErrorFrame, ErrorFrame,
Frame, Frame,
StartFrame, StartFrame,
StartInterruptionFrame,
SystemFrame, SystemFrame,
TranscriptionFrame, TranscriptionFrame,
URLImageRawFrame) URLImageRawFrame)
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.ai_services import AIService, AsyncAIService, TTSService, ImageGenService from pipecat.services.ai_services import AsyncAIService, TTSService, ImageGenService
from pipecat.services.openai import BaseOpenAILLMService from pipecat.services.openai import BaseOpenAILLMService
from loguru import logger from loguru import logger
@@ -83,7 +82,7 @@ class AzureTTSService(TTSService):
return True return True
async def run_tts(self, text: str) -> AsyncGenerator[Frame, None]: async def run_tts(self, text: str) -> AsyncGenerator[Frame, None]:
logger.debug(f"Generating TTS: {text}") logger.debug(f"Generating TTS: [{text}]")
await self.start_ttfb_metrics() await self.start_ttfb_metrics()
@@ -148,9 +147,11 @@ class AzureSTTService(AsyncAIService):
async def stop(self, frame: EndFrame): async def stop(self, frame: EndFrame):
self._speech_recognizer.stop_continuous_recognition_async() self._speech_recognizer.stop_continuous_recognition_async()
self._audio_stream.close()
async def cancel(self, frame: CancelFrame): async def cancel(self, frame: CancelFrame):
self._speech_recognizer.stop_continuous_recognition_async() self._speech_recognizer.stop_continuous_recognition_async()
self._audio_stream.close()
def _on_handle_recognized(self, event): def _on_handle_recognized(self, event):
if event.result.reason == ResultReason.RecognizedSpeech and len(event.result.text) > 0: if event.result.reason == ResultReason.RecognizedSpeech and len(event.result.text) > 0: