jason810496 opened a new pull request, #68554:
URL: https://github.com/apache/airflow/pull/68554
- related: #67937 (original SDK fix, cherry-picked here)
- closes: #67797
## Why
In coordinator mode a failing Go task with retries left was marked terminal
`FAILED` because the Go SDK never sent a `RetryTask` frame back to the
supervisor, so configured retries were silently lost.
## What
- `go-sdk/pkg/execution`: add `ShouldRetry` / `MaxTries` to `TIRunContext`,
add `RetryTaskMsg`, and emit it on task failure/panic when `should_retry` is set
- `go-sdk` example bundle: `load` task now reads `try_number`, fails on its
first attempt and succeeds on the retry (`retries=1`).
- e2e (`go_sdk_tests`): assert `load` transitions through UP_FOR_RETRY and
ends `success` on `try_number` 2.
## Verification
All 9 `go_sdk` e2e tests pass:
```
E2E_TEST_MODE=go_sdk uv run --project airflow-e2e-tests pytest \
airflow-e2e-tests/tests/airflow_e2e_tests/go_sdk_tests/test_go_sdk_dag.py -xvs
```
---
##### Was generative AI tooling used to co-author this PR?
- [x] Yes, with help of Claude Code Opus 4.8 following [the
guidelines](https://github.com/apache/airflow/blob/main/contributing-docs/05_pull_requests.rst#gen-ai-assisted-contributions)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
