fix dbt support

Author: izeigerman

sqlmesh/dbt/basemodel.py

@@ -10,6 +10,7 @@
 
 from sqlmesh.core import dialect as d
 from sqlmesh.core.config.base import UpdateStrategy
+from sqlmesh.core.config.common import VirtualEnvironmentMode
 from sqlmesh.core.model import Model
 from sqlmesh.dbt.column import (
     ColumnConfig,
@@ -30,7 +31,6 @@
 from sqlmesh.utils.pydantic import field_validator
 
 if t.TYPE_CHECKING:
-    from sqlmesh.core.config.common import VirtualEnvironmentMode
     from sqlmesh.core.audit.definition import ModelAudit
     from sqlmesh.dbt.context import DbtContext
 

sqlmesh/dbt/loader.py

@@ -117,7 +117,11 @@ def _load_models(
 
         def _to_sqlmesh(config: BMC, context: DbtContext) -> Model:
             logger.debug("Converting '%s' to sqlmesh format", config.canonical_name(context))
-            return config.to_sqlmesh(context, audit_definitions=audits)
+            return config.to_sqlmesh(
+                context,
+                audit_definitions=audits,
+                virtual_environment_mode=self.config.virtual_environment_mode,
+            )
 
         for project in self._load_projects():
             context = project.context.copy()

sqlmesh/dbt/model.py

@@ -8,6 +8,7 @@
 
 from sqlmesh.core import dialect as d
 from sqlmesh.core.config.base import UpdateStrategy
+from sqlmesh.core.config.common import VirtualEnvironmentMode
 from sqlmesh.core.console import get_console
 from sqlmesh.core.model import (
     EmbeddedKind,
@@ -29,7 +30,6 @@
 from sqlmesh.utils.pydantic import field_validator
 
 if t.TYPE_CHECKING:
-    from sqlmesh.core.config.common import VirtualEnvironmentMode
     from sqlmesh.core.audit.definition import ModelAudit
     from sqlmesh.dbt.context import DbtContext
 

sqlmesh/dbt/seed.py

@@ -15,12 +15,12 @@
     SUPPORTS_DELIMITER = False
 from sqlglot import exp
 
+from sqlmesh.core.config.common import VirtualEnvironmentMode
 from sqlmesh.core.model import Model, SeedKind, create_seed_model
 from sqlmesh.dbt.basemodel import BaseModelConfig
 from sqlmesh.dbt.column import ColumnConfig
 
 if t.TYPE_CHECKING:
-    from sqlmesh.core.config.common import VirtualEnvironmentMode
     from sqlmesh.core.audit.definition import ModelAudit
     from sqlmesh.dbt.context import DbtContext