fix(ingest): gracefully handle missing system metadata in client (dat…

…ahub-project#11592)
sleeperdeep · Dec 17, 2024 · 991ead3 · 991ead3
1 parent d9523a9
commit 991ead3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/graph/client.py b/metadata-ingestion/src/datahub/ingestion/graph/client.py
@@ -559,8 +559,10 @@ def get_entity_as_mcps(
             post_json_obj = post_json_transform(aspect_json)
             aspect_value = aspect_type.from_obj(post_json_obj["value"])
 
-            system_metadata_raw = post_json_obj["systemMetadata"]
-            system_metadata = SystemMetadataClass.from_obj(system_metadata_raw)
+            system_metadata_raw = post_json_obj.get("systemMetadata")
+            system_metadata = None
+            if system_metadata_raw:
+                system_metadata = SystemMetadataClass.from_obj(system_metadata_raw)
 
             mcpw = MetadataChangeProposalWrapper(
                 entityUrn=entity_urn,
@@ -590,7 +592,7 @@ def get_entity_semityped(
             not be present in the dictionary. The entity's key aspect will always be present.
         """
 
-        mcps = self.get_entity_as_mcps(entity_urn, aspects)
+        mcps = self.get_entity_as_mcps(entity_urn, aspects=aspects)
 
         result: AspectBag = {}
         for mcp in mcps: