add cli version to ingestion headers

datahub-project · Nov 13, 2024 · be77247 · be77247
1 parent 470c858
commit be77247
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/metadata-ingestion/src/datahub/emitter/rest_emitter.py b/metadata-ingestion/src/datahub/emitter/rest_emitter.py
@@ -10,6 +10,7 @@
 from requests.adapters import HTTPAdapter, Retry
 from requests.exceptions import HTTPError, RequestException
 
+from datahub import nice_version_name
 from datahub.cli import config_utils
 from datahub.cli.cli_utils import ensure_has_system_metadata, fixup_gms_url
 from datahub.configuration.common import ConfigurationError, OperationalError
@@ -91,6 +92,7 @@ def __init__(
         self._session.headers.update(
             {
                 "X-RestLi-Protocol-Version": "2.0.0",
+                "X-Datahub-Cli-Version": nice_version_name(),
                 "Content-Type": "application/json",
             }
         )