From 739a832d87c00e3b5977a24bba5654fa5ea7a702 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 7 Jun 2024 15:16:31 +0800
Subject: feat: js/py generate declarations from comments (#30)

---
 scripts/run-tool.js | 31 +++++++++++++------------------
 1 file changed, 13 insertions(+), 18 deletions(-)

(limited to 'scripts/run-tool.js')

diff --git a/scripts/run-tool.js b/scripts/run-tool.js
index 43a6587..f99b2e0 100755
--- a/scripts/run-tool.js
+++ b/scripts/run-tool.js
@@ -58,23 +58,18 @@ const [funcName, funcData] = parseArgv();
 
 process.env["LLM_FUNCTION_NAME"] = funcName;
 
-if (process.env["LLM_FUNCTION_ACTION"] == "declarate") {
-  const { declarate } = loadFunc(funcName);
-  console.log(JSON.stringify(declarate(), null, 2));
-} else {
-  if (!funcData) {
-    console.log("No json data");
-    process.exit(1);
-  }
-
-  let args;
-  try {
-    args = JSON.parse(funcData);
-  } catch {
-    console.log("Invalid json data");
-    process.exit(1);
-  }
+if (!funcData) {
+  console.log("No json data");
+  process.exit(1);
+}
 
-  const { execute } = loadFunc(funcName);
-  execute(args);
+let args;
+try {
+  args = JSON.parse(funcData);
+} catch {
+  console.log("Invalid json data");
+  process.exit(1);
 }
+
+const { main } = loadFunc(funcName);
+main(args);
-- 
cgit v1.2.3