WIP

Author: z3z1ma

src/dbt_core_interface/project.py

@@ -35,8 +35,11 @@
 from dbt.parser.sql import SqlBlockParser, SqlMacroParser
 from dbt.task.sql import SqlCompileRunner
 from dbt.tracking import disable_tracking
+from dbt_common.clients.system import get_env
+from dbt_common.context import set_invocation_context
 
 disable_tracking()
+set_invocation_context(get_env())
 
 __all__ = ["DbtProject", "DbtConfiguration", "ExecutionResult", "CompilationResult"]
 
@@ -198,17 +201,18 @@ def _initialize_adapter(self) -> None:
             raise RuntimeError("Runtime config not available")
 
         adapter_cls = get_adapter_class_by_name(self._runtime_config.credentials.type)
-        self._adapter = t.cast(BaseAdapter, adapter_cls(self._runtime_config, get_mp_context()))  # pyright: ignore[reportInvalidCast]
+        try:
+            self._adapter = t.cast(BaseAdapter, adapter_cls(self._runtime_config, get_mp_context()))  # pyright: ignore[reportInvalidCast,reportCallIssue]
+        except ValueError:
+            self._adapter = t.cast(BaseAdapter, adapter_cls(self._runtime_config))  # pyright: ignore[reportInvalidCast]
 
-        # Verify connection
         with self._adapter.connection_named("master"):
             self._adapter.debug_query()
 
         logger.debug(f"Initialized adapter for {self.project_name}")
 
     def _parse_project(self) -> None:
         """Parse the dbt project and load manifest."""
-        # Set dbt flags
         set_from_args(
             argparse.Namespace(
                 target=self.config.target,
@@ -223,7 +227,6 @@ def _parse_project(self) -> None:
         )
 
         with self._manifest_lock:
-            # Load runtime config
             self._runtime_config = RuntimeConfig.from_args(
                 argparse.Namespace(
                     target=self.config.target,
@@ -234,13 +237,12 @@ def _parse_project(self) -> None:
                 )
             )
 
-            # Initialize adapter
             self._initialize_adapter()
 
-            # Load manifest
             manifest_loader = ManifestLoader(
                 self._runtime_config,
-                self._runtime_config.dependencies or {},
+                self._runtime_config.load_dependencies(),
+                self.adapter.connections.set_query_header,  # pyright: ignore[reportArgumentType]
             )
             self._manifest = manifest_loader.load()