|
|
from plugins_func.register import register_function, ToolType, ActionResponse, Action
|
|
|
from plugins_func.functions.hass_init import initialize_hass_handler
|
|
|
from config.logger import setup_logging
|
|
|
import asyncio
|
|
|
import requests
|
|
|
|
|
|
TAG = __name__
|
|
|
logger = setup_logging()
|
|
|
|
|
|
hass_play_music_function_desc = {
|
|
|
"type": "function",
|
|
|
"function": {
|
|
|
"name": "hass_play_music",
|
|
|
"description": "用户想听音乐、有声书的时候使用，在房间的媒体播放器（media_player）里播放对应音频",
|
|
|
"parameters": {
|
|
|
"type": "object",
|
|
|
"properties": {
|
|
|
"media_content_id": {
|
|
|
"type": "string",
|
|
|
"description": "可以是音乐或有声书的专辑名称、歌曲名、演唱者,如果未指定就填random",
|
|
|
},
|
|
|
"entity_id": {
|
|
|
"type": "string",
|
|
|
"description": "需要操作的音箱的设备id,homeassistant里的entity_id,media_player开头",
|
|
|
},
|
|
|
},
|
|
|
"required": ["media_content_id", "entity_id"],
|
|
|
},
|
|
|
},
|
|
|
}
|
|
|
|
|
|
|
|
|
@register_function(
|
|
|
"hass_play_music", hass_play_music_function_desc, ToolType.SYSTEM_CTL
|
|
|
)
|
|
|
def hass_play_music(conn, entity_id="", media_content_id="random"):
|
|
|
try:
|
|
|
# 执行音乐播放命令
|
|
|
future = asyncio.run_coroutine_threadsafe(
|
|
|
handle_hass_play_music(conn, entity_id, media_content_id), conn.loop
|
|
|
)
|
|
|
ha_response = future.result()
|
|
|
return ActionResponse(
|
|
|
action=Action.RESPONSE, result="退出意图已处理", response=ha_response
|
|
|
)
|
|
|
except Exception as e:
|
|
|
logger.bind(tag=TAG).error(f"处理音乐意图错误: {e}")
|
|
|
|
|
|
|
|
|
async def handle_hass_play_music(conn, entity_id, media_content_id):
|
|
|
ha_config = initialize_hass_handler(conn)
|
|
|
api_key = ha_config.get("api_key")
|
|
|
base_url = ha_config.get("base_url")
|
|
|
url = f"{base_url}/api/services/music_assistant/play_media"
|
|
|
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
|
|
|
data = {"entity_id": entity_id, "media_id": media_content_id}
|
|
|
response = requests.post(url, headers=headers, json=data)
|
|
|
if response.status_code == 200:
|
|
|
return f"正在播放{media_content_id}的音乐"
|
|
|
else:
|
|
|
return f"音乐播放失败，错误码: {response.status_code}"
|
