Add BQ Job labels to copy_bq_table

dbt-labs · Jan 30, 2025 · 83d55b0 · 83d55b0
1 parent a219818
commit 83d55b0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dbt/adapters/bigquery/connections.py b/dbt/adapters/bigquery/connections.py
@@ -424,6 +424,7 @@ def copy_bq_table(self, source, destination, write_disposition) -> None:
         destination_ref = self.table_ref(
             destination.database, destination.schema, destination.table
         )
+        labels = self.get_labels_from_query_comment()
 
         logger.debug(
             'Copying table(s) "{}" to "{}" with disposition: "{}"',
@@ -440,7 +441,7 @@ def copy_bq_table(self, source, destination, write_disposition) -> None:
             copy_job = client.copy_table(
                 source_ref_array,
                 destination_ref,
-                job_config=CopyJobConfig(write_disposition=write_disposition),
+                job_config=CopyJobConfig(write_disposition=write_disposition, labels=labels),
                 retry=self._retry.create_reopen_with_deadline(conn),
             )
             copy_job.result(timeout=self._retry.create_job_execution_timeout(fallback=300))