query-cache-common 1.9.1__tar.gz → 1.10.0__tar.gz

{query_cache_common-1.9.1 → query_cache_common-1.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: query-cache-common
-Version: 1.9.1
+Version: 1.10.0
 Summary: Common code for Query Cache shared across client and server
 License: 
                                          Apache License

{query_cache_common-1.9.1 → query_cache_common-1.10.0}/src/query_cache_common/models/services/execution_service_models.py

@@ -47,6 +47,7 @@ class SQLExecution(BaseSerDeModel):
     query_dependencies: t.List[shared_models.QueryDependency] = field(default_factory=list)
     semantic_extras: t.Dict[str, str] = field(default_factory=dict)
     labels: t.Dict[str, str] = field(default_factory=dict)
+    dbt_node_state: t.Optional[shared_models.DbtNodeState] = None
 
     @classmethod
     def from_submit_sql_request(
@@ -62,6 +63,7 @@ class SQLExecution(BaseSerDeModel):
             query_dependencies=req.query_dependencies,
             semantic_extras=req.semantic_extras,
             labels=req.labels,
+            dbt_node_state=req.dbt_node_state,
         )
 
 

{query_cache_common-1.9.1 → query_cache_common-1.10.0}/src/query_cache_common/models/services/sql_service_models.py

@@ -47,6 +47,7 @@ class SubmitEnrichedSQLRequest(BaseSerDeModel):
     clone_table_properties: t.Optional[TableProperties] = None
     stale_upstream_policy: shared_models.StaleUpstreamPolicy = shared_models.StaleUpstreamPolicy.ANY
     clone_chain_depth_limit: t.Optional[int] = None
+    dbt_node_state: t.Optional[shared_models.DbtNodeState] = None
 
 
 @proto_dataclass(sql_service_pb2.QueryHashMatchMetadataInfo)

{query_cache_common-1.9.1 → query_cache_common-1.10.0}/src/query_cache_common/models/shared_models.py

@@ -101,3 +101,19 @@ class ExplainedDecision(BaseSerDeModel):
             clone_rejection_reason=RejectionReason.NO_SUITABLE_MATCH_FOUND,
             is_stale=False,
         )
+
+
+@proto_dataclass(shared_pb2.DbtNodeState)
+class DbtNodeState(BaseSerDeModel):
+    node_unique_id: str
+    target_name: str
+    project_name: str
+    resource_type: str
+    node_hash: str
+    node_body_hash: str
+    node_configs_hash: str
+    node_relation_hash: str
+    node_macros_hash: str
+    node_contract_hash: str
+    profile_name: str
+    node_persisted_descriptions_hash: t.Optional[str] = None

{query_cache_common-1.9.1 → query_cache_common-1.10.0}/src/query_cache_common/utils.py

@@ -26,6 +26,25 @@ def str_to_bool(s: str | None) -> bool:
     return s.lower() in ("true", "1", "t", "y", "yes", "on")
 
 
+def to_bool(value: t.Any) -> bool:
+    """Coerce a bool, int, or string to a boolean.
+
+    Unlike str_to_bool, this accepts bools and ints directly (strings are delegated to
+    str_to_bool) and raises ValueError for any other type rather than silently
+    returning False.
+    """
+    if isinstance(value, bool):
+        return value
+    if isinstance(value, str):
+        return str_to_bool(value)
+    if isinstance(value, int):
+        return bool(value)
+    raise ValueError(
+        f"Cannot convert value of type {type(value).__name__} to bool: {value!r}. "
+        "Expected bool, int, or string (e.g., 'true', 'false', 'yes', 'no')."
+    )
+
+
 def current_epoch_millis() -> int:
     """Return the current time in milliseconds since Unix epoch."""
     return time.time_ns() // 1_000_000