diff --git a/sqlmesh/dbt/manifest.py b/sqlmesh/dbt/manifest.py index 26d30a76d..0bac148a9 100644 --- a/sqlmesh/dbt/manifest.py +++ b/sqlmesh/dbt/manifest.py @@ -2,6 +2,7 @@ import json import logging +import os import re import typing as t from argparse import Namespace @@ -260,6 +261,11 @@ def _load_manifest(self) -> Manifest: ) flags.set_from_args(args, None) + if DBT_VERSION >= (1, 8): + from dbt_common.context import set_invocation_context # type: ignore + + set_invocation_context(os.environ) + profile = self._load_profile() project = self._load_project(profile) @@ -274,10 +280,8 @@ def _load_manifest(self) -> Manifest: if DBT_VERSION >= (1, 8): from dbt.mp_context import get_mp_context # type: ignore - from dbt_common.context import set_invocation_context # type: ignore register_adapter(runtime_config, get_mp_context()) # type: ignore - set_invocation_context({}) else: register_adapter(runtime_config) # type: ignore