Allows a snapshot to have a list as a unique_key

dbt-labs · Apr 22, 2024 · 4a88879 · 4a88879
1 parent 5c8a4ab
commit 4a88879
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/core/dbt/artifacts/resources/v1/snapshot.py b/core/dbt/artifacts/resources/v1/snapshot.py
@@ -10,11 +10,11 @@
 class SnapshotConfig(NodeConfig):
     materialized: str = "snapshot"
     strategy: Optional[str] = None
-    unique_key: Optional[str] = None
     target_schema: Optional[str] = None
     target_database: Optional[str] = None
     updated_at: Optional[str] = None
     # Not using Optional because of serialization issues with a Union of str and List[str]
+    unique_key: Union[str, List[str], None] = None
     check_cols: Union[str, List[str], None] = None
 
     @classmethod

diff --git a/core/dbt/contracts/graph/model_config.py b/core/dbt/contracts/graph/model_config.py
@@ -1,5 +1,5 @@
 from dataclasses import field, dataclass
-from typing import Any, List, Optional, Dict, Type
+from typing import Any, List, Optional, Dict, Type, Union
 
 from dbt.artifacts.resources import (
     ExposureConfig,
@@ -42,7 +42,7 @@ class UnitTestNodeConfig(NodeConfig):
 @dataclass
 class EmptySnapshotConfig(NodeConfig):
     materialized: str = "snapshot"
-    unique_key: Optional[str] = None  # override NodeConfig unique_key definition
+    unique_key: Union[str, List[str], None] = None  # override NodeConfig unique_key definition
 
 
 RESOURCE_TYPES: Dict[NodeType, Type[BaseConfig]] = {