The GitHub Actions job "Tests (AMD)" on 
airflow.git/fix-datafusion-sql-error-crashes-task has succeeded.
Run started by GitHub user ColtenOuO (triggered by ColtenOuO).

Head commit for run:
af0bb6aea3730a152d84709f2733e24d45edee88 / ColtenOuO <[email protected]>
Let the agent retry on a rejected DataFusion query instead of failing the task

DataFusionToolset._query() re-raised SQLSafetyError as-is after logging it,
so any query that failed safety validation crashed the whole task instead
of giving the agent a chance to correct it. That validation isn't limited
to deliberate write attempts (CREATE/DROP/etc.) -- sqlglot parses in RAISE
mode, so a plain SQL syntax typo the agent generates also fails as a
SQLSafetyError and hit the same crash.

SQLToolset already treats every query error this way (wrapped in
ModelRetry, bounded by the tool's max_retries=1) so the model sees the
error and can fix its own SQL within the run. Bring DataFusionToolset's
SQLSafetyError handling in line with it.

Report URL: https://github.com/apache/airflow/actions/runs/31509319771

With regards,
GitHub Actions via GitBox


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to