add more log

pull/21891/head
ytqh 1 year ago
parent 775b2ad919
commit 3ec1cc0972

@ -230,26 +230,28 @@ class AdvancedChatAppGenerateTaskPipeline:
while (time.time() - start_listener_time) < TTS_AUTO_PLAY_TIMEOUT: while (time.time() - start_listener_time) < TTS_AUTO_PLAY_TIMEOUT:
try: try:
if not tts_publisher: if not tts_publisher:
logging.info(f"TTS: tts_publisher is None") logger.info(f"TTS: tts_publisher is None")
break break
audio_trunk = tts_publisher.check_and_get_audio() audio_trunk = tts_publisher.check_and_get_audio()
if audio_trunk is None: if audio_trunk is None:
# release cpu # release cpu
# sleep 20 ms ( 40ms => 1280 byte audio file,20ms => 640 byte audio file) # sleep 20 ms ( 40ms => 1280 byte audio file,20ms => 640 byte audio file)
time.sleep(TTS_AUTO_PLAY_YIELD_CPU_TIME) time.sleep(TTS_AUTO_PLAY_YIELD_CPU_TIME)
logger.info(f"TTS: audio trunk is None, waiting. message_id: {self._message_id}")
continue continue
if audio_trunk.status == "finish": if audio_trunk.status == "finish":
is_end_with_finish = True is_end_with_finish = True
break break
else: else:
logger.info(f"TTS: convert audio trunk to audio stream response. message_id: {self._message_id}")
start_listener_time = time.time() start_listener_time = time.time()
yield MessageAudioStreamResponse(audio=audio_trunk.audio, task_id=task_id) yield MessageAudioStreamResponse(audio=audio_trunk.audio, task_id=task_id)
except Exception as e: except Exception as e:
logger.exception(f"TTS: Failed to listen audio message, task_id: {task_id}") logger.exception(f"TTS: Failed to listen audio message, task_id: {task_id}")
break break
logging.info(f"TTS: audio end, is_end_with_finish: {is_end_with_finish}") logger.info(f"TTS: audio end, is_end_with_finish: {is_end_with_finish}")
if tts_publisher: if tts_publisher:
yield MessageAudioEndStreamResponse(audio="", task_id=task_id) yield MessageAudioEndStreamResponse(audio="", task_id=task_id)

Loading…
Cancel
Save