mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-22 23:23:55 +08:00
57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
from plugins_func.register import register_function, ToolType, ActionResponse, Action
|
|
|
|
plugin_loader_function_desc = {
|
|
"type": "function",
|
|
"function": {
|
|
"name": "plugin_loader",
|
|
"description": "当用户想加载或卸载插件/function时，调用此函数：支持的插件列表为[plugins]",
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"oper": {"type": "string", "description": "load or unload"},
|
|
"name": {"type": "string", "description": "要加载或卸载的插件名字"},
|
|
},
|
|
"required": ["oper", "name"],
|
|
},
|
|
},
|
|
}
|
|
|
|
|
|
@register_function("plugin_loader", plugin_loader_function_desc, ToolType.SYSTEM_CTL)
|
|
def plugin_loader(conn, oper: str, name: str):
|
|
"""插件加载"""
|
|
if oper not in ["load", "unload"]:
|
|
return ActionResponse(
|
|
action=Action.RESPONSE, result="插件操作失败", response="不支持的操作"
|
|
)
|
|
|
|
cur_support = conn.func_handler.current_support_functions()
|
|
if oper == "load":
|
|
if name in cur_support:
|
|
return ActionResponse(
|
|
action=Action.RESPONSE,
|
|
result="插件加载失败",
|
|
response=f"{name}插件已加载,无需重复加载",
|
|
)
|
|
func = conn.func_handler.function_registry.register_function(name)
|
|
if not func:
|
|
return ActionResponse(
|
|
action=Action.RESPONSE, result="插件加载失败", response="插件未找到"
|
|
)
|
|
res = f"{name}插件加载成功"
|
|
else:
|
|
if name not in cur_support:
|
|
return ActionResponse(
|
|
action=Action.RESPONSE,
|
|
result="插件卸载失败",
|
|
response=f"{name}插件未加载",
|
|
)
|
|
bOK = conn.func_handler.function_registry.unregister_function(name)
|
|
if not bOK:
|
|
return ActionResponse(
|
|
action=Action.RESPONSE, result="插件卸载失败", response="插件未找到"
|
|
)
|
|
res = f"{name}插件卸载成功"
|
|
conn.func_handler.upload_functions_desc()
|
|
return ActionResponse(action=Action.RESPONSE, result="插件操作成功", response=res)
|