feat: Add databricks_lakebase to DatabaseType enum

Author: jiatolentino

HISTORY.rst

@@ -2,6 +2,11 @@
 History
 =======
 
+1.0.1 (2026-05-11)
+------------------
+
+* Added ``databricks_lakebase`` to ``DatabaseType`` enum.
+
 1.0.0 (2026-04-21)
 ------------------
 

datamasque/client/models/connection.py

@@ -44,6 +44,7 @@ class DatabaseType(Enum):
     mssql_linked = "mssql_linked"
     snowflake = "snowflake"
     mongodb = "mongodb"
+    databricks_lakebase = "databricks_lakebase"
 
 
 class SnowflakeStageLocation(str, Enum):

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "datamasque-python"
-version = "1.0.0"
+version = "1.0.1"
 description = "Official Python client for the DataMasque data-masking API."
 authors = [
     { name = "DataMasque Ltd" },

setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 1.0.0
+current_version = 1.0.1
 commit = True
 tag = True
 

tests/test_connections.py

@@ -768,6 +768,27 @@ def test_database_connection_model_validate_keeps_schema_for_postgres():
     assert conn.db_schema == "public"
 
 
+def test_database_connection_model_validate_databricks_lakebase():
+    payload = {
+        "id": "abc-lakebase",
+        "name": "lakebase",
+        "mask_type": "database",
+        "db_type": "databricks_lakebase",
+        "host": "lakebase-host",
+        "port": 5432,
+        "database": "lakebasedb",
+        "user": "lakebase_user",
+        "schema": "public",
+        "is_read_only": False,
+    }
+
+    conn = DatabaseConnectionConfig.model_validate(payload)
+
+    assert isinstance(conn, DatabaseConnectionConfig)
+    assert conn.database_type is DatabaseType.databricks_lakebase
+    assert conn.db_schema == "public"
+
+
 def test_mssql_linked_connection_model_validate_includes_linked_server():
     payload = {
         "id": "48a7af45-f63f-4e05-bf9f-7b1cc3a0e89d",