refactor: restructure repo layout

llm-functions/scripts/create-tool.sh

@@ -0,0 +1,195 @@
+#!/usr/bin/env bash
+set -e
+
+# @describe Create a boilplate tool script
+#
+# Examples:
+#   ./scripts/create-tool.sh _test.py foo bar! baz+ qux*
+#
+# @option --description <text> The tool description
+# @flag --force Override the exist tool file
+# @arg name! The script file name
+# @arg params* The script parameters
+
+main() {
+    output="tools/$argc_name"
+    if [[ -f "$output" ]] && [[ -z "$argc_force" ]]; then
+        _die "$output already exists"
+    fi
+    ext="${argc_name##*.}"
+    description="${argc_description:-"The description for the tool"}"
+    support_exts=('.sh' '.js' '.py')
+    if [[ "$ext" == "$argc_name" ]]; then
+        _die "error: no extension name, pelease add one of ${support_exts[*]}" 
+    fi
+    case $ext in
+    sh) create_sh ;;
+    js) create_js ;;
+    py) create_py ;;
+    *) _die "error: invalid extension name: $ext, must be one of ${support_exts[*]}" ;; 
+    esac
+    echo "$output generated"
+}
+
+create_sh() {
+    cat <<-'EOF' > "$output"
+#!/usr/bin/env bash
+set -e
+
+EOF
+    echo "# @describe $description" >> "$output"
+    for param in "${argc_params[@]}"; do
+        echo "# @option --$(echo $param | sed 's/-/_/g')" >> "$output"
+    done
+    cat <<-'EOF' >> "$output"
+
+main() {
+    ( set -o posix ; set ) | grep ^argc_
+}
+
+eval "$(argc --argc-eval "$0" "$@")"
+EOF
+    chmod +x "$output"
+}
+
+create_js() {
+    properties=''
+    for param in "${argc_params[@]}"; do
+        if [[ "$param" == *'!' ]]; then
+            param="${param:0:$((${#param}-1))}"
+            property=" * @property {string} $param - "
+        elif [[ "$param" == *'+' ]]; then
+            param="${param:0:$((${#param}-1))}"
+            property=" * @property {string[]} $param - "
+        elif [[ "$param" == *'*' ]]; then
+            param="${param:0:$((${#param}-1))}"
+            property=" * @property {string[]} [$param] - "
+        else
+            property=" * @property {string} [$param] - "
+        fi
+        properties+=$'\n'"$property"
+    done
+    cat <<EOF > "$output"
+/**
+ * ${description}
+ * @typedef {Object} Args${properties}
+ * @param {Args} args
+ */
+exports.run = function (args) {
+  console.log(args);
+}
+EOF
+}
+
+create_py() {
+    has_array_param=false
+    has_optional_pram=false
+    required_properties=''
+    optional_properties=''
+    required_arguments=()
+    optional_arguments=()
+    indent="    "
+    for param in "${argc_params[@]}"; do
+        optional=false
+        if [[ "$param" == *'!' ]]; then
+            param="${param:0:$((${#param}-1))}"
+            type="str"
+        elif [[ "$param" == *'+' ]]; then
+            param="${param:0:$((${#param}-1))}"
+            type="List[str]"
+            has_array_param=true
+        elif [[ "$param" == *'*' ]]; then
+            param="${param:0:$((${#param}-1))}"
+            type="Optional[List[str]] = None"
+            optional=true
+            has_array_param=true
+        else
+            optional=true
+            type="Optional[str] = None"
+        fi
+        if [[ "$optional" == "true" ]]; then
+            has_optional_pram=true
+            optional_arguments+="$param: $type, "
+            optional_properties+=$'\n'"$indent$indent$param: -"
+        else
+            required_arguments+="$param: $type, "
+            required_properties+=$'\n'"$indent$indent$param: -"
+        fi
+    done
+    import_typing_members=()
+    if [[ "$has_array_param" == "true" ]]; then
+        import_typing_members+=("List")
+    fi
+    if [[ "$has_optional_pram" == "true" ]]; then
+        import_typing_members+=("Optional")
+    fi
+    imports=""
+    if [[ -n "$import_typing_members" ]]; then
+        members="$(echo "${import_typing_members[*]}" | sed 's/ /, /')"
+        imports="from typing import $members"$'\n'
+    fi
+    if [[ -n "$imports" ]]; then
+        imports="$imports"$'\n'
+    fi
+    cat <<EOF > "$output"
+${imports}
+def run(${required_arguments}${optional_arguments}):
+    """${description}
+    Args:${required_properties}${optional_properties}
+    """
+    pass
+EOF
+}
+
+build_schema() {
+    echo '{
+        "name": "'"${argc_name%%.*}"'",
+        "description": "",
+        "parameters": '"$(build_properties)"'
+    }' | jq '.' | sed '2,$s/^/  /g'
+}
+
+build_properties() {
+    required_params=()
+    properties=''
+    for param in "${argc_params[@]}"; do
+        if [[ "$param" == *'!' ]]; then
+            param="${param:0:$((${#param}-1))}"
+            required_params+=("$param")
+            property='{"'"$param"'":{"type":"string","description":""}}'
+        elif [[ "$param" == *'+' ]]; then
+            param="${param:0:$((${#param}-1))}"
+            required_params+=("$param")
+            property='{"'"$param"'":{"type":"array","description":"","items": {"type":"string"}}}'
+        elif [[ "$param" == *'*' ]]; then
+            param="${param:0:$((${#param}-1))}"
+            property='{"'"$param"'":{"type":"array","description":"","items": {"type":"string"}}}'
+        else
+            property='{"'"$param"'":{"type":"string","description":""}}'
+        fi
+        properties+="$property"
+    done
+    required=''
+    for param in "${required_params[@]}"; do
+        if [[ -z "$required" ]]; then
+            required=',"required":['
+        fi
+        required+="\"$param\","
+    done
+    if [[ -n "$required" ]]; then
+        required="${required:0:$((${#required}-1))}"
+        required+="]"
+    fi
+    echo '{
+        "type": "object",
+        "properties": '"$(echo "$properties" | jq -s 'add')$required"'
+    }' | jq '.'
+}
+
+_die() {
+    echo "$*" >&2
+    exit 1
+}
+
+# See more details at https://github.com/sigoden/argc
+eval "$(argc --argc-eval "$0" "$@")"