update:优化ha工具获取密钥的方式

main/xiaozhi-server/plugins_func/functions/hass_play_music.py

@@ -17,46 +17,43 @@ hass_play_music_function_desc = {
             "properties": {
                 "media_content_id": {
                     "type": "string",
-                    "description": "可以是音乐或有声书的专辑名称、歌曲名、演唱者,如果未指定就填random"
+                    "description": "可以是音乐或有声书的专辑名称、歌曲名、演唱者,如果未指定就填random",
                 },
                 "entity_id": {
                     "type": "string",
-                    "description": "需要操作的音箱的设备id,homeassistant里的entity_id,media_player开头"
-                }
+                    "description": "需要操作的音箱的设备id,homeassistant里的entity_id,media_player开头",
+                },
             },
-            "required": ["media_content_id", "entity_id"]
-        }
-    }
+            "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'):
+@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
+            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)
+        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):
-    HASS_CACHE = initialize_hass_handler(conn)
-    api_key = HASS_CACHE['api_key']
-    base_url = HASS_CACHE['base_url']
+    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
-    }
+    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}的音乐"