fix test

Author: sycai

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

@@ -8,8 +8,8 @@ WITH `bfcte_0` AS (
     AI.GENERATE_BOOL(
       prompt => (`bfcol_0`, ' is the same as ', `bfcol_0`),
       connection_id => 'test_connection_id',
-      request_type => 'SHARED',
-      model_params => JSON '{}'
+      endpoint => 'gemini-2.5-flash',
+      request_type => 'SHARED'
     ) AS `bfcol_1`
   FROM `bfcte_0`
 )

tests/unit/core/compile/sqlglot/expressions/snapshots/test_ai_ops/test_ai_generate_bool_with_model_param/out.sql

@@ -0,0 +1,18 @@
+WITH `bfcte_0` AS (
+  SELECT
+    `string_col` AS `bfcol_0`
+  FROM `bigframes-dev`.`sqlglot_test`.`scalar_types`
+), `bfcte_1` AS (
+  SELECT
+    *,
+    AI.GENERATE_BOOL(
+      prompt => (`bfcol_0`, ' is the same as ', `bfcol_0`),
+      connection_id => 'test_connection_id',
+      request_type => 'SHARED',
+      model_params => JSON '{}'
+    ) AS `bfcol_1`
+  FROM `bfcte_0`
+)
+SELECT
+  `bfcol_1` AS `result`
+FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/test_ai_ops.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import json
+import sys
 
 import pytest
 
@@ -26,6 +27,31 @@
 def test_ai_generate_bool(scalar_types_df: dataframe.DataFrame, snapshot):
     col_name = "string_col"
 
+    op = ops.AIGenerateBool(
+        prompt_context=(None, " is the same as ", None),
+        connection_id="test_connection_id",
+        endpoint="gemini-2.5-flash",
+        request_type="shared",
+        model_params=None,
+    )
+
+    sql = utils._apply_unary_ops(
+        scalar_types_df, [op.as_expr(col_name, col_name)], ["result"]
+    )
+
+    snapshot.assert_match(sql, "out.sql")
+
+
+def test_ai_generate_bool_with_model_param(
+    scalar_types_df: dataframe.DataFrame, snapshot
+):
+    if sys.version_info < (3, 10):
+        pytest.skip(
+            "Skip test because SQLGLot cannot compile model params to JSON at this env."
+        )
+
+    col_name = "string_col"
+
     op = ops.AIGenerateBool(
         prompt_context=(None, " is the same as ", None),
         connection_id="test_connection_id",