dbt-adapters 1.16.5__tar.gz → 1.16.6__tar.gz

dbt_adapters-1.16.6/.changes/1.16.6.md

@@ -0,0 +1,9 @@
+## dbt-adapters 1.16.6 - September 03, 2025
+
+### Under the Hood
+
+- Record adapter.add_query as Available record ([#1285](https://github.com/dbt-labs/dbt-adapters/issues/1285))
+- Record adapter.list_relations_without_caching as Available record ([#1308](https://github.com/dbt-labs/dbt-adapters/issues/1308))
+
+### Contributors
+- [@jasonlin45](https://github.com/jasonlin45) ([#1308](https://github.com/dbt-labs/dbt-adapters/issues/1308))

{dbt_adapters-1.16.5 → dbt_adapters-1.16.6}/CHANGELOG.md

@@ -10,14 +10,23 @@ and is generated by [Changie](https://github.com/miniscruff/changie)
 - Do not edit this file directly. This file is auto-generated using [changie](https://github.com/miniscruff/changie)
 - For details on how to document a change, see the [contributing guide](/CONTRIBUTING.md#changelog-entry)
 
+## dbt-adapters 1.16.6 - September 03, 2025
+
+### Under the Hood
+
+- Record adapter.add_query as Available record ([#1285](https://github.com/dbt-labs/dbt-adapters/issues/1285))
+- Record adapter.list_relations_without_caching as Available record ([#1308](https://github.com/dbt-labs/dbt-adapters/issues/1308))
+
+### Contributors
+- [@jasonlin45](https://github.com/jasonlin45) ([#1308](https://github.com/dbt-labs/dbt-adapters/issues/1308))
+
+
 ## dbt-adapters 1.16.5 - August 19, 2025
 
 ### Fixes
 
 - Correct column mapping lookup in unit test macro ([#1273](https://github.com/dbt-labs/dbt-adapters/issues/1273))
 
-
-
 ## dbt-adapters 1.16.4 - August 15, 2025
 
 ### Fixes

{dbt_adapters-1.16.5 → dbt_adapters-1.16.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbt-adapters
-Version: 1.16.5
+Version: 1.16.6
 Summary: The set of adapter protocols and base functionality that supports integration with dbt-core
 Project-URL: Homepage, https://github.com/dbt-labs/dbt-adapters/tree/main/dbt-adapters
 Project-URL: Documentation, https://docs.getdbt.com

dbt_adapters-1.16.6/src/dbt/adapters/__about__.py

@@ -0,0 +1 @@
+version = "1.16.6"

{dbt_adapters-1.16.5 → dbt_adapters-1.16.6}/src/dbt/adapters/base/impl.py

@@ -30,6 +30,7 @@ from dbt.adapters.record.base import (
     AdapterGetPartitionsMetadataRecord,
     AdapterConvertTypeRecord,
     AdapterStandardizeGrantsDictRecord,
+    AdapterListRelationsWithoutCachingRecord,
 )
 from dbt_common.behavior_flags import Behavior, BehaviorFlag
 from dbt_common.clients.jinja import CallableMacroGenerator
@@ -772,6 +773,12 @@ class BaseAdapter(metaclass=AdapterMeta):
             "`expand_target_column_types` is not implemented for this adapter!"
         )
 
+    @record_function(
+        AdapterListRelationsWithoutCachingRecord,
+        method=True,
+        index_on_thread_id=True,
+        id_field_name="thread_id",
+    )
     @abc.abstractmethod
     def list_relations_without_caching(self, schema_relation: BaseRelation) -> List[BaseRelation]:
         """List relations in the given schema, bypassing the cache.

{dbt_adapters-1.16.5 → dbt_adapters-1.16.6}/src/dbt/adapters/catalogs/__init__.py

@@ -3,6 +3,7 @@ from dbt.adapters.catalogs._exceptions import (
     DbtCatalogIntegrationAlreadyExistsError,
     DbtCatalogIntegrationNotFoundError,
     DbtCatalogIntegrationNotSupportedError,
+    InvalidCatalogIntegrationConfigError,
 )
 from dbt.adapters.catalogs._integration import (
     CatalogIntegration,

{dbt_adapters-1.16.5 → dbt_adapters-1.16.6}/src/dbt/adapters/catalogs/_exceptions.py

@@ -30,3 +30,10 @@ class DbtCatalogIntegrationNotSupportedError(DbtConfigError):
             f"Expected one of: {', '.join(supported_catalog_types)}"
         )
         super().__init__(msg)
+
+
+class InvalidCatalogIntegrationConfigError(DbtConfigError):
+    def __init__(self, catalog_name: str, msg: str) -> None:
+        self.catalog_name = catalog_name
+        msg = f"Invalid catalog integration config: {self.catalog_name}. {msg}"
+        super().__init__(msg)

{dbt_adapters-1.16.5 → dbt_adapters-1.16.6}/src/dbt/adapters/exceptions/compilation.py

@@ -255,3 +255,15 @@ class RelationWrongTypeError(CompilationError):
         )
 
         return msg
+
+
+class InvalidRelationConfigError(CompilationError):
+    def __init__(self, relation, config, msg):
+        self.relation = relation
+        self.config = config
+        self.msg = msg
+        super().__init__(msg=self.get_message())
+
+    def get_message(self) -> str:
+        msg = f"Invalid relation config: {self.config}"
+        return msg

{dbt_adapters-1.16.5 → dbt_adapters-1.16.6}/src/dbt/adapters/record/base.py

@@ -2,14 +2,15 @@
 
 import dataclasses
 
-from typing import Optional, Tuple, Dict, Any, TYPE_CHECKING
+from typing import Optional, Tuple, Dict, Any, TYPE_CHECKING, List
 
 from dbt.adapters.contracts.connection import AdapterResponse
-from dbt.adapters.record.serialization import serialize_agate_table
+from dbt.adapters.record.serialization import serialize_agate_table, serialize_bindings
 from dbt_common.record import Record, Recorder
 
 if TYPE_CHECKING:
     from agate import Table
+    from dbt.adapters.base.relation import BaseRelation
 
 
 @dataclasses.dataclass
@@ -161,3 +162,85 @@ class AdapterStandardizeGrantsDictRecord(Record):
     params_cls = AdapterStandardizeGrantsDictParams
     result_cls = AdapterStandardizeGrantsDictResult
     group = "Available"
+
+
+@dataclasses.dataclass
+class AdapterAddQueryParams:
+    thread_id: str
+    sql: str
+    auto_begin: bool = True
+    bindings: Optional[Any] = None
+    abridge_sql_log: bool = False
+
+    def _to_dict(self):
+        return {
+            "thread_id": self.thread_id,
+            "sql": self.sql,
+            "auto_begin": self.auto_begin,
+            "bindings": serialize_bindings(self.bindings),
+            "abridge_sql_log": self.abridge_sql_log,
+        }
+
+
+@dataclasses.dataclass
+class AdapterAddQueryResult:
+    return_val: Tuple[str, str]
+
+    def _to_dict(self):
+        return {
+            "return_val": {
+                "conn": "conn",
+                "cursor": "cursor",
+            }
+        }
+
+
+@Recorder.register_record_type
+class AdapterAddQueryRecord(Record):
+    params_cls = AdapterAddQueryParams
+    result_cls = AdapterAddQueryResult
+    group = "Available"
+
+
+@dataclasses.dataclass
+class AdapterListRelationsWithoutCachingParams:
+    thread_id: str
+    schema_relation: "BaseRelation"
+
+    def _to_dict(self):
+        from dbt.adapters.record.serialization import serialize_base_relation
+
+        return {
+            "thread_id": self.thread_id,
+            "schema_relation": serialize_base_relation(self.schema_relation),
+        }
+
+    def _from_dict(self, data: Dict[str, Any]):
+        from dbt.adapters.record.serialization import deserialize_base_relation
+
+        self.thread_id = data["thread_id"]
+        self.schema_relation = deserialize_base_relation(data["schema_relation"])
+
+
+@dataclasses.dataclass
+class AdapterListRelationsWithoutCachingResult:
+    return_val: List["BaseRelation"]
+
+    def _to_dict(self):
+        from dbt.adapters.record.serialization import serialize_base_relation_list
+
+        return {"return_val": serialize_base_relation_list(self.return_val)}
+
+    def _from_dict(self, data: Dict[str, Any]):
+        from dbt.adapters.record.serialization import deserialize_base_relation_list
+
+        self.return_val = deserialize_base_relation_list(data["return_val"])
+
+
+@Recorder.register_record_type
+class AdapterListRelationsWithoutCachingRecord(Record):
+    """Implements record/replay support for the BaseAdapter.list_relations_without_caching() method."""
+
+    params_cls = AdapterListRelationsWithoutCachingParams
+    result_cls = AdapterListRelationsWithoutCachingResult
+    group = "Available"

dbt_adapters-1.16.6/src/dbt/adapters/record/serialization.py

@@ -0,0 +1,63 @@
+from datetime import datetime, date
+from decimal import Decimal
+from typing import Any, Dict, TYPE_CHECKING, List, Union
+
+if TYPE_CHECKING:
+    from agate import Table
+    from dbt.adapters.base.relation import BaseRelation
+
+
+def _column_filter(val: Any) -> Any:
+    return (
+        float(val)
+        if isinstance(val, Decimal)
+        else (
+            str(val)
+            if isinstance(val, datetime)
+            else str(val) if isinstance(val, date) else str(val)
+        )
+    )
+
+
+def serialize_agate_table(table: "Table") -> Dict[str, Any]:
+    rows = []
+    for row in table.rows:
+        row = list(map(_column_filter, row))
+        rows.append(row)
+
+    return {
+        "column_names": table.column_names,
+        "column_types": [t.__class__.__name__ for t in table.column_types],
+        "rows": rows,
+    }
+
+
+def serialize_bindings(bindings: Any) -> Union[None, List[Any], str]:
+    if bindings is None:
+        return None
+    elif isinstance(bindings, list):
+        return list(map(_column_filter, bindings))
+    else:
+        return "bindings"
+
+
+def serialize_base_relation(relation: "BaseRelation") -> Dict[str, Any]:
+    """Serialize a BaseRelation object for recording."""
+    return relation.to_dict(omit_none=True)
+
+
+def serialize_base_relation_list(relations: List["BaseRelation"]) -> List[Dict[str, Any]]:
+    """Serialize a list of BaseRelation objects for recording."""
+    return [serialize_base_relation(relation) for relation in relations]
+
+
+def deserialize_base_relation(relation_dict: Dict[str, Any]) -> "BaseRelation":
+    """Deserialize a BaseRelation object from a dictionary."""
+    from dbt.adapters.base.relation import BaseRelation
+
+    return BaseRelation.from_dict(relation_dict)
+
+
+def deserialize_base_relation_list(relations_data: List[Dict[str, Any]]) -> List["BaseRelation"]:
+    """Deserialize a list of BaseRelation objects from dictionaries."""
+    return [deserialize_base_relation(relation_dict) for relation_dict in relations_data]

{dbt_adapters-1.16.5 → dbt_adapters-1.16.6}/src/dbt/adapters/sql/impl.py

@@ -8,7 +8,7 @@ from dbt.adapters.cache import _make_ref_key_dict
 from dbt.adapters.contracts.connection import AdapterResponse, Connection
 from dbt.adapters.events.types import ColTypeChange, SchemaCreation, SchemaDrop
 from dbt.adapters.exceptions import RelationTypeNullError
-from dbt.adapters.record.base import AdapterTestSqlRecord
+from dbt.adapters.record.base import AdapterTestSqlRecord, AdapterAddQueryRecord
 from dbt.adapters.sql.connections import SQLConnectionManager
 
 LIST_RELATIONS_MACRO_NAME = "list_relations_without_caching"
@@ -49,6 +49,9 @@ class SQLAdapter(BaseAdapter):
     connections: SQLConnectionManager
 
     @available.parse(lambda *a, **k: (None, None))
+    @record_function(
+        AdapterAddQueryRecord, method=True, index_on_thread_id=True, id_field_name="thread_id"
+    )
     def add_query(
         self,
         sql: str,

dbt_adapters-1.16.5/src/dbt/adapters/__about__.py

@@ -1 +0,0 @@
-version = "1.16.5"

dbt_adapters-1.16.5/src/dbt/adapters/record/serialization.py

@@ -1,31 +0,0 @@
-from datetime import datetime, date
-from decimal import Decimal
-from typing import Any, Dict, TYPE_CHECKING
-
-if TYPE_CHECKING:
-    from agate import Table
-
-
-def _column_filter(val: Any) -> Any:
-    return (
-        float(val)
-        if isinstance(val, Decimal)
-        else (
-            str(val)
-            if isinstance(val, datetime)
-            else str(val) if isinstance(val, date) else str(val)
-        )
-    )
-
-
-def serialize_agate_table(table: "Table") -> Dict[str, Any]:
-    rows = []
-    for row in table.rows:
-        row = list(map(_column_filter, row))
-        rows.append(row)
-
-    return {
-        "column_names": table.column_names,
-        "column_types": [t.__class__.__name__ for t in table.column_types],
-        "rows": rows,
-    }