From 2d420d59383b3fcc5dc92ffbaed86ebf2faf45ab Mon Sep 17 00:00:00 2001
From: Maha Rana Yadavalli <271375718+maharanay22@users.noreply.github.com>
Date: Tue, 13 Oct 2026 11:57:05 -0400
Subject: [PATCH] Add databricks-sqlalchemy to the Databricks handler
 requirements

databricks-sql-connector 4.x no longer ships the SQLAlchemy dialect,
it moved to the separate databricks-sqlalchemy package. The handler
still imports databricks.sqlalchemy, so after the bump to
databricks-sql-connector 4.2.3 it failed to load with
"No module named 'databricks.sqlalchemy'", and the unit tests for it
were skipped as "Databricks handler not installed".

Pin databricks-sqlalchemy 2.0.10, which supports connector 4.x and
SQLAlchemy 2, and map it to the databricks module for deptry.
---
 .../integrations/handlers/databricks_handler/requirements.txt    | 1 +
 tests/scripts/check_requirements.py                              | 1 +
 2 files changed, 2 insertions(+)

diff --git a/mindsdb/integrations/handlers/databricks_handler/requirements.txt b/mindsdb/integrations/handlers/databricks_handler/requirements.txt
index 0137133cc5..a375e8fce2 100644
--- a/mindsdb/integrations/handlers/databricks_handler/requirements.txt
+++ b/mindsdb/integrations/handlers/databricks_handler/requirements.txt
@@ -1 +1,2 @@
 databricks-sql-connector==4.2.3
+databricks-sqlalchemy==2.0.10
diff --git a/tests/scripts/check_requirements.py b/tests/scripts/check_requirements.py
index f3ec6de303..8192060d27 100644
--- a/tests/scripts/check_requirements.py
+++ b/tests/scripts/check_requirements.py
@@ -210,6 +210,7 @@ def get_requirements_with_DEP002(path):
     "pysqlite3": ["pysqlite3"],
     "atlassian-python-api": ["atlassian"],
     "databricks-sql-connector": ["databricks"],
+    "databricks-sqlalchemy": ["databricks"],
     "elasticsearch-dbapi": ["es"],
     "pygithub": ["github"],
     "python-gitlab": ["gitlab"],