altimate-datapilot-cli 0.2.1__tar.gz → 0.2.3__tar.gz

{altimate_datapilot_cli-0.2.1 → altimate_datapilot_cli-0.2.3}/.bumpversion.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.2.1
+current_version = 0.2.3
 commit = True
 tag = True
 

{altimate_datapilot_cli-0.2.1/src/altimate_datapilot_cli.egg-info → altimate_datapilot_cli-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: altimate-datapilot-cli
-Version: 0.2.1
+Version: 0.2.3
 Summary: Assistant for Data Teams
 Home-page: https://github.com/AltimateAI/datapilot-cli
 Author: Altimate Inc

{altimate_datapilot_cli-0.2.1 → altimate_datapilot_cli-0.2.3}/docs/conf.py

@@ -15,7 +15,7 @@ project = "datapilot-cli"
 year = "2024"
 author = "Altimate Inc."
 copyright = f"{year}, {author}"
-version = release = "0.2.1"
+version = release = "0.2.3"
 
 pygments_style = "trac"
 templates_path = ["."]

{altimate_datapilot_cli-0.2.1 → altimate_datapilot_cli-0.2.3}/setup.py

@@ -13,7 +13,7 @@ def read(*names, **kwargs):
 
 setup(
     name="altimate-datapilot-cli",
-    version="0.2.1",
+    version="0.2.3",
     license="MIT",
     description="Assistant for Data Teams",
     long_description="{}\n{}".format(

{altimate_datapilot_cli-0.2.1 → altimate_datapilot_cli-0.2.3/src/altimate_datapilot_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: altimate-datapilot-cli
-Version: 0.2.1
+Version: 0.2.3
 Summary: Assistant for Data Teams
 Home-page: https://github.com/AltimateAI/datapilot-cli
 Author: Altimate Inc

altimate_datapilot_cli-0.2.3/src/datapilot/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.3"

{altimate_datapilot_cli-0.2.1 → altimate_datapilot_cli-0.2.3}/src/vendor/dbt_artifacts_parser/parser.py

@@ -14,6 +14,8 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 #
+import logging
+import re
 from typing import Union
 
 from vendor.dbt_artifacts_parser.parsers.catalog.catalog_v1 import CatalogV1
@@ -41,6 +43,18 @@ from vendor.dbt_artifacts_parser.parsers.sources.sources_v3 import SourcesV3
 from vendor.dbt_artifacts_parser.parsers.utils import get_dbt_schema_version
 from vendor.dbt_artifacts_parser.parsers.version_map import ArtifactTypes
 
+logger = logging.getLogger(__name__)
+
+# Fields with strict discriminated unions that break on dbt schema changes
+# but are not consumed by downstream wrappers
+_UNUSED_STRICT_FIELDS = {"disabled"}
+
+# Regex to extract manifest version number from schema URL
+_MANIFEST_VERSION_RE = re.compile(r"https://schemas\.getdbt\.com/dbt/manifest/v(\d+)\.json")
+
+# The latest manifest class we support, used as fallback for unknown versions
+_LATEST_MANIFEST_CLASS = ManifestV12
+
 
 #
 # catalog
@@ -71,6 +85,27 @@ def parse_catalog_v1(catalog: dict) -> CatalogV1:
 #
 # manifest
 #
+def _strip_unused_fields(manifest: dict) -> dict:
+    """Remove fields that have strict discriminated unions but are unused downstream.
+
+    These fields (e.g. `disabled`) use complex Pydantic unions that break when
+    dbt Cloud changes its schema, but our wrappers never read them.
+    """
+    return {k: v for k, v in manifest.items() if k not in _UNUSED_STRICT_FIELDS}
+
+
+def _try_parse_manifest(manifest: dict, model_class):
+    """Attempt to parse manifest, falling back to stripping unused fields on failure."""
+    try:
+        return model_class(**manifest)
+    except Exception:
+        stripped = _strip_unused_fields(manifest)
+        try:
+            return model_class(**stripped)
+        except Exception:
+            raise
+
+
 def parse_manifest(
     manifest: dict,
 ) -> Union[
@@ -100,31 +135,36 @@ def parse_manifest(
         ]
     """
     dbt_schema_version = get_dbt_schema_version(artifact_json=manifest)
-    if dbt_schema_version == ArtifactTypes.MANIFEST_V1.value.dbt_schema_version:
-        return ManifestV1(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V2.value.dbt_schema_version:
-        return ManifestV2(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V3.value.dbt_schema_version:
-        return ManifestV3(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V4.value.dbt_schema_version:
-        return ManifestV4(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V5.value.dbt_schema_version:
-        return ManifestV5(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V6.value.dbt_schema_version:
-        return ManifestV6(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V7.value.dbt_schema_version:
-        return ManifestV7(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V8.value.dbt_schema_version:
-        return ManifestV8(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V9.value.dbt_schema_version:
-        return ManifestV9(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V10.value.dbt_schema_version:
-        return ManifestV10(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V11.value.dbt_schema_version:
-        return ManifestV11(**manifest)
-    elif dbt_schema_version == ArtifactTypes.MANIFEST_V12.value.dbt_schema_version:
-        return ManifestV12(**manifest)
-    raise ValueError("Not a manifest.json")
+
+    version_to_class = {
+        ArtifactTypes.MANIFEST_V1.value.dbt_schema_version: ManifestV1,
+        ArtifactTypes.MANIFEST_V2.value.dbt_schema_version: ManifestV2,
+        ArtifactTypes.MANIFEST_V3.value.dbt_schema_version: ManifestV3,
+        ArtifactTypes.MANIFEST_V4.value.dbt_schema_version: ManifestV4,
+        ArtifactTypes.MANIFEST_V5.value.dbt_schema_version: ManifestV5,
+        ArtifactTypes.MANIFEST_V6.value.dbt_schema_version: ManifestV6,
+        ArtifactTypes.MANIFEST_V7.value.dbt_schema_version: ManifestV7,
+        ArtifactTypes.MANIFEST_V8.value.dbt_schema_version: ManifestV8,
+        ArtifactTypes.MANIFEST_V9.value.dbt_schema_version: ManifestV9,
+        ArtifactTypes.MANIFEST_V10.value.dbt_schema_version: ManifestV10,
+        ArtifactTypes.MANIFEST_V11.value.dbt_schema_version: ManifestV11,
+        ArtifactTypes.MANIFEST_V12.value.dbt_schema_version: ManifestV12,
+    }
+
+    model_class = version_to_class.get(dbt_schema_version)
+    if model_class:
+        return _try_parse_manifest(manifest, model_class)
+
+    # Forward-compatibility: unknown manifest version — try latest known class
+    match = _MANIFEST_VERSION_RE.match(dbt_schema_version)
+    if match:
+        logger.warning(
+            "Unknown manifest schema version %s, attempting parse with latest known class",
+            dbt_schema_version,
+        )
+        return _try_parse_manifest(manifest, _LATEST_MANIFEST_CLASS)
+
+    raise ValueError(f"Not a manifest.json (schema version: {dbt_schema_version})")
 
 
 def parse_manifest_v1(manifest: dict) -> ManifestV1:

{altimate_datapilot_cli-0.2.1 → altimate_datapilot_cli-0.2.3}/src/vendor/dbt_artifacts_parser/parsers/manifest/manifest_v12.py

@@ -288,6 +288,7 @@ class Nodes(BaseParserModel):
     unique_id: str
     fqn: list[str]
     alias: str
+    identifier: Optional[str] = None
     checksum: Checksum = Field(..., title="FileHash")
     config: Optional[Config] = Field(None, title="SeedConfig")
     tags: Optional[list[str]] = None
@@ -431,6 +432,7 @@ class Nodes1(BaseParserModel):
     unique_id: str
     fqn: list[str]
     alias: str
+    identifier: Optional[str] = None
     checksum: Checksum = Field(..., title="FileHash")
     config: Optional[Config3] = Field(None, title="NodeConfig")
     tags: Optional[list[str]] = None
@@ -538,6 +540,7 @@ class Nodes2(BaseParserModel):
     unique_id: str
     fqn: list[str]
     alias: str
+    identifier: Optional[str] = None
     checksum: Checksum = Field(..., title="FileHash")
     config: Optional[Config5] = Field(None, title="TestConfig")
     tags: Optional[list[str]] = None
@@ -674,6 +677,7 @@ class Nodes3(BaseParserModel):
     unique_id: str
     fqn: list[str]
     alias: str
+    identifier: Optional[str] = None
     checksum: Checksum = Field(..., title="FileHash")
     config: Optional[Config7] = Field(None, title="NodeConfig")
     tags: Optional[list[str]] = None
@@ -951,6 +955,7 @@ class Nodes4(BaseParserModel):
     unique_id: str
     fqn: list[str]
     alias: str
+    identifier: Optional[str] = None
     checksum: Checksum = Field(..., title="FileHash")
     config: Optional[Config9] = Field(None, title="ModelConfig")
     tags: Optional[list[str]] = None
@@ -1088,6 +1093,7 @@ class Nodes5(BaseParserModel):
     unique_id: str
     fqn: list[str]
     alias: str
+    identifier: Optional[str] = None
     checksum: Checksum = Field(..., title="FileHash")
     config: Optional[Config12] = Field(None, title="NodeConfig")
     tags: Optional[list[str]] = None
@@ -1204,6 +1210,7 @@ class Nodes6(BaseParserModel):
     unique_id: str
     fqn: list[str]
     alias: str
+    identifier: Optional[str] = None
     checksum: Checksum = Field(..., title="FileHash")
     config: Optional[Config14] = Field(None, title="TestConfig")
     tags: Optional[list[str]] = None
@@ -1421,6 +1428,7 @@ class Nodes7(BaseParserModel):
     unique_id: str
     fqn: list[str]
     alias: str
+    identifier: Optional[str] = None
     checksum: Checksum = Field(..., title="FileHash")
     config: Config16 = Field(..., title="SnapshotConfig")
     tags: Optional[list[str]] = None
@@ -2209,7 +2217,7 @@ class Disabled(BaseParserModel):
     database: Optional[str] = None
     schema_: str = Field(..., alias="schema")
     name: str
-    resource_type: Literal["seed"]
+    resource_type: Optional[Literal["seed"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -2335,7 +2343,7 @@ class Disabled1(BaseParserModel):
     database: Optional[str] = None
     schema_: str = Field(..., alias="schema")
     name: str
-    resource_type: Literal["analysis"]
+    resource_type: Optional[Literal["analysis"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -2442,7 +2450,7 @@ class Disabled2(BaseParserModel):
     database: Optional[str] = None
     schema_: str = Field(..., alias="schema")
     name: str
-    resource_type: Literal["test"]
+    resource_type: Optional[Literal["test"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -2577,7 +2585,7 @@ class Disabled3(BaseParserModel):
     database: Optional[str] = None
     schema_: str = Field(..., alias="schema")
     name: str
-    resource_type: Literal["operation"]
+    resource_type: Optional[Literal["operation"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -2828,7 +2836,7 @@ class Disabled4(BaseParserModel):
     database: Optional[str] = None
     schema_: str = Field(..., alias="schema")
     name: str
-    resource_type: Literal["model"]
+    resource_type: Optional[Literal["model"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -2965,7 +2973,7 @@ class Disabled5(BaseParserModel):
     database: Optional[str] = None
     schema_: str = Field(..., alias="schema")
     name: str
-    resource_type: Literal["sql_operation"]
+    resource_type: Optional[Literal["sql_operation"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -3072,7 +3080,7 @@ class Disabled6(BaseParserModel):
     database: Optional[str] = None
     schema_: str = Field(..., alias="schema")
     name: str
-    resource_type: Literal["test"]
+    resource_type: Optional[Literal["test"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -3277,7 +3285,7 @@ class Disabled7(BaseParserModel):
     database: Optional[str] = None
     schema_: str = Field(..., alias="schema")
     name: str
-    resource_type: Literal["snapshot"]
+    resource_type: Optional[Literal["snapshot"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -3435,7 +3443,7 @@ class Disabled8(BaseParserModel):
     database: Optional[str] = None
     schema_: str = Field(..., alias="schema")
     name: str
-    resource_type: Literal["source"]
+    resource_type: Optional[Literal["source"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -3488,7 +3496,7 @@ class Disabled9(BaseParserModel):
         extra="allow",
     )
     name: str
-    resource_type: Literal["exposure"]
+    resource_type: Optional[Literal["exposure"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -3712,7 +3720,7 @@ class Disabled10(BaseParserModel):
         extra="allow",
     )
     name: str
-    resource_type: Literal["metric"]
+    resource_type: Optional[Literal["metric"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -3812,7 +3820,7 @@ class Disabled11(BaseParserModel):
         extra="allow",
     )
     name: str
-    resource_type: Literal["saved_query"]
+    resource_type: Optional[Literal["saved_query"]] = None
     package_name: str
     path: str
     original_file_path: str
@@ -4015,7 +4023,7 @@ class Disabled12(BaseParserModel):
         extra="allow",
     )
     name: str
-    resource_type: ResourceType
+    resource_type: Optional[ResourceType] = None
     package_name: str
     path: str
     original_file_path: str
@@ -4099,7 +4107,7 @@ class Disabled13(BaseParserModel):
     given: list[GivenItem]
     expect: Expect = Field(..., title="UnitTestOutputFixture")
     name: str
-    resource_type: ResourceType
+    resource_type: Optional[ResourceType] = None
     package_name: str
     path: str
     original_file_path: str
@@ -4109,7 +4117,7 @@ class Disabled13(BaseParserModel):
     overrides: Optional[Overrides] = None
     depends_on: Optional[DependsOn12] = Field(None, title="DependsOn")
     config: Optional[Config53] = Field(None, title="UnitTestConfig")
-    checksum: Optional[str] = None
+    checksum: Optional[Union[str, Checksum]] = None
     schema_: Optional[str] = Field(None, alias="schema")
     created_at: Optional[float] = None
     versions: Optional[Versions] = None
@@ -4381,7 +4389,7 @@ class UnitTests(BaseParserModel):
     overrides: Optional[Overrides] = None
     depends_on: Optional[DependsOn12] = Field(None, title="DependsOn")
     config: Optional[Config60] = Field(None, title="UnitTestConfig")
-    checksum: Optional[str] = None
+    checksum: Optional[Union[str, Checksum]] = None
     schema_: Optional[str] = Field(None, alias="schema")
     created_at: Optional[float] = None
     versions: Optional[Versions] = None

altimate_datapilot_cli-0.2.1/src/datapilot/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.2.1"