|
|
from plugins_func.register import register_function,ToolType, ActionResponse, Action
|
|
|
from config.logger import setup_logging
|
|
|
|
|
|
TAG = __name__
|
|
|
logger = setup_logging()
|
|
|
|
|
|
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)
|
