fix(ingest/nifi): add error handling for version (datahub-project#11385)

acryldata · Sep 16, 2024 · d82e478 · d82e478
1 parent dc35465
commit d82e478
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/nifi.py b/metadata-ingestion/src/datahub/ingestion/source/nifi.py
@@ -638,7 +638,12 @@ def create_nifi_flow(self):
         )
         nifi_version: Optional[str] = None
         if about_response.ok:
-            nifi_version = about_response.json().get("about", {}).get("version")
+            try:
+                nifi_version = about_response.json().get("about", {}).get("version")
+            except Exception as e:
+                logger.error(
+                    f"Unable to parse about response from Nifi: {about_response} due to {e}"
+                )
         else:
             logger.warning("Failed to fetch version for nifi")
         cluster_response = self.session.get(