altimate-datapilot-cli 0.0.14__py3-none-any.whl → 0.0.15__py3-none-any.whl

{altimate_datapilot_cli-0.0.14.dist-info → altimate_datapilot_cli-0.0.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: altimate-datapilot-cli
-Version: 0.0.14
+Version: 0.0.15
 Summary: Assistant for Data Teams
 Home-page: https://github.com/AltimateAI/datapilot-cli
 Author: Altimate Inc
@@ -30,12 +30,12 @@ Classifier: Topic :: Utilities
 Requires-Python: >=3.8
 License-File: LICENSE
 License-File: AUTHORS.rst
-Requires-Dist: click==8.1.7
-Requires-Dist: dbt-artifacts-parser==0.6.0
-Requires-Dist: ruamel.yaml==0.18.6
-Requires-Dist: tabulate==0.9.0
-Requires-Dist: requests==2.31.0
-Requires-Dist: sqlglot==25.30.0
+Requires-Dist: click~=8.1.7
+Requires-Dist: dbt-artifacts-parser~=0.8.1
+Requires-Dist: ruamel.yaml~=0.18.6
+Requires-Dist: tabulate~=0.9.0
+Requires-Dist: requests~=2.31.0
+Requires-Dist: sqlglot~=25.30.0
 
 ========
 Overview

{altimate_datapilot_cli-0.0.14.dist-info → altimate_datapilot_cli-0.0.15.dist-info}/RECORD

@@ -1,5 +1,6 @@
-datapilot/__init__.py,sha256=71Tw8stu19MsLPyF1q_DF5mb_BefzOTPYggLlSLvCms,23
+datapilot/__init__.py,sha256=go20U3RCVaJ2N55RnX4tO5rinfUCRV0puFyrHCto8yw,23
 datapilot/__main__.py,sha256=I9USmeNnK-cAHb6LZfydJC0LeNSE8enieeY55wpR6uw,380
+datapilot/constants.py,sha256=h0gfZ8qVyTEi0k7NFOQyH4Z1cPU2zb2W7GsoXASgpxg,112
 datapilot/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datapilot/cli/main.py,sha256=VSdqlkCiu8GSG9qQh8q0BzyocsQc4lKWxZAPEsjXF18,181
 datapilot/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -44,8 +45,8 @@ datapilot/core/platforms/dbt/insights/checks/check_column_name_contract.py,sha25
 datapilot/core/platforms/dbt/insights/checks/check_macro_args_have_desc.py,sha256=Px-gBR0dx9z-xLOcp2ctDuKMRKK4ai554woMpeBFGFs,3185
 datapilot/core/platforms/dbt/insights/checks/check_macro_has_desc.py,sha256=UrMYyZpi97gIBvH7leu5z27BowaFfPl0tOI4dqMSJmQ,2802
 datapilot/core/platforms/dbt/insights/checks/check_model_has_all_columns.py,sha256=LULWYrGHrXiEcFfv5VxbrQ3quw4h5mLn3Sd_pG4ayug,4277
-datapilot/core/platforms/dbt/insights/checks/check_model_has_labels_keys.py,sha256=ZaIRC8Euq5kpF8JI8SFOqauSR_PWmYAXHCmd1-1oVD8,5183
-datapilot/core/platforms/dbt/insights/checks/check_model_has_meta_keys.py,sha256=r_GsGvQCAfcZBl8fbzw4T9jccbxwqlCoMeMnoX8jSXw,5016
+datapilot/core/platforms/dbt/insights/checks/check_model_has_labels_keys.py,sha256=-cxWasRTJbZBhCd4JtTeygEJ90-m7v1YyhUzn0Jdjmw,5189
+datapilot/core/platforms/dbt/insights/checks/check_model_has_meta_keys.py,sha256=QClFNsYlxy8nvmf8LyJENk8kT624Cas9l78qSXjs868,5009
 datapilot/core/platforms/dbt/insights/checks/check_model_has_properties_file.py,sha256=hyl51rqSGmbj5DINtL5Dvex80SGSmN2o-qf_U6eaNMg,2968
 datapilot/core/platforms/dbt/insights/checks/check_model_has_tests_by_group.py,sha256=d2Csu6EAapFx6eIXuCaCdNllbsjIBVab5QmlIv-SqGM,5706
 datapilot/core/platforms/dbt/insights/checks/check_model_has_tests_by_name.py,sha256=bu87eQ5yZYZHtmdMEoIUA1Ynv8Tw3UGwCYKB_1_AXDI,5187
@@ -59,10 +60,10 @@ datapilot/core/platforms/dbt/insights/checks/check_model_tags.py,sha256=-268fA2k
 datapilot/core/platforms/dbt/insights/checks/check_source_childs.py,sha256=L8W66Hauwjg98OCE3NFfTzXIbmT0XehD2TysiaPe-Eo,4254
 datapilot/core/platforms/dbt/insights/checks/check_source_columns_have_desc.py,sha256=UzohMY6PGH_d-VCVJxvAbDI_DqeQqukmCpKU9W9fk_g,4453
 datapilot/core/platforms/dbt/insights/checks/check_source_has_all_columns.py,sha256=AkAyNas0_dl7EN-hZitN33nTaAM11zsNvlGJjLnZOd0,4650
-datapilot/core/platforms/dbt/insights/checks/check_source_has_freshness.py,sha256=CFT_2uYghd_Aak6EOkwQMyauzebVTskBMYmjLZPZlr4,4111
-datapilot/core/platforms/dbt/insights/checks/check_source_has_labels_keys.py,sha256=l8uKIdg-CCgC-w4-Ay_UTdNlAkrRlTY1Aw8G4pFvAXs,5256
+datapilot/core/platforms/dbt/insights/checks/check_source_has_freshness.py,sha256=1KP6g_-98hyjs3dTibFJFgBm97x_hJCeIYHdUR2HKDs,4117
+datapilot/core/platforms/dbt/insights/checks/check_source_has_labels_keys.py,sha256=UMEd2inyjQMNk7scnn196yoPtQYfOynS4nVGoyGpsqc,5262
 datapilot/core/platforms/dbt/insights/checks/check_source_has_loader.py,sha256=d3KixZeAVTlcKl8t-ap2n7pZU0qXDBeXYzjf813GkEE,2772
-datapilot/core/platforms/dbt/insights/checks/check_source_has_meta_keys.py,sha256=omzKNdBAYYwggt36Jz4BzaOnDqVRL1mJee0xNtYsKSo,5246
+datapilot/core/platforms/dbt/insights/checks/check_source_has_meta_keys.py,sha256=e4RNkMgGLRxOzqidMRZWKQgDdlFlB7LhHpoWlB5cJrQ,5239
 datapilot/core/platforms/dbt/insights/checks/check_source_has_tests.py,sha256=M7WWX1ud49FnInS_htvmNa_8gIjVuhk9tIuq7Q-4P5E,3711
 datapilot/core/platforms/dbt/insights/checks/check_source_has_tests_by_group.py,sha256=tvflQTxgdejTecDBspD6PdLLDHIWXY-_4uOzPWWqh2w,5550
 datapilot/core/platforms/dbt/insights/checks/check_source_has_tests_by_name.py,sha256=wDNYB0ZGx_hV8DxXewR1TtZ_HnozZN1Dyk0YOlGUoEs,5159
@@ -109,8 +110,8 @@ datapilot/core/platforms/dbt/insights/structure/model_naming_conventions.py,sha2
 datapilot/core/platforms/dbt/insights/structure/source_directories_structure.py,sha256=K-xTjYltuzcHH51s6PbFYcrnqpSCYluTbrzW1JT_vIU,3675
 datapilot/core/platforms/dbt/insights/structure/test_directory_structure.py,sha256=XL6YQgWmf397W2OwBmT7L8zUN93M1bOZBTp4PlngqlI,3484
 datapilot/core/platforms/dbt/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datapilot/core/platforms/dbt/schemas/catalog.py,sha256=izhzchuME5BTuNAj6iyDHpo6NAPsggG4rd6z6WxOWV8,1878
-datapilot/core/platforms/dbt/schemas/manifest.py,sha256=tkmGYWkn9DxnYzvHVH8GiypT3DhcHAVugzETfC9Q19o,14127
+datapilot/core/platforms/dbt/schemas/catalog.py,sha256=t8qDw2QTI6e3TMeVt_apw9L-WAe8JyPyZS6nf61m4rc,2023
+datapilot/core/platforms/dbt/schemas/manifest.py,sha256=zBr6-NgpKx06hQnw5yHmBwR0lOl2L8zyNISvN-eWD6g,14401
 datapilot/core/platforms/dbt/wrappers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datapilot/core/platforms/dbt/wrappers/catalog/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datapilot/core/platforms/dbt/wrappers/catalog/wrapper.py,sha256=LDjt4pSiqGAmvI5DSPzEubwq13N1MKv9xakLNhG-BWY,196
@@ -120,13 +121,13 @@ datapilot/core/platforms/dbt/wrappers/manifest/__init__.py,sha256=47DEQpj8HBSa-_
 datapilot/core/platforms/dbt/wrappers/manifest/wrapper.py,sha256=Q9DG-OEyJAa4ePoVhxG78pehfNotNc9A3xtz171fLhU,1189
 datapilot/core/platforms/dbt/wrappers/manifest/v10/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datapilot/core/platforms/dbt/wrappers/manifest/v10/schemas.py,sha256=VxqIFQbAL88ObQW1E8wPpiuooCdawdACJUFhgAYBxmo,1523
-datapilot/core/platforms/dbt/wrappers/manifest/v10/wrapper.py,sha256=7ewyoNPQbjcyNH90cugZh9Nad-cM3bF7XvlZKZ1uEGE,17536
+datapilot/core/platforms/dbt/wrappers/manifest/v10/wrapper.py,sha256=jXF_HDoBMR5r7IPNcL3YGAvY0ZEFpTmda0uNiElbJbY,18057
 datapilot/core/platforms/dbt/wrappers/manifest/v11/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datapilot/core/platforms/dbt/wrappers/manifest/v11/schemas.py,sha256=Mb0N48czcwQTz5mxQx2QlVINzz50A5FUm0kMJ-Nzs6A,1523
-datapilot/core/platforms/dbt/wrappers/manifest/v11/wrapper.py,sha256=DXOxuRHphwKfAr3WPoULteeZGdF8IHlHQu2lm382hrs,17536
+datapilot/core/platforms/dbt/wrappers/manifest/v11/wrapper.py,sha256=vuGFupQHyAJokWkgdzXmZd67S04nRkH89fEYCXeJWa0,17991
 datapilot/core/platforms/dbt/wrappers/manifest/v12/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datapilot/core/platforms/dbt/wrappers/manifest/v12/schemas.py,sha256=Muu2CVkTKbOtPrOcitysh6bxDrdLVw-Cu6HZT8J7Si8,1248
-datapilot/core/platforms/dbt/wrappers/manifest/v12/wrapper.py,sha256=ONARXJ3MhttSNUGnIc73CAmSiE2pZctHztwKLmZUGqg,17767
+datapilot/core/platforms/dbt/wrappers/manifest/v12/schemas.py,sha256=2rMEQ_BCcjiibfrlDKwgCE8vFX_6BzSInpilpVTWrNE,1269
+datapilot/core/platforms/dbt/wrappers/manifest/v12/wrapper.py,sha256=lAQY7rAI2luQep4sPogi1swVzcAZ_z5VzGwIl2XpKe0,17926
 datapilot/core/platforms/dbt/wrappers/run_results/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datapilot/core/platforms/dbt/wrappers/run_results/run_results.py,sha256=3E_y1gAF491WmXt-Z_Fqhr5BU-kVnzjHpZZv5UpOx-s,1267
 datapilot/exceptions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -139,10 +140,10 @@ datapilot/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datapilot/utils/utils.py,sha256=MY8q6ZBJ0hkrTuH7gWMxAlEAQGrajXFMabEhtGtT7sc,11524
 datapilot/utils/formatting/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datapilot/utils/formatting/utils.py,sha256=rAVmIYuldvw9VvCSwG2kMTEgiT7cEconp_F1sAWVyCo,1377
-altimate_datapilot_cli-0.0.14.dist-info/AUTHORS.rst,sha256=S4H4zw_v3GVyz5_55jF5Gf_YNG3s5Y0VgbQaEov9PFk,50
-altimate_datapilot_cli-0.0.14.dist-info/LICENSE,sha256=Mf7VqpsmU2QR5_s2Cb_ZeeMB2Q9KW7YXJENZPFZRK1k,1100
-altimate_datapilot_cli-0.0.14.dist-info/METADATA,sha256=s4DZVAhnfFUP7rj4HVF7H7Ote-5mozGCG8k6UySl5K4,2382
-altimate_datapilot_cli-0.0.14.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-altimate_datapilot_cli-0.0.14.dist-info/entry_points.txt,sha256=0zwgKxN40RLVB5jSmlJz7IH_FBqRtpFdbrdZn-xuQIY,141
-altimate_datapilot_cli-0.0.14.dist-info/top_level.txt,sha256=gAOFOdwB00vcxv74y4M1J-nQtPvEatU8-mYViEBcToo,10
-altimate_datapilot_cli-0.0.14.dist-info/RECORD,,
+altimate_datapilot_cli-0.0.15.dist-info/AUTHORS.rst,sha256=S4H4zw_v3GVyz5_55jF5Gf_YNG3s5Y0VgbQaEov9PFk,50
+altimate_datapilot_cli-0.0.15.dist-info/LICENSE,sha256=Mf7VqpsmU2QR5_s2Cb_ZeeMB2Q9KW7YXJENZPFZRK1k,1100
+altimate_datapilot_cli-0.0.15.dist-info/METADATA,sha256=WKXEoLY1k78gtF1S4mw3IYJZUu2Butum_gkDHf_n0dI,2382
+altimate_datapilot_cli-0.0.15.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+altimate_datapilot_cli-0.0.15.dist-info/entry_points.txt,sha256=0zwgKxN40RLVB5jSmlJz7IH_FBqRtpFdbrdZn-xuQIY,141
+altimate_datapilot_cli-0.0.15.dist-info/top_level.txt,sha256=gAOFOdwB00vcxv74y4M1J-nQtPvEatU8-mYViEBcToo,10
+altimate_datapilot_cli-0.0.15.dist-info/RECORD,,

datapilot/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.14"
+__version__ = "0.0.15"

datapilot/constants.py

@@ -0,0 +1,7 @@
+from enum import Enum
+
+
+class Extra(str, Enum):
+    allow = "allow"
+    forbid = "forbid"
+    ignore = "ignore"

datapilot/core/platforms/dbt/insights/checks/check_model_has_labels_keys.py

@@ -76,7 +76,7 @@ class CheckModelHasLabelsKeys(ChecksInsight):
     def _check_labels_keys(self, node_id) -> Tuple[int, Set[str]]:
         status_code = 0
         missing_keys = set(self.labels_keys) - set(self.get_node(node_id).label)
-        config = self.get_node(node_id).config.dict() if self.get_node(node_id).config else {}
+        config = self.get_node(node_id).config.model_dump() if self.get_node(node_id).config else {}
         labels = config.get("labels", {})
         label_keys = set(labels.keys())
         extra_keys = set()

datapilot/core/platforms/dbt/insights/checks/check_model_has_meta_keys.py

@@ -72,7 +72,7 @@ class CheckModelHasMetaKeys(ChecksInsight):
     def _check_meta_keys(self, node_id) -> Tuple[int, Set[str], Set[str]]:
         status_code = 0
         model = self.get_node(node_id)
-        meta = model.meta.dict() if model.meta else {}
+        meta = model.meta if model.meta else {}
         model_meta_keys = set(meta.keys())
         missing_keys = None
         extra_keys = None

datapilot/core/platforms/dbt/insights/checks/check_source_has_freshness.py

@@ -61,7 +61,7 @@ class CheckSourceHasFreshness(ChecksInsight):
 
     def _check_source_has_freshness(self, source_id: str) -> List[str]:
         source = self.get_node(source_id)
-        freshness = source.freshness.dict() if source.freshness else {}
+        freshness = source.freshness.model_dump() if source.freshness else {}
 
         if not freshness:
             return False

datapilot/core/platforms/dbt/insights/checks/check_source_has_labels_keys.py

@@ -74,7 +74,7 @@ class CheckSourceHasLabelsKeys(ChecksInsight):
     def _check_labels_keys(self, node_id) -> Tuple[int, Set[str]]:
         status_code = 0
         missing_keys = set(self.labels_keys) - set(self.get_node(node_id).label)
-        config = self.get_node(node_id).config.dict() if self.get_node(node_id).config else {}
+        config = self.get_node(node_id).config.model_dump() if self.get_node(node_id).config else {}
         labels = config.get("labels", {})
         label_keys = set(labels.keys())
         extra_keys = set()

datapilot/core/platforms/dbt/insights/checks/check_source_has_meta_keys.py

@@ -81,7 +81,7 @@ class CheckSourceHasMetaKeys(ChecksInsight):
     def _check_source_has_meta_keys(self, source_unique_id: str):
         status_code = 0
         model = self.get_node(source_unique_id)
-        meta = model.meta.dict() if model.meta else {}
+        meta = model.meta if model.meta else {}
         model_meta_keys = set(meta.keys())
         missing_keys = None
         extra_keys = None

datapilot/core/platforms/dbt/schemas/catalog.py

@@ -5,13 +5,16 @@ from typing import List
 from typing import Optional
 from typing import Union
 
-from pydantic.config import Extra
+from pydantic import ConfigDict
 from pydantic.main import BaseModel
 
+from datapilot.constants import Extra
+
 
 class AltimateCatalogMetadata(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     dbt_schema_version: Optional[str] = "https://schemas.getdbt.com/dbt/catalog/v1.json"
     dbt_version: Optional[str] = "0.19.0"
@@ -21,8 +24,9 @@ class AltimateCatalogMetadata(BaseModel):
 
 
 class AltimateCatalogTableMetadata(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     type: str
     database: Optional[Optional[str]] = None
@@ -33,8 +37,9 @@ class AltimateCatalogTableMetadata(BaseModel):
 
 
 class AltimateCatalogColumnMetadata(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     type: str
     comment: Optional[Optional[str]] = None
@@ -43,8 +48,9 @@ class AltimateCatalogColumnMetadata(BaseModel):
 
 
 class AltimateCatalogStatsItem(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     id: str
     label: str
@@ -54,8 +60,9 @@ class AltimateCatalogStatsItem(BaseModel):
 
 
 class AltimateCatalogTable(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     metadata: AltimateCatalogTableMetadata
     columns: Dict[str, AltimateCatalogColumnMetadata]
@@ -64,8 +71,9 @@ class AltimateCatalogTable(BaseModel):
 
 
 class AltimateCatalogCatalogV1(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     metadata: AltimateCatalogMetadata
     nodes: Dict[str, AltimateCatalogTable]

datapilot/core/platforms/dbt/schemas/manifest.py

@@ -19,7 +19,9 @@ from dbt_artifacts_parser.parsers.manifest.manifest_v10 import ManifestV10
 from dbt_artifacts_parser.parsers.manifest.manifest_v11 import ManifestV11
 from dbt_artifacts_parser.parsers.manifest.manifest_v11 import SupportedLanguage
 from pydantic import BaseModel
-from pydantic import Extra
+from pydantic import ConfigDict
+
+from datapilot.constants import Extra
 
 
 class DBTVersion(BaseModel):
@@ -46,16 +48,17 @@ Catalog = CatalogV1
 
 
 class AltimateDocs(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     show: Optional[bool] = True
     node_color: Optional[Optional[str]] = None
 
 
 class AltimateDependsOn(BaseModel):
-    nodes: Optional[List[str]]
-    macros: Optional[List[str]]
+    nodes: Optional[List[str]] = None
+    macros: Optional[List[str]] = None
 
 
 class AltimateManifestColumnInfo(BaseModel):
@@ -98,8 +101,9 @@ class AltimateAccess(Enum):
 
 
 class AltimateDBTContract(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     enforced: Optional[bool] = False
     alias_types: Optional[bool] = True
@@ -107,8 +111,9 @@ class AltimateDBTContract(BaseModel):
 
 
 class AltimateHook(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     sql: str
     transaction: Optional[bool] = True
@@ -117,8 +122,9 @@ class AltimateHook(BaseModel):
 
 # TODO: Need to add the rest of the fields
 class AltimateNodeConfig(BaseModel):
-    class Config:
-        extra = Extra.allow
+    model_config = ConfigDict(
+        extra=Extra.allow,
+    )
 
     _extra: Optional[Dict[str, Any]] = None
     enabled: Optional[bool] = True
@@ -203,8 +209,9 @@ class AltimateSourceConfig(BaseModel):
 
 
 class AltimateDeferRelation(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     database: Optional[str]
     schema_name: str
@@ -213,8 +220,9 @@ class AltimateDeferRelation(BaseModel):
 
 
 class AltimateSeedConfig(BaseModel):
-    class Config:
-        extra = Extra.allow
+    model_config = ConfigDict(
+        extra=Extra.allow,
+    )
 
     _extra: Optional[Dict[str, Any]] = None
     enabled: Optional[bool] = True
@@ -314,8 +322,9 @@ class AltimateExposureType(Enum):
 
 
 class AltimateOwner(BaseModel):
-    class Config:
-        extra = Extra.allow
+    model_config = ConfigDict(
+        extra=Extra.allow,
+    )
 
     _extra: Optional[Dict[str, Any]] = None
     email: Optional[Optional[str]] = None
@@ -329,8 +338,9 @@ class AltimateMaturityEnum(Enum):
 
 
 class AltimateRefArgs(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     name: str
     package: Optional[Optional[str]] = None
@@ -338,8 +348,9 @@ class AltimateRefArgs(BaseModel):
 
 
 class AltimateExposureConfig(BaseModel):
-    class Config:
-        extra = Extra.allow
+    model_config = ConfigDict(
+        extra=Extra.allow,
+    )
 
     _extra: Optional[Dict[str, Any]] = None
     enabled: Optional[bool] = True
@@ -371,8 +382,9 @@ class AltimateManifestExposureNode(BaseModel):
 
 
 class AltimateTestMetadata(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     name: str
     kwargs: Optional[Dict[str, Any]] = None
@@ -380,8 +392,9 @@ class AltimateTestMetadata(BaseModel):
 
 
 class AltimateTestConfig(BaseModel):
-    class Config:
-        extra = Extra.allow
+    model_config = ConfigDict(
+        extra=Extra.allow,
+    )
 
     _extra: Optional[Dict[str, Any]] = None
     enabled: Optional[bool] = True
@@ -433,8 +446,9 @@ class AltimateManifestTestNode(BaseModel):
 
 
 class AltimateMacroArgument(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(
+        extra=Extra.forbid,
+    )
 
     name: str
     type: Optional[Optional[str]] = None

datapilot/core/platforms/dbt/wrappers/manifest/v10/wrapper.py

@@ -116,6 +116,7 @@ class ManifestV10Wrapper(BaseManifestWrapper):
             contract=contract,
             meta=node.meta,
             patch_path=node.patch_path,
+            access=node.access.value,
         )
 
     def _get_source(self, source: SourceNode) -> AltimateManifestSourceNode:
@@ -133,10 +134,10 @@ class ManifestV10Wrapper(BaseManifestWrapper):
             source_description=source.source_description,
             loader=source.loader,
             identifier=source.identifier,
-            quoting=AltimateQuoting(**source.quoting.dict()) if source.quoting else None,
+            quoting=AltimateQuoting(**source.quoting.model_dump()) if source.quoting else None,
             loaded_at_field=source.loaded_at_field,
-            freshness=AltimateFreshnessThreshold(**source.freshness.dict()) if source.freshness else None,
-            external=AltimateExternalTable(**source.external.dict()) if source.external else None,
+            freshness=AltimateFreshnessThreshold(**source.freshness.model_dump()) if source.freshness else None,
+            external=AltimateExternalTable(**source.external.model_dump()) if source.external else None,
             description=source.description,
             columns={
                 name: AltimateManifestColumnInfo(
@@ -153,7 +154,7 @@ class ManifestV10Wrapper(BaseManifestWrapper):
             relation_name=source.relation_name,
             source_meta=source.source_meta,
             tags=source.tags,
-            config=AltimateSourceConfig(**source.config.dict()) if source.config else None,
+            config=AltimateSourceConfig(**source.config.model_dump()) if source.config else None,
             patch_path=source.patch_path,
             unrendered_config=source.unrendered_config,
             created_at=source.created_at,
@@ -177,9 +178,9 @@ class ManifestV10Wrapper(BaseManifestWrapper):
             ),
             description=macro.description,
             meta=macro.meta,
-            docs=macro.docs,
+            docs=macro.docs.model_dump() if macro.docs else None,
             patch_path=macro.patch_path,
-            arguments=[AltimateMacroArgument(**arg.dict()) for arg in macro.arguments] if macro.arguments else None,
+            arguments=[AltimateMacroArgument(**arg.model_dump()) for arg in macro.arguments] if macro.arguments else None,
             created_at=macro.created_at,
             supported_languages=macro.supported_languages,
         )
@@ -194,22 +195,24 @@ class ManifestV10Wrapper(BaseManifestWrapper):
             unique_id=exposure.unique_id,
             fqn=exposure.fqn,
             type=AltimateExposureType(exposure.type.value) if exposure.type else None,
-            owner=AltimateOwner(**exposure.owner.dict()) if exposure.owner else None,
+            owner=AltimateOwner(**exposure.owner.model_dump()) if exposure.owner else None,
             description=exposure.description,
             label=exposure.label,
             maturity=AltimateMaturityEnum(exposure.maturity.value) if exposure.maturity else None,
             meta=exposure.meta,
             tags=exposure.tags,
-            config=AltimateSourceConfig(**exposure.config.dict()) if exposure.config else None,
+            config=AltimateSourceConfig(**exposure.config.model_dump()) if exposure.config else None,
             unrendered_config=exposure.unrendered_config,
             url=exposure.url,
-            depends_on=AltimateDependsOn(
-                nodes=exposure.depends_on.nodes,
-                macros=exposure.depends_on.macros,
-            )
-            if exposure.depends_on
-            else None,
-            refs=[AltimateRefArgs(**ref.dict()) for ref in exposure.refs] if exposure.refs else None,
+            depends_on=(
+                AltimateDependsOn(
+                    nodes=exposure.depends_on.nodes,
+                    macros=exposure.depends_on.macros,
+                )
+                if exposure.depends_on
+                else None
+            ),
+            refs=[AltimateRefArgs(**ref.model_dump()) for ref in exposure.refs] if exposure.refs else None,
             sources=exposure.sources,
             metrics=exposure.metrics,
             created_at=exposure.created_at,
@@ -219,7 +222,7 @@ class ManifestV10Wrapper(BaseManifestWrapper):
         test_metadata = None
         if isinstance(test, GenericTestNode):
             test_type = GENERIC
-            test_metadata = AltimateTestMetadata(**test.test_metadata.dict()) if test.test_metadata else None
+            test_metadata = AltimateTestMetadata(**test.test_metadata.model_dump()) if test.test_metadata else None
         elif isinstance(test, SingularTestNode):
             test_type = SINGULAR
         else:
@@ -235,42 +238,48 @@ class ManifestV10Wrapper(BaseManifestWrapper):
             unique_id=test.unique_id,
             fqn=test.fqn,
             alias=test.alias,
-            checksum=AltimateFileHash(
-                name=test.checksum.name,
-                checksum=test.checksum.checksum,
-            )
-            if test.checksum
-            else None,
-            config=AltimateTestConfig(**test.config.dict()) if test.config else None,
+            checksum=(
+                AltimateFileHash(
+                    name=test.checksum.name,
+                    checksum=test.checksum.checksum,
+                )
+                if test.checksum
+                else None
+            ),
+            config=AltimateTestConfig(**test.config.model_dump()) if test.config else None,
             description=test.description,
             tags=test.tags,
-            columns={
-                name: AltimateManifestColumnInfo(
-                    name=column.name,
-                    description=column.description,
-                    meta=column.meta,
-                    data_type=column.data_type,
-                    quote=column.quote,
-                    tags=column.tags,
-                )
-                for name, column in test.columns.items()
-            }
-            if test.columns
-            else None,
+            columns=(
+                {
+                    name: AltimateManifestColumnInfo(
+                        name=column.name,
+                        description=column.description,
+                        meta=column.meta,
+                        data_type=column.data_type,
+                        quote=column.quote,
+                        tags=column.tags,
+                    )
+                    for name, column in test.columns.items()
+                }
+                if test.columns
+                else None
+            ),
             meta=test.meta,
             relation_name=test.relation_name,
             group=test.group,
             raw_code=test.raw_code,
             language=test.language,
-            refs=[AltimateRefArgs(**ref.dict()) for ref in test.refs] if test.refs else None,
+            refs=[AltimateRefArgs(**ref.model_dump()) for ref in test.refs] if test.refs else None,
             sources=test.sources,
             metrics=test.metrics,
-            depends_on=AltimateDependsOn(
-                nodes=test.depends_on.nodes,
-                macros=test.depends_on.macros,
-            )
-            if test.depends_on
-            else None,
+            depends_on=(
+                AltimateDependsOn(
+                    nodes=test.depends_on.nodes,
+                    macros=test.depends_on.macros,
+                )
+                if test.depends_on
+                else None
+            ),
             compiled_path=test.compiled_path,
             compiled=test.compiled,
             compiled_code=test.compiled_code,
@@ -288,31 +297,35 @@ class ManifestV10Wrapper(BaseManifestWrapper):
             unique_id=seed.unique_id,
             fqn=seed.fqn,
             alias=seed.alias,
-            checksum=AltimateFileHash(
-                name=seed.checksum.name,
-                checksum=seed.checksum.checksum,
-            )
-            if seed.checksum
-            else None,
-            config=AltimateSeedConfig(**seed.config.dict()) if seed.config else None,
+            checksum=(
+                AltimateFileHash(
+                    name=seed.checksum.name,
+                    checksum=seed.checksum.checksum,
+                )
+                if seed.checksum
+                else None
+            ),
+            config=AltimateSeedConfig(**seed.config.model_dump()) if seed.config else None,
             description=seed.description,
             tags=seed.tags,
-            columns={
-                name: AltimateManifestColumnInfo(
-                    name=column.name,
-                    description=column.description,
-                    meta=column.meta,
-                    data_type=column.data_type,
-                    quote=column.quote,
-                    tags=column.tags,
-                )
-                for name, column in seed.columns.items()
-            }
-            if seed.columns
-            else None,
+            columns=(
+                {
+                    name: AltimateManifestColumnInfo(
+                        name=column.name,
+                        description=column.description,
+                        meta=column.meta,
+                        data_type=column.data_type,
+                        quote=column.quote,
+                        tags=column.tags,
+                    )
+                    for name, column in seed.columns.items()
+                }
+                if seed.columns
+                else None
+            ),
             meta=seed.meta,
             group=seed.group,
-            docs=seed.docs.dict() if seed.docs else None,
+            docs=seed.docs.model_dump() if seed.docs else None,
             patch_path=seed.patch_path,
             build_path=seed.build_path,
             deferred=seed.deferred,

datapilot/core/platforms/dbt/wrappers/manifest/v11/wrapper.py

@@ -61,7 +61,7 @@ class ManifestV11Wrapper(BaseManifestWrapper):
             language,
             contract,
         ) = ([], [], None, None, None, None, None, "", "", None)
-        if node.resource_type.value != SEED:
+        if node.resource_type != SEED:
             sources = node.sources
             metrics = node.metrics
             depends_on_nodes = node.depends_on.nodes if node.depends_on else None
@@ -77,7 +77,7 @@ class ManifestV11Wrapper(BaseManifestWrapper):
             database=node.database,
             schema_name=node.schema_,
             name=node.name,
-            resource_type=AltimateResourceType(node.resource_type.value),
+            resource_type=AltimateResourceType(node.resource_type),
             package_name=node.package_name,
             path=node.path,
             description=node.description,
@@ -116,12 +116,13 @@ class ManifestV11Wrapper(BaseManifestWrapper):
             contract=contract,
             meta=node.meta,
             patch_path=node.patch_path,
+            access=node.access.value,
         )
 
     def _get_source(self, source: SourceNode) -> AltimateManifestSourceNode:
         return AltimateManifestSourceNode(
             database=source.database,
-            resource_type=AltimateResourceType(source.resource_type.value),
+            resource_type=AltimateResourceType(source.resource_type),
             schema_name=source.schema_,
             name=source.name,
             package_name=source.package_name,
@@ -133,10 +134,10 @@ class ManifestV11Wrapper(BaseManifestWrapper):
             source_description=source.source_description,
             loader=source.loader,
             identifier=source.identifier,
-            quoting=AltimateQuoting(**source.quoting.dict()) if source.quoting else None,
+            quoting=AltimateQuoting(**source.quoting.model_dump()) if source.quoting else None,
             loaded_at_field=source.loaded_at_field,
-            freshness=AltimateFreshnessThreshold(**source.freshness.dict()) if source.freshness else None,
-            external=AltimateExternalTable(**source.external.dict()) if source.external else None,
+            freshness=AltimateFreshnessThreshold(**source.freshness.model_dump()) if source.freshness else None,
+            external=AltimateExternalTable(**source.external.model_dump()) if source.external else None,
             description=source.description,
             columns={
                 name: AltimateManifestColumnInfo(
@@ -153,7 +154,7 @@ class ManifestV11Wrapper(BaseManifestWrapper):
             relation_name=source.relation_name,
             source_meta=source.source_meta,
             tags=source.tags,
-            config=AltimateSourceConfig(**source.config.dict()) if source.config else None,
+            config=AltimateSourceConfig(**source.config.model_dump()) if source.config else None,
             patch_path=source.patch_path,
             unrendered_config=source.unrendered_config,
             created_at=source.created_at,
@@ -162,7 +163,7 @@ class ManifestV11Wrapper(BaseManifestWrapper):
     def _get_macro(self, macro: MacroNode) -> AltimateManifestMacroNode:
         return AltimateManifestMacroNode(
             name=macro.name,
-            resource_type=AltimateResourceType(macro.resource_type.value),
+            resource_type=AltimateResourceType(macro.resource_type),
             package_name=macro.package_name,
             path=macro.path,
             original_file_path=macro.original_file_path,
@@ -177,9 +178,9 @@ class ManifestV11Wrapper(BaseManifestWrapper):
             ),
             description=macro.description,
             meta=macro.meta,
-            docs=macro.docs,
+            docs=macro.docs.model_dump() if macro.docs else None,
             patch_path=macro.patch_path,
-            arguments=[AltimateMacroArgument(**arg.dict()) for arg in macro.arguments] if macro.arguments else None,
+            arguments=[AltimateMacroArgument(**arg.model_dump()) for arg in macro.arguments] if macro.arguments else None,
             created_at=macro.created_at,
             supported_languages=macro.supported_languages,
         )
@@ -187,29 +188,31 @@ class ManifestV11Wrapper(BaseManifestWrapper):
     def _get_exposure(self, exposure: ExposureNode) -> AltimateManifestExposureNode:
         return AltimateManifestExposureNode(
             name=exposure.name,
-            resource_type=AltimateResourceType(exposure.resource_type.value),
+            resource_type=AltimateResourceType(exposure.resource_type),
             package_name=exposure.package_name,
             path=exposure.path,
             original_file_path=exposure.original_file_path,
             unique_id=exposure.unique_id,
             fqn=exposure.fqn,
             type=AltimateExposureType(exposure.type.value) if exposure.type else None,
-            owner=AltimateOwner(**exposure.owner.dict()) if exposure.owner else None,
+            owner=AltimateOwner(**exposure.owner.model_dump()) if exposure.owner else None,
             description=exposure.description,
             label=exposure.label,
             maturity=AltimateMaturityEnum(exposure.maturity.value) if exposure.maturity else None,
             meta=exposure.meta,
             tags=exposure.tags,
-            config=AltimateSourceConfig(**exposure.config.dict()) if exposure.config else None,
+            config=AltimateSourceConfig(**exposure.config.model_dump()) if exposure.config else None,
             unrendered_config=exposure.unrendered_config,
             url=exposure.url,
-            depends_on=AltimateDependsOn(
-                nodes=exposure.depends_on.nodes,
-                macros=exposure.depends_on.macros,
-            )
-            if exposure.depends_on
-            else None,
-            refs=[AltimateRefArgs(**ref.dict()) for ref in exposure.refs] if exposure.refs else None,
+            depends_on=(
+                AltimateDependsOn(
+                    nodes=exposure.depends_on.nodes,
+                    macros=exposure.depends_on.macros,
+                )
+                if exposure.depends_on
+                else None
+            ),
+            refs=[AltimateRefArgs(**ref.model_dump()) for ref in exposure.refs] if exposure.refs else None,
             sources=exposure.sources,
             metrics=exposure.metrics,
             created_at=exposure.created_at,
@@ -219,7 +222,7 @@ class ManifestV11Wrapper(BaseManifestWrapper):
         test_metadata = None
         if isinstance(test, GenericTestNode):
             test_type = GENERIC
-            test_metadata = AltimateTestMetadata(**test.test_metadata.dict()) if test.test_metadata else None
+            test_metadata = AltimateTestMetadata(**test.test_metadata.model_dump()) if test.test_metadata else None
         elif isinstance(test, SingularTestNode):
             test_type = SINGULAR
         else:
@@ -228,49 +231,55 @@ class ManifestV11Wrapper(BaseManifestWrapper):
             test_metadata=test_metadata,
             test_type=test_type,
             name=test.name,
-            resource_type=AltimateResourceType(test.resource_type.value),
+            resource_type=AltimateResourceType(test.resource_type),
             package_name=test.package_name,
             path=test.path,
             original_file_path=test.original_file_path,
             unique_id=test.unique_id,
             fqn=test.fqn,
             alias=test.alias,
-            checksum=AltimateFileHash(
-                name=test.checksum.name,
-                checksum=test.checksum.checksum,
-            )
-            if test.checksum
-            else None,
-            config=AltimateTestConfig(**test.config.dict()) if test.config else None,
+            checksum=(
+                AltimateFileHash(
+                    name=test.checksum.name,
+                    checksum=test.checksum.checksum,
+                )
+                if test.checksum
+                else None
+            ),
+            config=AltimateTestConfig(**test.config.model_dump()) if test.config else None,
             description=test.description,
             tags=test.tags,
-            columns={
-                name: AltimateManifestColumnInfo(
-                    name=column.name,
-                    description=column.description,
-                    meta=column.meta,
-                    data_type=column.data_type,
-                    quote=column.quote,
-                    tags=column.tags,
-                )
-                for name, column in test.columns.items()
-            }
-            if test.columns
-            else None,
+            columns=(
+                {
+                    name: AltimateManifestColumnInfo(
+                        name=column.name,
+                        description=column.description,
+                        meta=column.meta,
+                        data_type=column.data_type,
+                        quote=column.quote,
+                        tags=column.tags,
+                    )
+                    for name, column in test.columns.items()
+                }
+                if test.columns
+                else None
+            ),
             meta=test.meta,
             relation_name=test.relation_name,
             group=test.group,
             raw_code=test.raw_code,
             language=test.language,
-            refs=[AltimateRefArgs(**ref.dict()) for ref in test.refs] if test.refs else None,
+            refs=[AltimateRefArgs(**ref.model_dump()) for ref in test.refs] if test.refs else None,
             sources=test.sources,
             metrics=test.metrics,
-            depends_on=AltimateDependsOn(
-                nodes=test.depends_on.nodes,
-                macros=test.depends_on.macros,
-            )
-            if test.depends_on
-            else None,
+            depends_on=(
+                AltimateDependsOn(
+                    nodes=test.depends_on.nodes,
+                    macros=test.depends_on.macros,
+                )
+                if test.depends_on
+                else None
+            ),
             compiled_path=test.compiled_path,
             compiled=test.compiled,
             compiled_code=test.compiled_code,
@@ -281,38 +290,42 @@ class ManifestV11Wrapper(BaseManifestWrapper):
             database=seed.database,
             schema_name=seed.schema_,
             name=seed.name,
-            resource_type=AltimateResourceType(seed.resource_type.value),
+            resource_type=AltimateResourceType(seed.resource_type),
             package_name=seed.package_name,
             path=seed.path,
             original_file_path=seed.original_file_path,
             unique_id=seed.unique_id,
             fqn=seed.fqn,
             alias=seed.alias,
-            checksum=AltimateFileHash(
-                name=seed.checksum.name,
-                checksum=seed.checksum.checksum,
-            )
-            if seed.checksum
-            else None,
-            config=AltimateSeedConfig(**seed.config.dict()) if seed.config else None,
+            checksum=(
+                AltimateFileHash(
+                    name=seed.checksum.name,
+                    checksum=seed.checksum.checksum,
+                )
+                if seed.checksum
+                else None
+            ),
+            config=AltimateSeedConfig(**seed.config.model_dump()) if seed.config else None,
             description=seed.description,
             tags=seed.tags,
-            columns={
-                name: AltimateManifestColumnInfo(
-                    name=column.name,
-                    description=column.description,
-                    meta=column.meta,
-                    data_type=column.data_type,
-                    quote=column.quote,
-                    tags=column.tags,
-                )
-                for name, column in seed.columns.items()
-            }
-            if seed.columns
-            else None,
+            columns=(
+                {
+                    name: AltimateManifestColumnInfo(
+                        name=column.name,
+                        description=column.description,
+                        meta=column.meta,
+                        data_type=column.data_type,
+                        quote=column.quote,
+                        tags=column.tags,
+                    )
+                    for name, column in seed.columns.items()
+                }
+                if seed.columns
+                else None
+            ),
             meta=seed.meta,
             group=seed.group,
-            docs=seed.docs.dict() if seed.docs else None,
+            docs=seed.docs.model_dump() if seed.docs else None,
             patch_path=seed.patch_path,
             build_path=seed.build_path,
             deferred=seed.deferred,
@@ -327,7 +340,7 @@ class ManifestV11Wrapper(BaseManifestWrapper):
         nodes = {}
         for node in self.manifest.nodes.values():
             if (
-                node.resource_type.value
+                node.resource_type
                 in [
                     AltimateResourceType.seed.value,
                     AltimateResourceType.test.value,
@@ -350,7 +363,7 @@ class ManifestV11Wrapper(BaseManifestWrapper):
     def get_macros(self) -> Dict[str, AltimateManifestMacroNode]:
         macros = {}
         for macro in self.manifest.macros.values():
-            if macro.resource_type.value == AltimateResourceType.macro.value and macro.package_name == self.get_package():
+            if macro.resource_type == AltimateResourceType.macro.value and macro.package_name == self.get_package():
                 macros[macro.unique_id] = self._get_macro(macro)
         return macros
 
@@ -371,7 +384,7 @@ class ManifestV11Wrapper(BaseManifestWrapper):
 
         for node in self.manifest.nodes.values():
             # Check if the node is a test and of the correct type
-            if node.resource_type.value == AltimateResourceType.test.value:
+            if node.resource_type == AltimateResourceType.test.value:
                 if any(isinstance(node, t) for t in types):
                     tests[node.unique_id] = self._get_tests(node)
         return tests
@@ -379,7 +392,7 @@ class ManifestV11Wrapper(BaseManifestWrapper):
     def get_seeds(self) -> Dict[str, AltimateSeedNode]:
         seeds = {}
         for seed in self.manifest.nodes.values():
-            if seed.resource_type.value == AltimateResourceType.seed.value:
+            if seed.resource_type == AltimateResourceType.seed.value:
                 seeds[seed.unique_id] = self._get_seed(seed)
         return seeds
 

datapilot/core/platforms/dbt/wrappers/manifest/v12/schemas.py

@@ -4,33 +4,33 @@ from typing import Union
 
 from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Exposures
 from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Macros
-from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Node
-from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Node1
-from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Node2
-from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Node3
-from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Node4
-from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Node5
-from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Node6
-from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Node7
+from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Nodes
+from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Nodes1
+from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Nodes2
+from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Nodes3
+from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Nodes4
+from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Nodes5
+from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Nodes6
+from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Nodes7
 from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Sources
 
 from datapilot.core.platforms.dbt.constants import GENERIC
 from datapilot.core.platforms.dbt.constants import SINGULAR
 
-ManifestNode = Union[Node, Node1, Node2, Node3, Node4, Node5, Node6, Node7]
+ManifestNode = Union[Nodes, Nodes1, Nodes2, Nodes3, Nodes4, Nodes5, Nodes6, Nodes7]
 
 SourceNode = Sources
 
 ExposureNode = Exposures
 
-TestNode = Union[Node6, Node2]
+TestNode = Union[Nodes6, Nodes2]
 
 MacroNode = Macros
 
 TEST_TYPE_TO_NODE_MAP: Dict[str, Type] = {
-    GENERIC: [Node6],
-    SINGULAR: [Node2],
+    GENERIC: [Nodes6],
+    SINGULAR: [Nodes2],
 }
 
 
-SeedNodeMap = Node
+SeedNodeMap = Nodes

datapilot/core/platforms/dbt/wrappers/manifest/v12/wrapper.py

@@ -3,8 +3,8 @@ from typing import Optional
 from typing import Set
 
 from dbt_artifacts_parser.parsers.manifest.manifest_v12 import ManifestV12
-from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Node2
-from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Node6
+from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Nodes2
+from dbt_artifacts_parser.parsers.manifest.manifest_v12 import Nodes6
 
 from datapilot.core.platforms.dbt.constants import GENERIC
 from datapilot.core.platforms.dbt.constants import OTHER_TEST_NODE
@@ -116,6 +116,7 @@ class ManifestV12Wrapper(BaseManifestWrapper):
             contract=contract,
             meta=node.meta,
             patch_path=node.patch_path,
+            access=node.access.value,
         )
 
     def _get_source(self, source: SourceNode) -> AltimateManifestSourceNode:
@@ -133,10 +134,10 @@ class ManifestV12Wrapper(BaseManifestWrapper):
             source_description=source.source_description,
             loader=source.loader,
             identifier=source.identifier,
-            quoting=AltimateQuoting(**source.quoting.dict()) if source.quoting else None,
+            quoting=AltimateQuoting(**source.quoting.model_dump()) if source.quoting else None,
             loaded_at_field=source.loaded_at_field,
-            freshness=AltimateFreshnessThreshold(**source.freshness.dict()) if source.freshness else None,
-            external=AltimateExternalTable(**source.external.dict()) if source.external else None,
+            freshness=AltimateFreshnessThreshold(**source.freshness.model_dump()) if source.freshness else None,
+            external=AltimateExternalTable(**source.external.model_dump()) if source.external else None,
             description=source.description,
             columns={
                 name: AltimateManifestColumnInfo(
@@ -153,7 +154,7 @@ class ManifestV12Wrapper(BaseManifestWrapper):
             relation_name=source.relation_name,
             source_meta=source.source_meta,
             tags=source.tags,
-            config=AltimateSourceConfig(**source.config.dict()) if source.config else None,
+            config=AltimateSourceConfig(**source.config.model_dump()) if source.config else None,
             patch_path=source.patch_path,
             unrendered_config=source.unrendered_config,
             created_at=source.created_at,
@@ -177,9 +178,9 @@ class ManifestV12Wrapper(BaseManifestWrapper):
             ),
             description=macro.description,
             meta=macro.meta,
-            docs=macro.docs,
+            docs=macro.docs.model_dump() if macro.docs else None,
             patch_path=macro.patch_path,
-            arguments=[AltimateMacroArgument(**arg.dict()) for arg in macro.arguments] if macro.arguments else None,
+            arguments=[AltimateMacroArgument(**arg.model_dump()) for arg in macro.arguments] if macro.arguments else None,
             created_at=macro.created_at,
             supported_languages=macro.supported_languages,
         )
@@ -194,13 +195,13 @@ class ManifestV12Wrapper(BaseManifestWrapper):
             unique_id=exposure.unique_id,
             fqn=exposure.fqn,
             type=AltimateExposureType(exposure.type.value) if exposure.type else None,
-            owner=AltimateOwner(**exposure.owner.dict()) if exposure.owner else None,
+            owner=AltimateOwner(**exposure.owner.model_dump()) if exposure.owner else None,
             description=exposure.description,
             label=exposure.label,
             maturity=AltimateMaturityEnum(exposure.maturity.value) if exposure.maturity else None,
             meta=exposure.meta,
             tags=exposure.tags,
-            config=AltimateSourceConfig(**exposure.config.dict()) if exposure.config else None,
+            config=AltimateSourceConfig(**exposure.config.model_dump()) if exposure.config else None,
             unrendered_config=exposure.unrendered_config,
             url=exposure.url,
             depends_on=(
@@ -211,7 +212,7 @@ class ManifestV12Wrapper(BaseManifestWrapper):
                 if exposure.depends_on
                 else None
             ),
-            refs=[AltimateRefArgs(**ref.dict()) for ref in exposure.refs] if exposure.refs else None,
+            refs=[AltimateRefArgs(**ref.model_dump()) for ref in exposure.refs] if exposure.refs else None,
             sources=exposure.sources,
             metrics=exposure.metrics,
             created_at=exposure.created_at,
@@ -219,10 +220,10 @@ class ManifestV12Wrapper(BaseManifestWrapper):
 
     def _get_tests(self, test: TestNode) -> AltimateManifestTestNode:
         test_metadata = None
-        if isinstance(test, Node6):
+        if isinstance(test, Nodes6):
             test_type = GENERIC
-            test_metadata = AltimateTestMetadata(**test.test_metadata.dict()) if test.test_metadata else None
-        elif isinstance(test, Node2):
+            test_metadata = AltimateTestMetadata(**test.test_metadata.model_dump()) if test.test_metadata else None
+        elif isinstance(test, Nodes2):
             test_type = SINGULAR
         else:
             test_type = OTHER_TEST_NODE
@@ -245,7 +246,7 @@ class ManifestV12Wrapper(BaseManifestWrapper):
                 if test.checksum
                 else None
             ),
-            config=AltimateTestConfig(**test.config.dict()) if test.config else None,
+            config=AltimateTestConfig(**test.config.model_dump()) if test.config else None,
             description=test.description,
             tags=test.tags,
             columns=(
@@ -268,7 +269,7 @@ class ManifestV12Wrapper(BaseManifestWrapper):
             group=test.group,
             raw_code=test.raw_code,
             language=test.language,
-            refs=[AltimateRefArgs(**ref.dict()) for ref in test.refs] if test.refs else None,
+            refs=[AltimateRefArgs(**ref.model_dump()) for ref in test.refs] if test.refs else None,
             sources=test.sources,
             metrics=test.metrics,
             depends_on=(
@@ -304,7 +305,7 @@ class ManifestV12Wrapper(BaseManifestWrapper):
                 if seed.checksum
                 else None
             ),
-            config=AltimateSeedConfig(**seed.config.dict()) if seed.config else None,
+            config=AltimateSeedConfig(**seed.config.model_dump()) if seed.config else None,
             description=seed.description,
             tags=seed.tags,
             columns=(
@@ -324,7 +325,7 @@ class ManifestV12Wrapper(BaseManifestWrapper):
             ),
             meta=seed.meta,
             group=seed.group,
-            docs=seed.docs.dict() if seed.docs else None,
+            docs=seed.docs.model_dump() if seed.docs else None,
             patch_path=seed.patch_path,
             build_path=seed.build_path,
             deferred=False,
@@ -375,7 +376,7 @@ class ManifestV12Wrapper(BaseManifestWrapper):
     def get_tests(self, type=None) -> Dict[str, AltimateManifestTestNode]:
         tests = {}
         # Initialize types_union with TestNode
-        types = [Node2, Node6]
+        types = [Nodes2, Nodes6]
 
         # Add other types to the union if provided
         if type: