test(bigframes): Disable ibis fallback for system tests (#17333)

Author: TrevorBergeron

packages/bigframes/bigframes/core/compile/sqlglot/expressions/ai_ops.py

@@ -111,7 +111,8 @@ def _construct_prompt(
         else:
             prompt.append(sge.Literal.string(elem))
 
-    return sge.Kwarg(this=param_name, expression=sge.Tuple(expressions=prompt))
+    # Need Struct rather than tuple syntax, as tuple syntax is ambiguous for single arg
+    return sge.Kwarg(this=param_name, expression=sge.Struct(expressions=prompt))
 
 
 def _construct_named_args(op: ops.ScalarOp) -> list[sge.Kwarg]:

packages/bigframes/tests/system/conftest.py

@@ -1325,6 +1325,14 @@ def usa_names_grouped_table(
         return session.bqclient.get_table(table_id)
 
 
+@pytest.fixture(scope="session", autouse=True)
+def use_sqlglot_compiler():
+    original_setting = bigframes.options.experiments.sql_compiler
+    bigframes.options.experiments.sql_compiler = "experimental"
+    yield
+    bigframes.options.experiments.sql_compiler = original_setting
+
+
 @pytest.fixture()
 def restore_sampling_settings():
     enable_downsampling = bigframes.options.sampling.enable_downsampling

packages/bigframes/tests/unit/core/compile/sqlglot/expressions/snapshots/test_ai_ops/test_ai_classify/None/out.sql

@@ -1,3 +1,3 @@
 SELECT
-  AI.CLASSIFY(input => (`string_col`), categories => ['greeting', 'rejection']) AS `result`
+  AI.CLASSIFY(input => STRUCT(`string_col`), categories => ['greeting', 'rejection']) AS `result`
 FROM `bigframes-dev`.`sqlglot_test`.`scalar_types` AS `bft_0`

packages/bigframes/tests/unit/core/compile/sqlglot/expressions/snapshots/test_ai_ops/test_ai_classify/bigframes-dev.us.bigframes-default-connection/out.sql

@@ -1,6 +1,6 @@
 SELECT
   AI.CLASSIFY(
-    input => (`string_col`),
+    input => STRUCT(`string_col`),
     categories => ['greeting', 'rejection'],
     connection_id => 'bigframes-dev.us.bigframes-default-connection'
   ) AS `result`

packages/bigframes/tests/unit/core/compile/sqlglot/expressions/snapshots/test_ai_ops/test_ai_classify_multi_with_list_examples/out.sql

@@ -1,6 +1,6 @@
 SELECT
   AI.CLASSIFY(
-    input => (`string_col`),
+    input => STRUCT(`string_col`),
     categories => ['greeting', 'rejection'],
     examples => [('hi', ['greeting', 'positive']), ('bye', ['rejection', 'negative'])],
     output_mode => 'multi'

packages/bigframes/tests/unit/core/compile/sqlglot/expressions/snapshots/test_ai_ops/test_ai_classify_with_output_mode/out.sql

@@ -1,6 +1,6 @@
 SELECT
   AI.CLASSIFY(
-    input => (`string_col`),
+    input => STRUCT(`string_col`),
     categories => ['greeting', 'rejection'],
     output_mode => 'multi'
   ) AS `result`

packages/bigframes/tests/unit/core/compile/sqlglot/expressions/snapshots/test_ai_ops/test_ai_classify_with_params/out.sql

@@ -1,6 +1,6 @@
 SELECT
   AI.CLASSIFY(
-    input => (`string_col`),
+    input => STRUCT(`string_col`),
     categories => ['greeting', 'rejection'],
     examples => [('hi', 'greeting'), ('bye', 'rejection')],
     endpoint => 'gemini-2.5-flash',

packages/bigframes/tests/unit/core/compile/sqlglot/expressions/snapshots/test_ai_ops/test_ai_generate/out.sql

@@ -1,6 +1,6 @@
 SELECT
   AI.GENERATE(
-    prompt => (`string_col`, ' is the same as ', `string_col`),
+    prompt => STRUCT(`string_col`, ' is the same as ', `string_col`),
     endpoint => 'gemini-2.5-flash',
     request_type => 'SHARED'
   ) AS `result`

packages/bigframes/tests/unit/core/compile/sqlglot/expressions/snapshots/test_ai_ops/test_ai_generate_bool/out.sql

@@ -1,6 +1,6 @@
 SELECT
   AI.GENERATE_BOOL(
-    prompt => (`string_col`, ' is the same as ', `string_col`),
+    prompt => STRUCT(`string_col`, ' is the same as ', `string_col`),
     endpoint => 'gemini-2.5-flash'
   ) AS `result`
 FROM `bigframes-dev`.`sqlglot_test`.`scalar_types` AS `bft_0`

packages/bigframes/tests/unit/core/compile/sqlglot/expressions/snapshots/test_ai_ops/test_ai_generate_bool_with_connection_id/out.sql

@@ -1,6 +1,6 @@
 SELECT
   AI.GENERATE_BOOL(
-    prompt => (`string_col`, ' is the same as ', `string_col`),
+    prompt => STRUCT(`string_col`, ' is the same as ', `string_col`),
     connection_id => 'bigframes-dev.us.bigframes-default-connection',
     endpoint => 'gemini-2.5-flash'
   ) AS `result`