Fix: Support custom dbt project loaders (#3920)

Author: izeigerman

sqlmesh/dbt/loader.py

@@ -56,8 +56,12 @@ def sqlmesh_config(
     if register_comments is not None:
         target_to_sqlmesh_args["register_comments"] = register_comments
 
+    loader = kwargs.pop("loader", DbtLoader)
+    if not issubclass(loader, DbtLoader):
+        raise ConfigError("The loader must be a DbtLoader.")
+
     return Config(
-        loader=DbtLoader,
+        loader=loader,
         model_defaults=model_defaults,
         variables=variables or {},
         **{

tests/dbt/test_config.py

@@ -901,6 +901,20 @@ def test_connection_args(tmp_path):
     assert not config.gateways["in_memory"].connection.register_comments
 
 
+def test_custom_dbt_loader():
+    from sqlmesh.core.loader import SqlMeshLoader
+    from sqlmesh.dbt.loader import DbtLoader
+
+    dbt_project_dir = "tests/fixtures/dbt/sushi_test"
+    with pytest.raises(ConfigError, match="The loader must be a DbtLoader."):
+        sqlmesh_config(dbt_project_dir, loader=SqlMeshLoader)
+
+    class CustomDbtLoader(DbtLoader):
+        pass
+
+    sqlmesh_config(dbt_project_dir, loader=CustomDbtLoader)
+
+
 @pytest.mark.cicdonly
 def test_db_type_to_relation_class():
     from dbt.adapters.bigquery.relation import BigQueryRelation