feat(snowflake)!: Annotated type for ARRAY_CONSTRUCT_COMPACT #6496

Author: fivetran-kwoodbeck

sqlglot/expressions.py

@@ -6030,7 +6030,7 @@ class ArrayConcatAgg(AggFunc):
 
 
 class ArrayConstructCompact(Func):
-    arg_types = {"expressions": True}
+    arg_types = {"expressions": False}
     is_var_len_args = True
 
 

sqlglot/typing/snowflake.py

@@ -190,6 +190,7 @@ def _annotate_math_with_float_decfloat(
             exp.ApproxTopK,
             exp.ApproxTopKEstimate,
             exp.ArrayAgg,
+            exp.ArrayConstructCompact,
             exp.ArrayUniqueAgg,
             exp.ArrayUnionAgg,
             exp.RegexpExtractAll,

tests/fixtures/optimizer/annotate_functions.sql

@@ -1704,6 +1704,14 @@ DATE;
 ADD_MONTHS(tbl.timestamp_col, -1);
 TIMESTAMP;
 
+# dialect: snowflake
+ARRAY_CONSTRUCT_COMPACT();
+ARRAY;
+
+# dialect: snowflake
+ARRAY_CONSTRUCT_COMPACT(1, null, 2);
+ARRAY;
+
 # dialect: snowflake
 ASIN(tbl.double_col);
 DOUBLE;