68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
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
|
||
from typing import TYPE_CHECKING
|
||
|
||
if TYPE_CHECKING:
|
||
from core.connection import ConnectionHandler
|
||
|
||
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: "ConnectionHandler", 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: "ConnectionHandler", 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}"
|