astronomer/astronomer-cosmos

DBT Docs generate allow for execution config

Closed this issue · 1 comments

Description

As part of the dbtdocsoperator there isn't an option to specify the execution config unlike with the task group or DAG. This means that if dbt is install on venv then it is not able to work.

Also the connection_id is mandatory but I am able to connect to S3 for other tasks without specifying the connection id so can this be made optional? (Seems to fail on the pydantic validation)

Use case/motivation

Allows for more use of the dbt docs generator.

Related issues

No response

Are you willing to submit a PR?

  • Yes, I am willing to submit a PR!

Was able to get it working now. You can still use the dbt_executable_path and just use a dummy value for the connection_id.

generate_dbt_docs_aws = DbtDocsS3Operator(
task_id="generate_dbt_docs_aws",
project_dir=f"{os.environ['AIRFLOW_HOME']}/dags/shared/dbt",
profile_config=profile_config,
dbt_executable_path=f"{os.environ['AIRFLOW_HOME']}/dbt_venv/bin/dbt",
# docs-specific arguments
connection_id="dummy",
bucket_name=f"bucket",
folder_dir=f"folder")