diff --git a/data_diff/__main__.py b/data_diff/__main__.py
index dfd12577..82334932 100644
--- a/data_diff/__main__.py
+++ b/data_diff/__main__.py
@@ -33,7 +33,7 @@
     "-": "red",
 }
 
-set_entrypoint_name("CLI")
+set_entrypoint_name(os.getenv("DATAFOLD_TRIGGERED_BY", "CLI"))
 
 
 def _get_log_handlers(is_dbt: Optional[bool] = False) -> Dict[str, logging.Handler]:
diff --git a/data_diff/dbt.py b/data_diff/dbt.py
index fd3223de..fdebb532 100644
--- a/data_diff/dbt.py
+++ b/data_diff/dbt.py
@@ -74,7 +74,7 @@ def dbt_diff(
 ) -> None:
     print_version_info()
     diff_threads = []
-    set_entrypoint_name("CLI-dbt")
+    set_entrypoint_name(os.getenv("DATAFOLD_TRIGGERED_BY", "CLI-dbt"))
     dbt_parser = DbtParser(profiles_dir_override, project_dir_override, state)
     models = dbt_parser.get_models(dbt_selection)
     config = dbt_parser.get_datadiff_config()