84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
|
import json
|
||
|
|
|
||
|
|
from xiaozhi.services.protocols.typing import AbortReason, ListeningMode
|
||
|
|
|
||
|
|
|
||
|
|
class Protocol:
|
||
|
|
def __init__(self):
|
||
|
|
self.session_id = ""
|
||
|
|
self.on_incoming_json = None
|
||
|
|
self.on_incoming_audio = None
|
||
|
|
self.on_audio_channel_opened = None
|
||
|
|
self.on_audio_channel_closed = None
|
||
|
|
self.on_network_error = None
|
||
|
|
|
||
|
|
def on_incoming_json(self, callback):
|
||
|
|
"""设置JSON消息接收回调函数"""
|
||
|
|
self.on_incoming_json = callback
|
||
|
|
|
||
|
|
def on_incoming_audio(self, callback):
|
||
|
|
"""设置音频数据接收回调函数"""
|
||
|
|
self.on_incoming_audio = callback
|
||
|
|
|
||
|
|
def on_audio_channel_opened(self, callback):
|
||
|
|
"""设置音频通道打开回调函数"""
|
||
|
|
self.on_audio_channel_opened = callback
|
||
|
|
|
||
|
|
def on_audio_channel_closed(self, callback):
|
||
|
|
"""设置音频通道关闭回调函数"""
|
||
|
|
self.on_audio_channel_closed = callback
|
||
|
|
|
||
|
|
def on_network_error(self, callback):
|
||
|
|
"""设置网络错误回调函数"""
|
||
|
|
self.on_network_error = callback
|
||
|
|
|
||
|
|
async def send_text(self, message):
|
||
|
|
"""发送文本消息的抽象方法，需要在子类中实现"""
|
||
|
|
raise NotImplementedError("send_text方法必须由子类实现")
|
||
|
|
|
||
|
|
async def send_abort_speaking(self, reason):
|
||
|
|
"""发送中止语音的消息"""
|
||
|
|
message = {"session_id": self.session_id, "type": "abort"}
|
||
|
|
if reason == AbortReason.WAKE_WORD_DETECTED:
|
||
|
|
message["reason"] = "wake_word_detected"
|
||
|
|
await self.send_text(json.dumps(message))
|
||
|
|
|
||
|
|
|
||
|
|
async def send_start_listening(self, mode):
|
||
|
|
"""发送开始监听的消息"""
|
||
|
|
mode_map = {
|
||
|
|
ListeningMode.ALWAYS_ON: "realtime",
|
||
|
|
ListeningMode.AUTO_STOP: "auto",
|
||
|
|
ListeningMode.MANUAL: "manual",
|
||
|
|
}
|
||
|
|
message = {
|
||
|
|
"session_id": self.session_id,
|
||
|
|
"type": "listen",
|
||
|
|
"state": "start",
|
||
|
|
"mode": mode_map[mode],
|
||
|
|
}
|
||
|
|
await self.send_text(json.dumps(message))
|
||
|
|
|
||
|
|
async def send_stop_listening(self):
|
||
|
|
"""发送停止监听的消息"""
|
||
|
|
message = {"session_id": self.session_id, "type": "listen", "state": "stop"}
|
||
|
|
await self.send_text(json.dumps(message))
|
||
|
|
|
||
|
|
async def send_iot_descriptors(self, descriptors):
|
||
|
|
"""发送物联网设备描述信息"""
|
||
|
|
message = {
|
||
|
|
"session_id": self.session_id,
|
||
|
|
"type": "iot",
|
||
|
|
"descriptors": json.loads(descriptors),
|
||
|
|
}
|
||
|
|
await self.send_text(json.dumps(message))
|
||
|
|
|
||
|
|
async def send_iot_states(self, states):
|
||
|
|
"""发送物联网设备状态信息"""
|
||
|
|
message = {
|
||
|
|
"session_id": self.session_id,
|
||
|
|
"type": "iot",
|
||
|
|
"states": json.loads(states),
|
||
|
|
}
|
||
|
|
await self.send_text(json.dumps(message))
|