added schema override

dlt/helpers/dbt_cloud/__init__.py

@@ -21,21 +21,24 @@ def run_dbt_cloud_job(
         api_token=credentials.api_token,
         account_id=credentials.account_id,
     )
-    json_data = {}
+    json_data = {"cause": credentials.cause}
 
-    if credentials.package_sha:
-        json_data["git_sha"] = credentials.package_sha
+    if credentials.git_sha:
+        json_data["git_sha"] = credentials.git_sha
 
-    if credentials.package_repository_branch:
-        json_data["git_branch"] = credentials.package_repository_branch
+    elif credentials.git_branch:
+        json_data["git_branch"] = credentials.git_branch
+
+    if credentials.schema_override:
+        json_data["schema_override"] = credentials.schema_override
 
     if data:
         json_data.update(json_data)
 
     run_id = operator.trigger_job_run(job_id=credentials.job_id, data=json_data)
+    status = operator.get_run_status(run_id)
 
-    status = {}
-    if wait_for_outcome:
+    if wait_for_outcome and status["in_progress"]:
         while True:
             status = operator.get_run_status(run_id)
             if not status["in_progress"]:

dlt/helpers/dbt_cloud/client.py

@@ -106,7 +106,7 @@ def trigger_job_run(self, job_id: int | str, data: Optional[dict] = None) -> int
                 f"account_id and job_id are required, got account_id: {self.account_id} and job_id: {job_id}"
             )
 
-        json_body = {"cause": "Triggered via API"}
+        json_body = {}
         if data:
             json_body.update(data)
 

dlt/helpers/dbt_cloud/configuration.py

@@ -15,5 +15,7 @@ class DBTCloudConfiguration(BaseConfiguration):
     environment_id: Optional[str]
     run_id: Optional[str]
 
-    package_sha: Optional[str]
-    package_repository_branch: Optional[str]
+    cause: str = "Triggered via API"
+    git_sha: Optional[str]
+    git_branch: Optional[str]
+    schema_override: Optional[str]