logging
This commit is contained in:
@@ -7,6 +7,7 @@
|
||||
import asyncio
|
||||
import base64
|
||||
import json
|
||||
import random
|
||||
import uuid
|
||||
from typing import AsyncGenerator, List, Optional, Union
|
||||
|
||||
@@ -222,6 +223,10 @@ class CartesiaTTSService(WordTTSService):
|
||||
async def _receive_task_handler(self):
|
||||
try:
|
||||
async for message in self._get_websocket():
|
||||
# Randomly cancel the asyncio task 1% of the time
|
||||
if random.random() < 0.01:
|
||||
logger.info(f"Cancelling task for {self} due to random chance")
|
||||
asyncio.current_task().cancel()
|
||||
msg = json.loads(message)
|
||||
if not msg or msg["context_id"] != self._context_id:
|
||||
continue
|
||||
@@ -256,6 +261,7 @@ class CartesiaTTSService(WordTTSService):
|
||||
logger.error(f"Cartesia error, unknown message type: {msg}")
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
# await self.push_error(ErrorFrame(f"{self} cancelled", True))
|
||||
except Exception as e:
|
||||
logger.error(f"{self} exception: {e}")
|
||||
|
||||
|
||||
@@ -71,6 +71,7 @@ class BaseInputTransport(FrameProcessor):
|
||||
return self._params.vad_analyzer
|
||||
|
||||
async def push_audio_frame(self, frame: InputAudioRawFrame):
|
||||
logger.info(f"Pushing audio qsize: {self._audio_in_queue.qsize()}")
|
||||
if self._params.audio_in_enabled or self._params.vad_enabled:
|
||||
await self._audio_in_queue.put(frame)
|
||||
|
||||
@@ -167,6 +168,7 @@ class BaseInputTransport(FrameProcessor):
|
||||
return vad_state
|
||||
|
||||
async def _audio_task_handler(self):
|
||||
logger.info("_audio_task_handler started")
|
||||
vad_state: VADState = VADState.QUIET
|
||||
while True:
|
||||
try:
|
||||
|
||||
@@ -106,6 +106,7 @@ class WebsocketServerInputTransport(BaseInputTransport):
|
||||
continue
|
||||
|
||||
if isinstance(frame, AudioRawFrame):
|
||||
logger.info("websocket_server")
|
||||
await self.push_audio_frame(
|
||||
InputAudioRawFrame(
|
||||
audio=frame.audio,
|
||||
|
||||
Reference in New Issue
Block a user