diff --git a/providers/databricks/src/airflow/providers/databricks/hooks/databricks_sql.py b/providers/databricks/src/airflow/providers/databricks/hooks/databricks_sql.py index b13100330d32f..24be7acc92733 100644 --- a/providers/databricks/src/airflow/providers/databricks/hooks/databricks_sql.py +++ b/providers/databricks/src/airflow/providers/databricks/hooks/databricks_sql.py @@ -241,7 +241,7 @@ def get_conn(self) -> AirflowConnection: "catalog": self.catalog, "session_configuration": session_config or None, "http_headers": self.http_headers, - "_user_agent_entry": self.user_agent_value, + "user_agent_entry": self.user_agent_value, **self._get_extra_config(), **self.additional_params, }