Add DBT project to run benchmark for async execution mode#11
Merged
Conversation
Contributor
There was a problem hiding this comment.
Pull Request Overview
Adds a focused dbt project and Airflow DAG to benchmark async execution performance using Cosmos and BigQuery, with seeds and “slow” models to inflate runtime and a script to scale model count.
- Introduces altered_jaffle_shop dbt project with seeds, staging, core, and heavy benchmarking models.
- Adds an Airflow DbtDag configured for AIRFLOW_ASYNC execution and a script to auto-generate additional long-running models.
- Provides configurable model parameters via a seed to control workload size.
Reviewed Changes
Copilot reviewed 24 out of 29 changed files in this pull request and generated 9 comments.
Show a summary per file
| File | Description |
|---|---|
| dbt/altered_jaffle_shop/seeds/raw_payments.csv | Seed data for payments used by staging model. |
| dbt/altered_jaffle_shop/seeds/raw_orders.csv | Seed data for orders used by staging model. |
| dbt/altered_jaffle_shop/seeds/raw_customers.csv | Seed data for customers used by staging model. |
| dbt/altered_jaffle_shop/seeds/model_params.csv | Seed controlling workload parameters for long models. |
| dbt/altered_jaffle_shop/profiles.yml | Example dbt profile for BigQuery (likely unused with Cosmos mapping). |
| dbt/altered_jaffle_shop/models/staging/stg_payments.sql | Staging model converting cents to dollars. |
| dbt/altered_jaffle_shop/models/staging/stg_orders.sql | Staging model for orders. |
| dbt/altered_jaffle_shop/models/staging/stg_customers.sql | Staging model for customers. |
| dbt/altered_jaffle_shop/models/schema.yml | Schema tests and docs for orders and customers. |
| dbt/altered_jaffle_shop/models/orders.sql | Orders fact model aggregating payments by method. |
| dbt/altered_jaffle_shop/models/long_model_text_processing.sql | Heavy text-processing benchmark model. |
| dbt/altered_jaffle_shop/models/long_model_subquery_windows.sql | Heavy windowing benchmark model. |
| dbt/altered_jaffle_shop/models/long_model_cross_random.sql | Heavy cross-join benchmark model. |
| dbt/altered_jaffle_shop/models/docs.md | Docs for order status values. |
| dbt/altered_jaffle_shop/models/customers_slow_query.sql | Heavy windowing workload over customers. |
| dbt/altered_jaffle_shop/models/customers.sql | Customers mart model. |
| dbt/altered_jaffle_shop/macros/get_model_param.sql | Macro intended to read params from a seed. |
| dbt/altered_jaffle_shop/dbt_project.yml | Project config (profile name, paths, defaults). |
| dbt/altered_jaffle_shop/README.md | Project usage notes. |
| dbt/altered_jaffle_shop/.user.yml | User metadata. |
| dbt/altered_jaffle_shop/.gitignore | Ignore dbt artifacts. |
| dags/cosmos_async_dag.py | Airflow DAG using Cosmos with AIRFLOW_ASYNC. |
| benchmark/auto_generate_models.sh | Script to replicate long models for scaling. |
| README.md | Repository-level readme mentions new project and how to scale models. |
Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.
pankajkoti
reviewed
Apr 27, 2026
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
pankajkoti
reviewed
Apr 30, 2026
Collaborator
pankajkoti
left a comment
There was a problem hiding this comment.
LGTM, but have some comments inline if you could check and address once before merging.
Hoping that we have tested the benchmark with these scripts.
pankajkoti
approved these changes
Apr 30, 2026
Collaborator
|
Happy to merge once we have tested benchmarking with the latest changes and the comments are addressed. |
Co-authored-by: Pankaj Koti <pankajkoti699@gmail.com>
Co-authored-by: Pankaj Koti <pankajkoti699@gmail.com>
Co-authored-by: Pankaj Koti <pankajkoti699@gmail.com>
Contributor
Author
pankajastro
added a commit
that referenced
this pull request
May 8, 2026
<img width="1703" height="1008" alt="Screenshot 2026-04-28 at 4 05 36 PM" src="https://github.com/user-attachments/assets/ee2785fc-34e2-4cee-b092-33daf8480fec" /> <img width="1680" height="758" alt="Screenshot 2026-05-05 at 1 24 00 PM" src="https://github.com/user-attachments/assets/e8e6b3df-9e38-46e7-9278-79485b986d08" /> Depend on - astronomer/astronomer-cosmos#2616 - #11 closes: astronomer/oss-integrations-private#176 --------- Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
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
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 introduces a DBT project setup to benchmark async execution mode performance.
benchmark/auto_generate_models.shis available to generate additional models if needed for scaling benchmarks.dbt/altered_jaffle_shop/seeds/model_params.csv, which allows simulation of time-consuming transformation.This setup will help in evaluating how async execution mode scales with larger DAGs and more resource-intensive transformations.
closes: https://github.com/astronomer/oss-integrations-private/issues/175