mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-22 23:23:55 +08:00
45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
from config.logger import setup_logging
|
|
from core.utils.util import check_model_key
|
|
|
|
TAG = __name__
|
|
logger = setup_logging()
|
|
|
|
|
|
def append_devices_to_prompt(conn):
|
|
if conn.intent_type == "function_call":
|
|
funcs = conn.config["Intent"][conn.config["selected_module"]["Intent"]].get(
|
|
"functions", []
|
|
)
|
|
if "hass_get_state" in funcs or "hass_set_state" in funcs:
|
|
prompt = "\n下面是我家智能设备列表（位置，设备名，entity_id），可以通过homeassistant控制\n"
|
|
devices = conn.config["plugins"]["home_assistant"].get("devices", [])
|
|
if len(devices) == 0:
|
|
return
|
|
for device in devices:
|
|
prompt += device + "\n"
|
|
conn.prompt += prompt
|
|
# 更新提示词
|
|
conn.dialogue.update_system_message(conn.prompt)
|
|
|
|
|
|
def initialize_hass_handler(conn):
|
|
ha_config = {}
|
|
if conn.load_function_plugin:
|
|
if conn.config["plugins"].get("home_assistant"):
|
|
ha_config["base_url"] = conn.config["plugins"]["home_assistant"].get(
|
|
"base_url"
|
|
)
|
|
ha_config["api_key"] = conn.config["plugins"]["home_assistant"].get(
|
|
"api_key"
|
|
)
|
|
check_model_key("home_assistant", ha_config.get("api_key"))
|
|
elif conn.config["plugins"].get("hass_get_state"):
|
|
ha_config["base_url"] = conn.config["plugins"]["hass_get_state"].get(
|
|
"base_url"
|
|
)
|
|
ha_config["api_key"] = conn.config["plugins"]["hass_get_state"].get(
|
|
"api_key"
|
|
)
|
|
check_model_key("home_assistant", ha_config.get("api_key"))
|
|
return ha_config
|