expose OpenLineage's lineage_root_* macros in plugin (#50532)

mobuchowski · web-flow · commit a500cafafe86 · 2025-05-19T15:16:32.000-04:00
Signed-off-by: Maciej Obuchowski &lt;maciej.obuchowski@datadoghq.com&gt;
diff --git a/providers/openlineage/src/airflow/providers/openlineage/plugins/macros.py b/providers/openlineage/src/airflow/providers/openlineage/plugins/macros.py
@@ -90,6 +90,16 @@ def lineage_parent_id(task_instance: TaskInstance):
 
 
 def lineage_root_parent_id(task_instance: TaskInstance):
+    """
+    Macro function which returns a unique identifier of given task that can be used to create root information for ParentRunFacet.
+
+    This identifier is composed of the namespace, dag name, and generated run id for given dag, structured
+    as '{namespace}/{job_name}/{run_id}'.
+
+    .. seealso::
+        For more information take a look at the guide:
+        :ref:`howto/macros:openlineage`
+    """
     return "/".join(
         (
             lineage_job_namespace(),
diff --git a/providers/openlineage/src/airflow/providers/openlineage/plugins/openlineage.py b/providers/openlineage/src/airflow/providers/openlineage/plugins/openlineage.py
@@ -23,6 +23,9 @@
     lineage_job_name,
     lineage_job_namespace,
     lineage_parent_id,
+    lineage_root_job_name,
+    lineage_root_parent_id,
+    lineage_root_run_id,
     lineage_run_id,
 )
 
@@ -37,7 +40,15 @@ class OpenLineageProviderPlugin(AirflowPlugin):
 
     name = "OpenLineageProviderPlugin"
     if not conf.is_disabled():
-        macros = [lineage_job_namespace, lineage_job_name, lineage_run_id, lineage_parent_id]
+        macros = [
+            lineage_job_namespace,
+            lineage_job_name,
+            lineage_run_id,
+            lineage_parent_id,
+            lineage_root_run_id,
+            lineage_root_job_name,
+            lineage_root_parent_id,
+        ]
         listeners = [get_openlineage_listener()]
         from airflow.lineage.hook import HookLineageReader
 
