diff options
| author | sigoden <sigoden@gmail.com> | 2024-11-30 17:14:21 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-11-30 17:14:21 +0800 |
| commit | d2a537c3743963bfb6add44731dd3f3f6c785a2e (patch) | |
| tree | e499f1217135741622995ba253ad34ea9164f358 /scripts/run-agent.py | |
| parent | c4ba6ff379bdce768f6053671309e2e2f4a6fc79 (diff) | |
| download | llm-functions-docker-d2a537c3743963bfb6add44731dd3f3f6c785a2e.tar.gz | |
refactor: several improvements (#132)
- js export expression
- check args regarding run-agent/run-tool
Diffstat (limited to 'scripts/run-agent.py')
| -rwxr-xr-x | scripts/run-agent.py | 22 |
1 files changed, 14 insertions, 8 deletions
diff --git a/scripts/run-agent.py b/scripts/run-agent.py index 780267d..6b16b14 100755 --- a/scripts/run-agent.py +++ b/scripts/run-agent.py @@ -32,20 +32,26 @@ def parse_argv(this_file_name): agent_name = argv[0] agent_func = "" - agent_data = None + agent_data = "" if agent_name.endswith(this_file_name): - agent_name = sys.argv[1] - agent_func = sys.argv[2] - agent_data = sys.argv[3] + if len(sys.argv) > 3: + agent_name = sys.argv[1] + agent_func = sys.argv[2] + agent_data = sys.argv[3] else: - agent_name = os.path.basename(agent_name) - agent_func = sys.argv[1] - agent_data = sys.argv[2] + if len(sys.argv) > 2: + agent_name = os.path.basename(agent_name) + agent_func = sys.argv[1] + agent_data = sys.argv[2] - if agent_name.endswith(".py"): + if agent_name and agent_name.endswith(".py"): agent_name = agent_name[:-3] + if (not agent_data) or (not agent_func) or (not agent_name): + print("Usage: ./run-agent.py <agent-name> <agent-func> <agent-data>", file=sys.stderr) + sys.exit(1) + return agent_name, agent_func, agent_data |
