diff --git a/soda-bigquery/src/soda_bigquery/common/data_sources/bigquery_data_source_connection.py b/soda-bigquery/src/soda_bigquery/common/data_sources/bigquery_data_source_connection.py index 72eeb2f0b..3e5fcdf05 100644 --- a/soda-bigquery/src/soda_bigquery/common/data_sources/bigquery_data_source_connection.py +++ b/soda-bigquery/src/soda_bigquery/common/data_sources/bigquery_data_source_connection.py @@ -27,6 +27,7 @@ class BigQueryConnectionProperties(DataSourceConnectionProperties, ABC): project_id: Optional[str] = Field(None, description="BigQuery project ID") + storage_project_id: Optional[str] = Field(None, description="BigQuery storage project ID") location: Optional[str] = Field(None, description="BigQuery location") client_options: Optional[dict] = Field(None, description="Client options") labels: Optional[dict] = Field({}, description="Labels") @@ -117,6 +118,10 @@ def _apply_optional_params(self, config: BigQueryConnectionProperties): self.location = config.location self.client_options = config.client_options + # Storage project ID is currently not used, because the project is configured via the DQN in the data contract. + # When we implement discovery, we'll need to use this value. + self.storage_project_id = config.storage_project_id if config.storage_project_id else self.project_id + self.labels = config.labels def _create_connection(