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", []
|
)
|
|
# 安全地获取插件配置
|
plugins_config = conn.config.get("plugins", {})
|
config_source = (
|
"home_assistant"
|
if plugins_config.get("home_assistant")
|
else "hass_state"
|
)
|
|
if "hass_get_state" in funcs or "hass_set_state" in funcs:
|
prompt = "\n下面是我家智能设备列表（位置，设备名，entity_id），可以通过homeassistant控制\n"
|
deviceStr = plugins_config.get(config_source, {}).get("devices", "")
|
conn.prompt += prompt + deviceStr + "\n"
|
# 更新提示词
|
conn.dialogue.update_system_message(conn.prompt)
|
|
|
def initialize_hass_handler(conn):
|
ha_config = {}
|
if not conn.load_function_plugin:
|
return ha_config
|
|
# 安全地获取插件配置
|
plugins_config = conn.config.get("plugins", {})
|
# 确定配置来源
|
config_source = (
|
"home_assistant" if plugins_config.get("home_assistant") else "hass_state"
|
)
|
if not plugins_config.get(config_source):
|
return ha_config
|
|
# 统一获取配置
|
plugin_config = plugins_config[config_source]
|
ha_config["base_url"] = plugin_config.get("base_url")
|
ha_config["api_key"] = plugin_config.get("api_key")
|
|
# 统一检查API密钥
|
model_key_msg = check_model_key("home_assistant", ha_config.get("api_key"))
|
if model_key_msg:
|
logger.bind(tag=TAG).error(model_key_msg)
|
|
return ha_config
|
