79 lines
3.0 KiB
Python
79 lines
3.0 KiB
Python
import asyncio
|
||
from utils.logger import session_logger
|
||
from services.interrupt_handler import interrupt_handler
|
||
from services.tts_error_manager import tts_error_manager
|
||
|
||
|
||
class InterruptionHelper:
|
||
|
||
|
||
@staticmethod
|
||
async def notify_client_tts_error(device_id: str, session_id: str):
|
||
|
||
try:
|
||
if not await tts_error_manager.can_send_error(device_id, session_id):
|
||
session_logger.info(device_id, session_id, "已经发送过TTS_ERROR，不再重复发送")
|
||
return False
|
||
|
||
from services.connection_manager import connection_manager
|
||
|
||
websocket = await connection_manager.get_connection(device_id)
|
||
if websocket and websocket.client_state.name == "CONNECTED":
|
||
await websocket.send_text("TTS_ERROR")
|
||
session_logger.info(device_id, session_id, "已通知客户端TTS错误")
|
||
return True
|
||
except Exception as e:
|
||
session_logger.error(device_id, session_id, f"发送TTS错误通知失败: {e}")
|
||
return False
|
||
|
||
@staticmethod
|
||
async def send_error_prompt_sound(device_id: str, session_id: str):
|
||
|
||
try:
|
||
from handlers.prompt_sound_handler import handle_prompt_sound_request
|
||
|
||
await handle_prompt_sound_request(device_id, "tts_error")
|
||
session_logger.info(device_id, session_id, "已发送错误提示音")
|
||
return True
|
||
except Exception as e:
|
||
session_logger.error(device_id, session_id, f"发送错误提示音失败: {e}")
|
||
return False
|
||
|
||
@staticmethod
|
||
async def handle_llm_timeout(device_id: str, session_id: str, llm_service=None):
|
||
|
||
session_key = (device_id, session_id)
|
||
await interrupt_handler.set_interrupt_state(session_key, True)
|
||
if llm_service and hasattr(llm_service, "closed"):
|
||
llm_service.closed = True
|
||
await InterruptionHelper.send_error_prompt_sound(device_id, session_id)
|
||
await InterruptionHelper.notify_client_tts_error(device_id, session_id)
|
||
session_logger.error(device_id, session_id, "已处理LLM超时")
|
||
|
||
@staticmethod
|
||
async def register_queue_cleanup(session_key: tuple, queue: asyncio.Queue):
|
||
|
||
if queue is None:
|
||
return
|
||
|
||
async def cleanup_queue():
|
||
try:
|
||
await queue.put(None)
|
||
while not queue.empty():
|
||
try:
|
||
queue.get_nowait()
|
||
queue.task_done()
|
||
except asyncio.QueueEmpty:
|
||
break
|
||
device_id, session_id = session_key
|
||
session_logger.info(device_id, session_id, "队列已清空")
|
||
except Exception as e:
|
||
device_id, session_id = session_key
|
||
session_logger.error(device_id, session_id, f"清空队列时出错: {e}")
|
||
|
||
await interrupt_handler.register_cleanup_handler(
|
||
session_key,
|
||
cleanup_queue,
|
||
priority=interrupt_handler.PRIORITY["QUEUE_CLEANUP"],
|
||
)
|