Refactor airflow/graph.py to simplify code-base

[tatiana]

The goal with this PR is to refactor the airflow/graph.py to make the code more manageable.

The PR #1759 should be merged before the current changes are applied.

The complexity of building DAGs has increased significantly with #1759. As part of this PR, we ended up duplicating configurations and argument passing in multiple parts of the code, adding more complexity to the airflow/graph.py module.

This PR also aims to add documentation changes that will help users of node_converters and node_conversion_by_task_group.

Breaking changes

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

[Copilot]

Pull Request Overview

This PR refactors the airflow/graph.py module to reduce code complexity and duplication introduced in PR #1759. The refactoring consolidates multiple configuration parameters into a RenderConfig object and adds support for finer-grained node conversion control.

Key changes:

• Consolidates individual configuration parameters (test_behavior, source_rendering_behavior, enable_owner_inheritance, etc.) into a single render_config parameter
• Introduces node_conversion_by_task_group flag to control whether node converters apply at task group or individual task level
• Adds new generate_or_convert_task function to centralize task creation/conversion logic

Reviewed Changes

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

Show a summary per file

File	Description
cosmos/init.py	Version bump to 1.12.0a2
CHANGELOG.rst	Documents breaking changes and new features for v1.12.0a1
cosmos/config.py	Adds node_conversion_by_task_group parameter to RenderConfig
docs/configuration/render-config.rst	Documents the new node_conversion_by_task_group parameter and updates node converter documentation
cosmos/airflow/graph.py	Major refactoring to consolidate parameters into render_config and add generate_or_convert_task function
tests/airflow/test_graph.py	Updates tests to use RenderConfig wrapper and adds comprehensive node converter tests
tests/dbt/test_pruning.py	Updates tests to pass render_config instead of individual parameters

Comments suppressed due to low confidence (1)

tests/airflow/test_graph.py:1

• The test_behavior parameter is passed both as a standalone argument (line 367) and inside the RenderConfig (implicitly using the default). This is inconsistent with the refactoring goal. The standalone test_behavior parameter should be removed and only set within the RenderConfig.

import os

---

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