WIP

Author: gaurpulkit

src/datapilot/clients/altimate/constants.py

@@ -6,4 +6,5 @@
     "catalog",
     "run_results",
     "sources",
+    "semantic_manifest",
 }

src/datapilot/core/platforms/dbt/cli/cli.py

@@ -159,6 +159,7 @@ def project_health(
 @click.option("--catalog-path", required=False, prompt=False, help="Path to the catalog file.")
 @click.option("--run-results-path", required=False, prompt=False, help="Path to the run_results.json file.")
 @click.option("--sources-path", required=False, prompt=False, help="Path to the sources.json file (source freshness results).")
+@click.option("--semantic-manifest-path", required=False, prompt=False, help="Path to the semantic_manifest.json file.")
 def onboard(
     token,
     instance_name,
@@ -170,6 +171,7 @@ def onboard(
     catalog_path,
     run_results_path,
     sources_path,
+    semantic_manifest_path,
 ):
     """Onboard a manifest file to DBT. You can specify either --dbt_integration_id or --dbt_integration_name."""
 
@@ -270,6 +272,16 @@ def onboard(
         else:
             click.echo(f"{response['message']}")
 
+    if semantic_manifest_path:
+        response = onboard_file(
+            token, instance_name, dbt_integration_id, dbt_integration_environment, "semantic_manifest", semantic_manifest_path, backend_url
+        )
+        if response["ok"]:
+            click.echo("Semantic manifest onboarded successfully!")
+            artifacts_uploaded.append("semantic_manifest")
+        else:
+            click.echo(f"{response['message']}")
+
     # Start ingestion
     response = start_dbt_ingestion(token, instance_name, dbt_integration_id, dbt_integration_environment, backend_url)
     if response["ok"]:

tests/clients/altimate/test_utils.py

@@ -5,7 +5,7 @@
 class TestOnboardFile:
     def test_supported_artifact_types(self):
         """Test that all expected artifact types are supported."""
-        expected_types = {"manifest", "catalog", "run_results", "sources"}
+        expected_types = {"manifest", "catalog", "run_results", "sources", "semantic_manifest"}
         assert SUPPORTED_ARTIFACT_TYPES == expected_types
 
     def test_unsupported_file_type_returns_error(self):

tests/core/platform/dbt/test_cli.py

@@ -126,3 +126,4 @@ def test_onboard_help_shows_all_artifact_options():
     assert "--catalog-path" in result.output
     assert "--run-results-path" in result.output
     assert "--sources-path" in result.output
+    assert "--semantic-manifest-path" in result.output