-
Notifications
You must be signed in to change notification settings - Fork 16.3k
Add options_mapping support to HITLBranchOperator #55093
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Merged
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
kaxil
reviewed
Aug 30, 2025
providers/standard/src/airflow/providers/standard/operators/hitl.py
Outdated
Show resolved
Hide resolved
4ff2c9f to
c2cd7a7
Compare
Lee-W
approved these changes
Sep 1, 2025
This change introduces an optional `options_mapping` argument for HITLBranchOperator, allowing human-facing option labels to be mapped to specific downstream task_ids. This makes the operator more business-friendly (e.g. "Approve" → "publish") while preserving Airflow’s branch validation. Also adds unit tests covering: - valid mapping for single and multiple choices - fallback to unmapped options - error when mapping points to non-downstream task_ids - invalid mapping keys - invalid mapping value types
407a71e to
a0bc195
Compare
nothingmin
pushed a commit
to nothingmin/airflow
that referenced
this pull request
Sep 2, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Sep 30, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 1, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 2, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 3, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 4, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 5, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 5, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 7, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 8, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 9, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 10, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 11, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 12, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 14, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 15, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 17, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
abdulrahman305 bot
pushed a commit
to abdulrahman305/airflow
that referenced
this pull request
Oct 19, 2025
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR enhances
HITLBranchOperatorto accept an optionaloptions_mappingdict that maps human-facing option labels to downstreamtask_ids. Previously, HITL branching required users to select rawtask_ids, which isn’t user-friendly in business-facing workflows. With this change, a workflow can present clearer options (e.g., “Full Deploy” / “Canary”) while still resolving to valid downstream tasks.Changes
options)task_idsof valid downstream tasksAirflowException)Rationale
This improves HITL usability for end-users by decoupling the labels people choose from the
task_idsAirflow needs that may not necessarily be known by the approver, while keeping strong validation in place.Example
If the user selects Approve, the operator branches to the
publishtask.