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

{query_cache_common-1.9.0 → query_cache_common-1.9.1}/.gitignore

@@ -213,9 +213,11 @@ data/
 # Service account credentials
 .creds/
 
-# Load test generated files
-tooling/load_test/data/*.json
+# Load test generated files (flat JSONs + timestamped run dirs)
+tooling/load_test/data/
 tooling/load_test/proto/
+# Active k8s load-test run ID written by k8s prepare-run.
+tooling/load_test/k8s/.run-id
 
 # Turbo integration test
 tests/integration/dbt_run_cache_turbo/projects/
@@ -229,3 +231,6 @@ docs/superpowers/
 
 # Git worktrees
 .worktrees/
+
+# Google Antigravity CLI (agy)
+.antigravitycli/

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

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

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

@@ -32,17 +32,20 @@ def current_epoch_millis() -> int:
 
 
 def extract_select_from_ctas(create: CreateStatement) -> SqlglotExpression:
-    """Extract the SELECT statement from a CREATE TABLE / VIEW AS statement.
+    """Extract the query from a CREATE TABLE / VIEW AS statement.
+
+    The query body may be a single SELECT or a set operation (UNION / INTERSECT /
+    EXCEPT), optionally wrapped in parentheses.
 
     Args:
         create: The CTAS statement.
 
     Returns:
-        The extracted SELECT statement.
+        The extracted query expression.
     """
     from sqlglot import exp
 
-    if not isinstance(create.expression, (exp.Subquery, exp.Select)):
+    if not isinstance(create.expression, (exp.Subquery, exp.Query)):
         raise ValueError(
             "Invalid CREATE statement. Only CREATE ... AS SELECT statements are supported"
         )