chore: add user agent to Databricks requests (#20660)

superset/db_engine_specs/databricks.py

@@ -16,11 +16,15 @@
 # under the License.
 
 from datetime import datetime
-from typing import Any, Dict, Optional
+from typing import Any, Dict, Optional, TYPE_CHECKING
 
+from superset.constants import USER_AGENT
 from superset.db_engine_specs.base import BaseEngineSpec
 from superset.db_engine_specs.hive import HiveEngineSpec
 
+if TYPE_CHECKING:
+    from superset.models.core import Database
+
 time_grain_expressions = {
     None: "{col}",
     "PT1S": "date_trunc('second', {col})",
@@ -71,3 +75,15 @@ class DatabricksNativeEngineSpec(DatabricksODBCEngineSpec):
     engine = "databricks"
     engine_name = "Databricks Native Connector"
     driver = "connector"
+
+    @staticmethod
+    def get_extra_params(database: "Database") -> Dict[str, Any]:
+        """
+        Add a user agent to be used in the requests.
+        """
+        extra = {
+            "http_headers": [("User-Agent", USER_AGENT)],
+            "_user_agent_entry": USER_AGENT,
+        }
+        extra.update(BaseEngineSpec.get_extra_params(database))
+        return extra