1 files changed, 3 insertions, 7 deletions
diff --git a/tools/execute_sql_code.sh b/tools/execute_sql_code.sh
index df57dea..34cd948 100755
--- a/tools/execute_sql_code.sh
+++ b/tools/execute_sql_code.sh
@@ -9,15 +9,11 @@ set -e
 # @env USQL_DSN! The database url, e.g. pgsql://user:pass@host/dbname
 # @env LLM_OUTPUT=/dev/stdout The output path
 
+ROOT_DIR="${LLM_ROOT_DIR:-$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)}"
+
 main() {
     if ! grep -qi '^select' <<<"$argc_code"; then
-        if [ -t 1 ]; then
-            read -r -p "Are you sure you want to continue? [Y/n] " ans
-            if [[ "$ans" == "N" || "$ans" == "n" ]]; then
-                echo "Aborted!"
-                exit 1
-            fi
-        fi
+        "$ROOT_DIR/utils/guard_operation.sh"
     fi
     usql -c "$argc_code" "$USQL_DSN" >> "$LLM_OUTPUT"
 }