databricks-sql-autotuner
Fail
Audited by Gen Agent Trust Hub on Oct 6, 2026
Risk Level: HIGHDYNAMIC_EXECUTIONEXTERNAL_DOWNLOADSCOMMAND_EXECUTIONINDIRECT_PROMPT_INJECTION
Full Analysis
- [DYNAMIC_EXECUTION]: The script
scripts/tune.pydynamically loads and executes a Python file from the user's workspace. - Evidence: The
load_session_udfsfunction inscripts/tune.pychecks for a file namedudf_setup.pyin the current directory and executes itsregister_udfsfunction usingimportlib. While intended for registering Spark UDFs, this allows for the execution of arbitrary code if a malicious file is placed in the directory where the skill is invoked. - [EXTERNAL_DOWNLOADS]: The skill downloads and executes an installation script for the
uvpackage manager from its official source. - Evidence: Both
SKILL.mdandscripts/env_setup.pyreference the commandcurl -LsSf https://astral.sh/uv/install.sh | sh. The domainastral.shbelongs to Astral, a well-known technology company in the Python ecosystem. - [COMMAND_EXECUTION]: The skill uses
subprocess.runto interact with system tools and local scripts. - Evidence:
scripts/discover.pyexecutesdatabricksCLI commands.scripts/init_run.pyusesgitfor branch and file management.scripts/env_setup.pymanages virtual environments viauv. These are necessary for the skill's functionality but involve executing shell commands based on internal logic. - [INDIRECT_PROMPT_INJECTION]: The skill has an attack surface for indirect prompt injection via user-supplied SQL queries.
- Ingestion points: User-provided SQL strings or file contents via the
/databricks-sql-autotunercommand (processed inSKILL.mdandscripts/tune.py). - Boundary markers: No specific delimiters or safety instructions are used to wrap the input SQL.
- Capability inventory: The skill can execute shell commands, perform Git operations, and run SQL queries on a Databricks cluster.
- Sanitization: No sanitization or validation of the SQL content is performed before execution.
Recommendations
- HIGH: Downloads and executes remote code from: https://astral.sh/uv/install.sh - DO NOT USE without thorough review
Audit Metadata