Add `setup_operator_args` in ExecutionConfig by pankajastro · Pull Request #2133 · astronomer/astronomer-cosmos

pankajastro · 2025-11-14T21:17:36Z

closes: #1972
closes: #1962

This PR adds the setup_operator_args parameter to ExecutionConfig.
It allows overriding the operator_args used for the producer task, enabling different argument values for the watcher and producer tasks.

netlify · 2025-11-14T21:17:42Z

✅ Deploy Preview for sunny-pastelito-5ecb04 ready!

Name	Link
🔨 Latest commit	`0790cff`
🔍 Latest deploy log	https://app.netlify.com/projects/sunny-pastelito-5ecb04/deploys/691c282d3e09b10008589fa8
😎 Deploy Preview	https://deploy-preview-2133--sunny-pastelito-5ecb04.netlify.app
📱 Preview on mobile	Toggle QR Code... Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify project configuration.

codecov · 2025-11-14T21:44:43Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 97.82%. Comparing base (1631bc3) to head (0790cff).
⚠️ Report is 1 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #2133   +/-   ##
=======================================
  Coverage   97.82%   97.82%           
=======================================
  Files          92       92           
  Lines        5970     5972    +2     
=======================================
+ Hits         5840     5842    +2     
  Misses        130      130

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

Copilot

Pull Request Overview

This PR adds the setup_operator_args parameter to ExecutionConfig, enabling users to override operator arguments specifically for the producer task in watcher execution mode. This allows different argument values between the watcher (consumer) tasks and the producer task.

Adds setup_operator_args field to ExecutionConfig dataclass
Passes execution_config to build_airflow_graph function
Applies setup_operator_args override when creating producer task in watcher mode

Reviewed Changes

Copilot reviewed 3 out of 3 changed files in this pull request and generated no comments.

File	Description
cosmos/config.py	Adds `setup_operator_args` field and documentation to `ExecutionConfig`
cosmos/airflow/graph.py	Imports `ExecutionConfig`, adds parameter to `build_airflow_graph`, and merges `setup_operator_args` with `task_args` for producer task
cosmos/converter.py	Passes `execution_config` to `build_airflow_graph` function call

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

tatiana

This looks great, @pankajastro ! Please, could you add tests and docs? Happy for the the docs to be in a follow-up PR if you prefer.

pankajastro · 2025-11-18T08:40:04Z

This looks great, @pankajastro ! Please, could you add tests and docs? Happy for the the docs to be in a follow-up PR if you prefer.

Added a test and created a docs PR: #2136

tatiana

This looks great, thank you very much, @pankajastro!

document the feature implemented in PR: #2133

@pankajastro

Breaking changes * Introduced in the PR #2080. The following functions are expected to be used internally only to Cosmos, so we hope these won't impact end-users, but we are documenting the changes just in case: - ``generate_task_or_group`` receives ``render_config`` instead of its individual configurations, such as ``test_behavior``, ``source_rendering_behavior`` and ``enable_owner_inheritance`` - ``create_task_metadata`` receives ``render_config`` instead of its individual configurations, such as ``test_behavior``, ``source_rendering_behavior`` and ``enable_owner_inheritance`` - ``create_task_metadata`` now expects the ``node_converters`` argument * Drop Python 3.9 support by @pankajastro in #2118 * Drop Airflow 2.4 support by @pankajastro in #2161 * Drop Airflow 2.5 support by @pankajastro in #2165 Features * Support applying ``node_converter`` at a task level instead of task group level by @anyapriya in #1759 * Allow overriding ``DbtProducerWatcherOperator`` parameters via ``ExecutionConfig.setup_operator_args`` by @pankajastro in #2133 * Use deferrable sensors by default in ``ExecutionMode.WATCHER`` by @pankajastro in #2084 * Support real-time consumer updates when using ``ExecutionMode.WATCHER`` and ``InvocationMode.SUBPROCESS`` by @pankajastro in #2152 * Update telemetry to v3 format with query parameters by @pankajkoti in #2192 * Add initial set of telemetry task listener metrics for Cosmos operators by @pankajkoti in #2195 Enhancements * Unify Airflow version handling into ``constants.py`` by @tatiana in #2089 * Refactor ``airflow/graph.py`` to simplify the code base by @tatiana in #2080 * Force watcher producer retries to zero by @pankajkoti in #2114 * Fail ``ExecutionMode.WATCHER`` consumer sensors immediately when the producer fails using Airflow context by @pankajkoti in #2126 * ``ExecutonMode.WATCHER``: fetch producer status asynchronously from the Airflow runtime so deferrable sensors fail immediately when the producer task fails by @pankajkoti in #2144 * Refactor ``ExecutionMode.WATCHER`` ``InvocationMode.SUBPROCESS`` log parser by @tatiana in #2183 * Replace map_index with is_mapped_task boolean in task telemetry metrics by @pankajkoti in #2210 * Collect cosmos profile metrics in task telemetry metrics by @pankajastro in #2198 * Remove unnecessary information from telemetry by @tatiana in #2211 Bug fixes * Clarify ``ExecutionMode.WATCHER`` deferrable failure messaging by @pankajkoti in #2124 * Remove empty test tasks when all tests are detached by @anyapriya in #2010 * Fix forwarding ``DbtProducerWatcherOperator`` ``dbt build`` flags by @michal-mrazek in #2127 * Add databricks oauth mock profile by @fjmacagno in #2164 * Register listeners in Airflow 3 plugin implementation by @pankajastro in #2187 * Fix resolution of ``packages-install-path`` when it uses ``env_var`` by @tatiana in #2194 * Fix ``template_fields`` in ``DbtConsumerWatcherSensor`` to include ``DbtRunLocalOperator`` template_fields`` by @tiovader and @emanuel-luis in #2209 * Emit asset events in ExecutionMode.AIRFLOW_ASYNC mode by @pankajastro in #2184 * Remove dag_run_id from telemetry tests by @tatiana in #2213 Docs * Document dataset-event limitation when using ``ExecutionMode.AIRFLOW_ASYNC`` by @varaprasadregani in #2143 * Expand ``ExecutionMode.KUBERNETES`` guidance by @tatiana in #2139 * Add docs for deferrable ``DbtConsumerWatcherSensor`` by @pankajastro in #2115 * Fix reStructuredText formatting by @dnskr in #2132 * Add docs for ``setup_operator_args`` param by @pankajastro in #2136 * Remove experimental flag for ``ExecutionMode.AIRFLOW_ASYNC`` by @pankajastro in #2153 * Clarify ``ExecutionMode.AIRFLOW_ASYNC`` dataset limits by @pankajkoti in #2167 * Update PRIVACY_NOTICE.rst by @tatiana in #2212 Others * Drop Python 3.9 support by @pankajastro in #2118 * Drop Airflow 2.4 support by @pankajastro in #2161 * Drop Airflow 2.5 support by @pankajastro in #2165 * Improve example DAG ``jaffle_shop_kubernetes.py`` by @tatiana in #2140 * Enable tests for Python 3.13 by @pankajastro in #2154 * Add Python 3.12 to CI integration tests matrix by @pankajastro in #2168 * Retry flaky Telemetry success test to stabilise CI by @pankajkoti in #2138 * Drop unused producer state xcom handling in ``ExecutionMode.WATCHER`` by @pankajkoti in #2145 * Remove unused Python3.9 uses from Github action CI by @pankajastro in #2117 * Run pre-commit on ``ExecutionMode.WATCHER`` modules by @pankajkoti in #2150 * Refactor: Use shared airflow version constant by @pankajkoti in #2157 * Pin ``pydantic<2.0`` for Airflow 2.6 compatibility by @pankajastro in #2172 * Remove duplicate ``dbt-duckdb`` dependency by @pankajastro in #2170 * Add targeted ``type: ignore`` for untyped decorators to fix ``mypy`` errors by @pankajastro in #2174 * Replace Legacy typing Aliases with Built-in Types for Python 3.10+ by @pankajastro in #2175 * Refactor to reuse ``load_method_from_module`` from ``_utils/importer.py`` by @pankajastro in #2176 * Remove try except block for cache import and unused python_version variable by @pankajastro in #2186 * Unpin Airflow to satisfy GitHub Security tab requirements by @pankajastro in #2171 * Update Python version for ``pyupgrade`` in ``pre-commit`` config by @pankajastro in #2190 * Add cooldown config in ``dependabot`` config by @pankajastro in #2189 * Adjust pre-commit so Python 3.10 or higher can be used by @tatiana in #2196 * Remove empty variables emission from telemetry metrics by @pankajkoti in #2197 * Reformat documented comments for historical URL formats by @pankajkoti in #2199 * Bump ``actions/checkout`` from ``5.0.0`` to ``5.0.1`` by @dependabot in #2135 * Bump ``actions/checkout`` to ``6.0.0`` in GitHub workflows by @dependabot in #2147 * Bump ``zizmorcore/zizmor-action`` from ``0.2.0`` to ``0.3.0`` by @dependabot in #2156 * Bump ``actions/checkout`` from ``5.0.1`` to ``6.0.0`` by @dependabot in #2155 * Bump ``actions/checkout`` from ``6.0.0`` to ``6.0.1`` by @dependabot in #2178 * Bump ``codecov/codecov-action`` from ``5.5.1`` to ``5.5.2`` by @dependabot in #2208 * pre-commit autoupdate by @pre-commit-ci[bot] in #2134, #2162, #2173, #2191, #2202 closes: astronomer/oss-integrations-private#275

pankajastro temporarily deployed to internal November 14, 2025 21:17 — with GitHub Actions Inactive

pankajastro temporarily deployed to internal November 17, 2025 07:47 — with GitHub Actions Inactive

pankajastro marked this pull request as ready for review November 17, 2025 08:37

Copilot AI review requested due to automatic review settings November 17, 2025 08:37

Copilot started reviewing on behalf of pankajastro November 17, 2025 08:38 View session

Copilot finished reviewing on behalf of pankajastro November 17, 2025 08:40

Copilot AI reviewed Nov 17, 2025

View reviewed changes

tatiana changed the title ~~Add setup_operator_args in ExecutionConfig~~ Add setup_operator_args in ExecutionConfig Nov 17, 2025

tatiana reviewed Nov 17, 2025

View reviewed changes

pankajastro temporarily deployed to internal November 17, 2025 20:12 — with GitHub Actions Inactive

pankajastro temporarily deployed to internal November 17, 2025 20:13 — with GitHub Actions Inactive

pankajastro temporarily deployed to internal November 17, 2025 20:25 — with GitHub Actions Inactive

pankajastro temporarily deployed to internal November 17, 2025 20:38 — with GitHub Actions Inactive

pankajastro temporarily deployed to internal November 17, 2025 20:41 — with GitHub Actions Inactive

pankajastro temporarily deployed to internal November 17, 2025 20:44 — with GitHub Actions Inactive

pankajastro temporarily deployed to internal November 17, 2025 20:47 — with GitHub Actions Inactive

pankajastro temporarily deployed to internal November 17, 2025 20:57 — with GitHub Actions Inactive

pankajastro temporarily deployed to internal November 17, 2025 21:06 — with GitHub Actions Inactive

pankajastro temporarily deployed to internal November 18, 2025 08:01 — with GitHub Actions Inactive

pankajastro added 9 commits November 18, 2025 13:32

Add setup_operator_args in ExecutionConfig

2b8d722

cleanup

3f7e97b

Add tests

e0871ad

Add tests

fca7bd0

Fix dependency

a464d3f

Fix dependency

04fc544

Fix dependency

65b304c

Fix dependency

71e5f4a

Fix dependency

b3b741b

pankajastro added 3 commits November 18, 2025 13:32

Fix dependency

68ff9d9

Fix dependency

ac0b1e2

Fix dependency

0790cff

pankajastro force-pushed the fix_1962 branch from 9e7736b to 0790cff Compare November 18, 2025 08:02

pankajastro temporarily deployed to internal November 18, 2025 08:03 — with GitHub Actions Inactive

pankajastro mentioned this pull request Nov 18, 2025

Add docs for setup_operator_args param #2136

Merged

tatiana approved these changes Nov 18, 2025

View reviewed changes

tatiana merged commit 0112d06 into main Nov 18, 2025
85 checks passed

tatiana deleted the fix_1962 branch November 18, 2025 11:07

This was referenced Nov 21, 2025

Verify watcher sensor retries honor setup_operator_args-style queue overrides (cluster policy) #2148

Closed

Release 1.12.0 #2149

Merged

pankajastro added a commit that referenced this pull request Nov 22, 2025

Add docs for setup_operator_args param (#2136)

db76233

document the feature implemented in PR: #2133

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Add `setup_operator_args` in ExecutionConfig#2133

Add `setup_operator_args` in ExecutionConfig#2133
tatiana merged 12 commits into
mainfrom
fix_1962

pankajastro commented Nov 14, 2025 •

edited

Loading

Uh oh!

netlify Bot commented Nov 14, 2025 •

edited

Loading

Uh oh!

codecov Bot commented Nov 14, 2025 •

edited

Loading

Uh oh!

Copilot AI left a comment

Uh oh!

tatiana left a comment

Uh oh!

pankajastro commented Nov 18, 2025

Uh oh!

tatiana left a comment

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

Conversation

pankajastro commented Nov 14, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

netlify Bot commented Nov 14, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

✅ Deploy Preview for sunny-pastelito-5ecb04 ready!

Uh oh!

codecov Bot commented Nov 14, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Codecov Report

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull Request Overview

Reviewed Changes

Uh oh!

tatiana left a comment

Choose a reason for hiding this comment

Uh oh!

pankajastro commented Nov 18, 2025

Uh oh!

tatiana left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

pankajastro commented Nov 14, 2025 •

edited

Loading

netlify Bot commented Nov 14, 2025 •

edited

Loading

codecov Bot commented Nov 14, 2025 •

edited

Loading