mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-21 22:53:56 +08:00
96 lines
3.6 KiB
Python
96 lines
3.6 KiB
Python
"""呼叫设备工具"""
|
|
import httpx
|
|
from config.logger import setup_logging
|
|
from plugins_func.register import register_function, ToolType, ActionResponse, Action
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from core.connection import ConnectionHandler
|
|
|
|
TAG = __name__
|
|
logger = setup_logging()
|
|
|
|
call_device_function_desc = {
|
|
"type": "function",
|
|
"function": {
|
|
"name": "call_device",
|
|
"description": (
|
|
"用于设备之间建立语音通话连接。"
|
|
"当用户说出以下意图时调用此工具：\n"
|
|
"1. 主动呼叫：用户说”呼叫XX/打电话给XX/连线XX/打给XX/帮我呼叫XX”时调用，nickname取XX。"
|
|
"例如：”呼叫张三”→nickname=”张三”、”帮我连线小陈”→nickname=”小陈”；\n"
|
|
"2. 接听来电：系统刚提示”您收到来自XX的来电，是否接听？”后，用户说”接听/接通/同意接听/同意连线/同意对话”时调用，"
|
|
"nickname取提示中的XX。\n"
|
|
"如果用户输入不是明确接听，也不是明确拒绝，不得调用call_device，必须先追问一次"
|
|
),
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"nickname": {"type": "string", "description": "目标设备的备注名，例如：小陈、小翁"},
|
|
},
|
|
"required": ["nickname"],
|
|
},
|
|
},
|
|
}
|
|
|
|
|
|
async def _request_api(url: str, params: dict, headers: dict):
|
|
async with httpx.AsyncClient(timeout=httpx.Timeout(10.0, connect=3.0)) as client:
|
|
return await client.get(url, params=params, headers=headers)
|
|
|
|
|
|
def _failed_reply(msg: str) -> ActionResponse:
|
|
return ActionResponse(action=Action.RESPONSE, response=msg)
|
|
|
|
|
|
def _is_answering(conn: "ConnectionHandler") -> bool:
|
|
"""检查是否为接听模式（conn.incoming_call 不为空）"""
|
|
return hasattr(conn, 'incoming_call') and conn.incoming_call is not None
|
|
|
|
|
|
@register_function("call_device", call_device_function_desc, ToolType.SYSTEM_CTL)
|
|
async def call_device(conn: "ConnectionHandler", nickname: str):
|
|
caller_mac = conn.headers.get("device-id")
|
|
if not caller_mac:
|
|
return _failed_reply("无法获取本机MAC地址")
|
|
|
|
api_config = conn.config.get("manager-api", {})
|
|
api_url = api_config.get("url")
|
|
api_secret = api_config.get("secret")
|
|
if not api_url or not api_secret:
|
|
logger.bind(tag=TAG).error("manager-api配置缺失")
|
|
return _failed_reply("配置错误，请稍后再试")
|
|
|
|
headers = {"Authorization": f"Bearer {api_secret}"}
|
|
|
|
# 区分是主动呼叫还是接听来电
|
|
is_answer = _is_answering(conn)
|
|
params = {"callerMac": caller_mac, "nickname": nickname}
|
|
if is_answer:
|
|
params["answer"] = "true"
|
|
|
|
# 查询通讯录并发起呼叫
|
|
try:
|
|
resp = await _request_api(
|
|
f"{api_url}/device/address-book/call",
|
|
params=params,
|
|
headers=headers,
|
|
)
|
|
result = resp.json()
|
|
except httpx.HTTPError as e:
|
|
logger.bind(tag=TAG).error(f"呼叫请求失败: {e}")
|
|
return _failed_reply("呼叫失败，请稍后再试")
|
|
|
|
if result.get("code") != 0:
|
|
return _failed_reply(result.get("msg", "呼叫失败"))
|
|
|
|
data = result.get("data", {})
|
|
if data.get("status") == "error":
|
|
return _failed_reply(data.get("message"))
|
|
|
|
if is_answer:
|
|
return ActionResponse(action=Action.NONE, response="已成功接听")
|
|
else:
|
|
conn.calling = True
|
|
return ActionResponse(action=Action.NONE, response=f"正在呼叫{nickname}，请等待对方接听")
|