refactor: improve run-agent/run-agent scripts (#160)

This commit is contained in:
sigoden
2025-02-10 14:47:37 +08:00
committed by GitHub
parent e4f85a056c
commit fad2c60605
4 changed files with 12 additions and 28 deletions
+1 -6
View File
@@ -106,15 +106,10 @@ async function loadEnv(filePath) {
}
async function run(agentName, agentPath, agentFunc, agentData) {
let mod;
if (os.platform() === "win32") {
agentPath = `file://${agentPath}`;
}
try {
mod = await import(agentPath);
} catch {
throw new Error(`Unable to load agent tools at '${agentPath}'`);
}
const mod = await import(agentPath);
if (!mod || !mod[agentFunc]) {
throw new Error(`Not module function '${agentFunc}' at '${agentPath}'`);
}
+5 -8
View File
@@ -94,14 +94,11 @@ def load_env(file_path):
def run(agent_name, agent_path, agent_func, agent_data):
try:
spec = importlib.util.spec_from_file_location(
os.path.basename(agent_path), agent_path
)
mod = importlib.util.module_from_spec(spec)
spec.loader.exec_module(mod)
except:
raise Exception(f"Unable to load agent tools at '{agent_path}'")
spec = importlib.util.spec_from_file_location(
os.path.basename(agent_path), agent_path
)
mod = importlib.util.module_from_spec(spec)
spec.loader.exec_module(mod)
if not hasattr(mod, agent_func):
raise Exception(f"Not module function '{agent_func}' at '{agent_path}'")
+1 -6
View File
@@ -93,15 +93,10 @@ async function loadEnv(filePath) {
}
async function run(toolName, toolPath, toolFunc, toolData) {
let mod;
if (os.platform() === "win32") {
toolPath = `file://${toolPath}`;
}
try {
mod = await import(toolPath);
} catch {
throw new Error(`Unable to load tool at '${toolPath}'`);
}
const mod = await import(toolPath);
if (!mod || !mod[toolFunc]) {
throw new Error(`Not module function '${toolFunc}' at '${toolPath}'`);
}
+5 -8
View File
@@ -89,14 +89,11 @@ def load_env(file_path):
def run(tool_name, tool_path, tool_func, tool_data):
try:
spec = importlib.util.spec_from_file_location(
os.path.basename(tool_path), tool_path
)
mod = importlib.util.module_from_spec(spec)
spec.loader.exec_module(mod)
except:
raise Exception(f"Unable to load tool at '{tool_path}'")
spec = importlib.util.spec_from_file_location(
os.path.basename(tool_path), tool_path
)
mod = importlib.util.module_from_spec(spec)
spec.loader.exec_module(mod)
if not hasattr(mod, tool_func):
raise Exception(f"Not module function '{tool_func}' at '{tool_path}'")