From d2a537c3743963bfb6add44731dd3f3f6c785a2e Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sat, 30 Nov 2024 17:14:21 +0800
Subject: refactor: several improvements (#132)

- js export expression
- check args regarding run-agent/run-tool
---
 scripts/run-agent.py | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

(limited to 'scripts/run-agent.py')

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
 
 
-- 
cgit v1.2.3