24 lines
549 B
Bash
Executable File
24 lines
549 B
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
# @describe Execute sql code.
|
|
# @option --code! The code to execute.
|
|
|
|
# @meta require-tools usql
|
|
|
|
# @env USQL_DSN! The database connection url. e.g. pgsql://user:pass@host:port
|
|
# @env LLM_OUTPUT=/dev/stdout The output path
|
|
|
|
# shellcheck disable=SC1090
|
|
source "$LLM_PROMPT_UTILS_FILE"
|
|
|
|
# shellcheck disable=SC2154
|
|
main() {
|
|
argc_code="$(jq -r '.code' <<< "$LLM_TOOL_RAW_JSON")"
|
|
|
|
if ! grep -qi '^select' <<<"$argc_code"; then
|
|
guard_operation ""
|
|
fi
|
|
usql -c "$argc_code" "$USQL_DSN" >> "$LLM_OUTPUT"
|
|
}
|