From 39447fb39e93b054fb2bb839fb774d1b6af080e7 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 6 Jun 2024 16:33:24 +0800
Subject: feat: set environment variable LLM_FUNCTION_NAME (#27)

---
 scripts/bin.sh | 29 +++++++++++++++--------------
 1 file changed, 15 insertions(+), 14 deletions(-)

(limited to 'scripts/bin.sh')

diff --git a/scripts/bin.sh b/scripts/bin.sh
index b5b1075..807343a 100755
--- a/scripts/bin.sh
+++ b/scripts/bin.sh
@@ -8,26 +8,27 @@ if [[ -f "$LLM_FUNCTIONS_DIR/.env" ]]; then
 fi
 
 if [[ "$0" == *bin.sh ]]; then
-    FUNC_FILE="$1"
-    FUNC_DATA="$2"
+    func_name="$1"
+    func_data="$2"
 else
-    FUNC_FILE="$(basename "$0")"
-    FUNC_DATA="$1"
+    func_name="$(basename "$0")"
+    func_data="$1"
 fi
-if [[ "$FUNC_FILE" != *'.sh' ]]; then
-    FUNC_FILE="$FUNC_FILE.sh"
+if [[ "$func_name" == *.sh ]]; then
+    func_name="${func_name:0:$((${#func_name}-3))}"
 fi
 
-FUNC_FILE="$LLM_FUNCTIONS_DIR/tools/$FUNC_FILE"
+export LLM_FUNCTION_NAME="$func_name"
+func_file="$LLM_FUNCTIONS_DIR/tools/$func_name.sh"
 
 export JQ=jq
 if [[ "$OS" == "Windows_NT" ]]; then
     export JQ="jq -b"
-    FUNC_FILE="$(cygpath -w "$FUNC_FILE")"
+    func_file="$(cygpath -w "$func_file")"
 fi
 
 if [[ "$LLM_FUNCTION_ACTION" == "declarate" ]]; then
-    argc --argc-export "$FUNC_FILE" | \
+    argc --argc-export "$func_file" | \
     $JQ -r '
     def parse_description(flag_option):
         if flag_option.describe == "" then
@@ -69,13 +70,13 @@ if [[ "$LLM_FUNCTION_ACTION" == "declarate" ]]; then
         parameters: parse_parameter([.flag_options[] | select(.id != "help" and .id != "version")])
     }'
 else
-    if [[ -z "$FUNC_DATA" ]]; then
+    if [[ -z "$func_data" ]]; then
         echo "No json data"
         exit 1
     fi
 
     data="$(
-        echo "$FUNC_DATA" | \
+        echo "$func_data" | \
         $JQ -r '
         to_entries | .[] | 
         (.key | split("_") | join("-")) as $key |
@@ -92,10 +93,10 @@ else
     }
     while IFS= read -r line; do
         if [[ "$line" == '--'* ]]; then
-            ARGS+=("$line")
+            args+=("$line")
         else
-            ARGS+=("$(echo "$line" | $JQ -r '.')")
+            args+=("$(echo "$line" | $JQ -r '.')")
         fi
     done <<< "$data"
-    "$FUNC_FILE" "${ARGS[@]}"
+    "$func_file" "${args[@]}"
 fi
\ No newline at end of file
-- 
cgit v1.2.3