dbt-firebolt 1.7.0__tar.gz → 1.8.1__tar.gz

{dbt_firebolt-1.7.0 → dbt_firebolt-1.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbt_firebolt
-Version: 1.7.0
+Version: 1.8.1
 Summary: The Firebolt adapter plugin for dbt (data build tool)
 Home-page: https://github.com/firebolt-db/dbt-firebolt
 Author: Firebolt

{dbt_firebolt-1.7.0 → dbt_firebolt-1.8.1}/dbt/adapters/firebolt/__init__.py

@@ -7,7 +7,7 @@ from dbt.adapters.firebolt.connections import (
 from dbt.adapters.firebolt.impl import FireboltAdapter
 from dbt.include import firebolt
 
-__version__ = "1.7.0"
+__version__ = "1.8.1"
 
 Plugin = AdapterPlugin(
     adapter=FireboltAdapter,  # type: ignore

{dbt_firebolt-1.7.0 → dbt_firebolt-1.8.1}/dbt/adapters/firebolt/connections.py

@@ -1,19 +1,23 @@
 from contextlib import contextmanager
 from dataclasses import dataclass
 from datetime import datetime
+from multiprocessing.context import SpawnContext
 from typing import Generator, Optional, Tuple, Union
 
-import dbt.exceptions
-from dbt.adapters.base import Credentials
-from dbt.adapters.sql import SQLConnectionManager
-from dbt.contracts.connection import (
+from dbt.adapters.contracts.connection import (
     AdapterRequiredConfig,
     AdapterResponse,
     Connection,
+    Credentials,
     QueryComment,
 )
-from dbt.events import AdapterLogger  # type: ignore
-from dbt.exceptions import DbtRuntimeError
+from dbt.adapters.events.logging import AdapterLogger
+from dbt.adapters.sql.connections import SQLConnectionManager
+from dbt_common.exceptions import (
+    DbtConfigError,
+    DbtRuntimeError,
+    NotImplementedError,
+)
 from firebolt.client import DEFAULT_API_URL
 from firebolt.client.auth import Auth, ClientCredentials, UsernamePassword
 from firebolt.db import ARRAY, DECIMAL
@@ -40,18 +44,22 @@ class FireboltCredentials(Credentials):
     account_name: Optional[str] = None
     retries: int = 1
 
+    _ALIASES = {
+        'host': 'api_endpoint',
+    }
+
     def __post_init__(self) -> None:
         # If user and password are not provided, assume client_id and client_secret
         # are provided instead
         if not self.user and not self.password:
             if not self.client_id or not self.client_secret:
-                raise dbt.exceptions.DbtProfileError(
+                raise DbtConfigError(
                     'Either user and password or client_id and client_secret'
                     ' must be provided'
                 )
         else:
             if self.client_id or self.client_secret:
-                raise dbt.exceptions.DbtProfileError(
+                raise DbtConfigError(
                     'Either user and password or client_id and client_secret'
                     ' must be provided'
                 )
@@ -99,13 +107,13 @@ class FireboltConnectionManager(SQLConnectionManager):
 
     TYPE = 'firebolt'
 
-    def __init__(self, profile: AdapterRequiredConfig):
+    def __init__(self, profile: AdapterRequiredConfig, mp_context: SpawnContext):
         # Query comment in appent mode only
         # This allows clearer view of queries in query_history
         if not hasattr(profile, 'query_comment'):
             setattr(profile, 'query_comment', QueryComment())
         profile.query_comment.append = True
-        super().__init__(profile)
+        super().__init__(profile, mp_context)
 
     def __str__(self) -> str:
         return 'FireboltConnectionManager()'
@@ -181,9 +189,7 @@ class FireboltConnectionManager(SQLConnectionManager):
 
     def cancel(self, connection: Connection) -> None:
         """Cancel the last query on the given connection."""
-        raise dbt.exceptions.NotImplementedError(
-            '`cancel` is not implemented for this adapter!'
-        )
+        raise NotImplementedError('`cancel` is not implemented for this adapter!')
 
     @classmethod
     def data_type_code_to_name(  # type: ignore[override] # FIR-29423

{dbt_firebolt-1.7.0 → dbt_firebolt-1.8.1}/dbt/adapters/firebolt/impl.py

@@ -5,9 +5,8 @@ from datetime import datetime
 from typing import Any, List, Mapping, Optional, Union
 
 import agate
-from dbt.adapters.base import available  # type: ignore
-from dbt.adapters.base.impl import AdapterConfig  # type: ignore
 from dbt.adapters.base.impl import ConstraintSupport
+from dbt.adapters.base.meta import available
 from dbt.adapters.base.relation import BaseRelation
 from dbt.adapters.capability import (
     Capability,
@@ -15,10 +14,11 @@ from dbt.adapters.capability import (
     CapabilitySupport,
     Support,
 )
-from dbt.adapters.sql import SQLAdapter  # type: ignore
-from dbt.contracts.graph.nodes import ConstraintType
-from dbt.dataclass_schema import ValidationError, dbtClassMixin
-from dbt.exceptions import (
+from dbt.adapters.protocol import AdapterConfig
+from dbt.adapters.sql.impl import SQLAdapter
+from dbt_common.contracts.constraints import ConstraintType
+from dbt_common.dataclass_schema import ValidationError, dbtClassMixin
+from dbt_common.exceptions import (
     CompilationError,
     DbtRuntimeError,
     NotImplementedError,

{dbt_firebolt-1.7.0 → dbt_firebolt-1.8.1}/dbt/adapters/firebolt/relation.py

@@ -1,10 +1,10 @@
 from dataclasses import dataclass, field
 from typing import Dict, FrozenSet, Optional
 
-from dbt.adapters.base import RelationType  # type: ignore
-from dbt.adapters.base.relation import BaseRelation, Policy  # type: ignore
-from dbt.adapters.relation_configs import RelationConfigBase  # type: ignore
-from dbt.exceptions import DbtRuntimeError
+from dbt.adapters.base.relation import BaseRelation
+from dbt.adapters.contracts.relation import Policy, RelationType
+from dbt.adapters.relation_configs.config_base import RelationConfigBase
+from dbt_common.exceptions import DbtRuntimeError
 
 
 @dataclass

{dbt_firebolt-1.7.0 → dbt_firebolt-1.8.1}/dbt_firebolt.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbt-firebolt
-Version: 1.7.0
+Version: 1.8.1
 Summary: The Firebolt adapter plugin for dbt (data build tool)
 Home-page: https://github.com/firebolt-db/dbt-firebolt
 Author: Firebolt

{dbt_firebolt-1.7.0 → dbt_firebolt-1.8.1}/dbt_firebolt.egg-info/requires.txt

@@ -1,5 +1,6 @@
-dbt-core<1.8,~=1.6
-firebolt-sdk>=1.1.0
+dbt-adapters<2.0,>=1.0
+dbt-core>=1.8.0
+firebolt-sdk>=1.5.0
 pydantic>=0.23
 
 [dev]

{dbt_firebolt-1.7.0 → dbt_firebolt-1.8.1}/setup.cfg

@@ -22,8 +22,9 @@ project_urls =
 [options]
 packages = find_namespace:
 install_requires = 
-	dbt-core~=1.6,<1.8
-	firebolt-sdk>=1.1.0
+	dbt-adapters>=1.0,<2.0
+	dbt-core>=1.8.0
+	firebolt-sdk>=1.5.0
 	pydantic>=0.23
 python_requires = >=3.8
 include_package_data = True