fix(trigger): fix baseUrl when checking run status

kestra-io · Sep 13, 2024 · 358c5bc · 358c5bc
1 parent 7855e13
commit 358c5bc
Showing 1 changed file with 17 additions and 10 deletions.
diff --git a/src/main/java/io/kestra/plugin/dbt/cloud/TriggerRun.java b/src/main/java/io/kestra/plugin/dbt/cloud/TriggerRun.java
@@ -12,14 +12,20 @@
 import io.micronaut.http.HttpResponse;
 import io.micronaut.http.uri.UriTemplate;
 import io.swagger.v3.oas.annotations.media.Schema;
-import lombok.*;
+import jakarta.validation.constraints.NotNull;
+import lombok.Builder;
+import lombok.EqualsAndHashCode;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.ToString;
 import lombok.experimental.SuperBuilder;
 import org.slf4j.Logger;
 
 import java.net.URI;
 import java.time.Duration;
-import java.util.*;
-import jakarta.validation.constraints.NotNull;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 
 @SuperBuilder
 @ToString
@@ -226,13 +232,14 @@ public TriggerRun.Output run(RunContext runContext) throws Exception {
         }
 
         CheckStatus checkStatusJob = CheckStatus.builder()
-                                        .runId(runId.toString())
-                                        .token(getToken())
-                                        .accountId(getAccountId())
-                                        .pollFrequency(getPollFrequency())
-                                        .maxDuration(getMaxDuration())
-                                        .parseRunResults(getParseRunResults())
-                                        .build();
+            .runId(runId.toString())
+            .baseUrl(getBaseUrl())
+            .token(getToken())
+            .accountId(getAccountId())
+            .pollFrequency(getPollFrequency())
+            .maxDuration(getMaxDuration())
+            .parseRunResults(getParseRunResults())
+            .build();
 
         CheckStatus.Output runOutput = checkStatusJob.run(runContext);