changes

Author: gaurpulkit

src/vendor/dbt_artifacts_parser/parsers/run_results/run_results_v6.py

@@ -64,6 +64,12 @@ class BatchResults(BaseParserModel):
     failed: Optional[list[list]] = None
 
 
+class AdapterResponse(BaseParserModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+
+
 class Result(BaseParserModel):
     model_config = ConfigDict(
         extra="allow",
@@ -72,7 +78,7 @@ class Result(BaseParserModel):
     timing: list[TimingItem]
     thread_id: str
     execution_time: float
-    adapter_response: dict[str, Any]
+    adapter_response: AdapterResponse
     message: Optional[str] = None
     failures: Optional[int] = None
     unique_id: str
@@ -82,11 +88,17 @@ class Result(BaseParserModel):
     batch_results: Optional[BatchResults] = None
 
 
+class Args(BaseParserModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+
+
 class RunResultsV6(BaseParserModel):
     model_config = ConfigDict(
         extra="allow",
     )
     metadata: Metadata = Field(..., title="BaseArtifactMetadata")
     results: list[Result]
     elapsed_time: float
-    args: Optional[dict[str, Any]] = None
+    args: Optional[Args] = None

src/vendor/dbt_artifacts_parser/parsers/sources/sources_v3.py

@@ -85,6 +85,12 @@ class TimingItem(BaseParserModel):
     completed_at: Optional[str] = None
 
 
+class AdapterResponse(BaseParserModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+
+
 class Results1(BaseParserModel):
     model_config = ConfigDict(
         extra="allow",
@@ -95,7 +101,7 @@ class Results1(BaseParserModel):
     max_loaded_at_time_ago_in_s: float
     status: Status1
     criteria: Criteria = Field(..., title="FreshnessThreshold")
-    adapter_response: dict[str, Any]
+    adapter_response: AdapterResponse
     timing: list[TimingItem]
     thread_id: str
     execution_time: float