Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

refactor: demo_tool.* inspect LLM_* env vars #33

Merged
merged 1 commit into from
Jun 7, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
45 changes: 37 additions & 8 deletions Argcfile.sh
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,10 @@ LANG_CMDS=( \
"py:python" \
)

# @cmd Call the function
# @cmd Run the tool
# @arg cmd![`_choice_cmd`] The function command
# @arg json The json data
call() {
run-tool() {
if _is_win; then
ext=".cmd"
fi
Expand Down Expand Up @@ -128,16 +128,18 @@ list-tools() {

# @cmd Test the project
test() {
mkdir -p tmp/tests
names_file=tmp/tests/functions.txt
declarations_file=tmp/tests/functions.json
argc list-tools > "$names_file"
argc build --names-file "$names_file" --declarations-file "$declarations_file"
argc test-tools
test-tools
}

# @cmd Test call functions
test-tools() {
tmp_dir="cache/tmp"
mkdir -p "$tmp_dir"
names_file="$tmp_dir/functions.txt"
declarations_file="$tmp_dir/functions.json"
argc list-tools > "$names_file"
argc build --names-file "$names_file" --declarations-file "$declarations_file"

if _is_win; then
ext=".cmd"
fi
Expand All @@ -162,6 +164,33 @@ test-tools() {
done
}

# @cmd Test all demo tools
test-demo-tools() {
for item in "${LANG_CMDS[@]}"; do
lang="${item%:*}"
echo "---- Test demo_tool.$lang ---"
argc build-bin "demo_tool.$lang"
argc run-tool demo_tool '{
"boolean": true,
"string": "Hello",
"string_enum": "foo",
"integer": 123,
"number": 3.14,
"array": [
"a",
"b",
"c"
],
"string_optional": "OptionalValue",
"array_optional": [
"x",
"y"
]
}'
echo
done
}

# @cmd Install this repo to aichat functions_dir
install() {
functions_dir="$(aichat --info | grep functions_dir | awk '{print $2}')"
Expand Down
2 changes: 1 addition & 1 deletion scripts/create-tool.sh
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ EOF
cat <<-'EOF' >> "$output"

main() {
( set -o posix ; set ) | grep ^argc_ # inspect all argc variables
( set -o posix ; set ) | grep ^argc_
}

eval "$(argc --argc-eval "$0" "$@")"
Expand Down
10 changes: 9 additions & 1 deletion tools/demo_tool.js
Original file line number Diff line number Diff line change
Expand Up @@ -12,5 +12,13 @@
* @param {Args} args
*/
exports.run = function run(args) {
console.log(JSON.stringify(args, null, 2));
for (const [key, value] of Object.entries(args)) {
console.log(`${key}: ${JSON.stringify(value)}`);
}

for (const [key, value] of Object.entries(process.env)) {
if (key.startsWith("LLM_")) {
console.log(`${key}: ${value}`);
}
}
}
16 changes: 7 additions & 9 deletions tools/demo_tool.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import os
from typing import List, Literal, Optional


def run(
boolean: bool,
string: str,
Expand All @@ -22,11 +22,9 @@ def run(
string_optional: Define a optional string property
array_optional: Define a optional string array property
"""
print(f"boolean: {boolean}")
print(f"string: {string}")
print(f"string_enum: {string_enum}")
print(f"integer: {integer}")
print(f"number: {number}")
print(f"array: {array}")
print(f"string_optional: {string_optional}")
print(f"array_optional: {array_optional}")
for key, value in locals().items():
print(f"{key}: {value}")

for key, value in os.environ.items():
if key.startswith("LLM_"):
print(f"{key}: {value}")
3 changes: 2 additions & 1 deletion tools/demo_tool.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,8 @@
# @option --array-optional* Define a optional string array property

main() {
( set -o posix ; set ) | grep ^argc_ # inspect all argc variables
( set -o posix ; set ) | grep ^argc_
printenv | grep '^LLM_'
}

eval "$(argc --argc-eval "$0" "$@")"