support case-insensitive type parsing

Author: sycai

bigframes/operations/output_schemas.py

@@ -31,23 +31,23 @@ def parse_sql_type(sql: str) -> pa.DataType:
     """
     sql = sql.strip()
 
-    if sql == "STRING":
+    if sql.upper() == "STRING":
         return pa.string()
 
-    if sql == "INT64":
+    if sql.upper() == "INT64":
         return pa.int64()
 
-    if sql == "FLOAT64":
+    if sql.upper() == "FLOAT64":
         return pa.float64()
 
-    if sql == "BOOL":
+    if sql.upper() == "BOOL":
         return pa.bool_()
 
-    if sql.startswith("ARRAY<") and sql.endswith(">"):
+    if sql.upper().startswith("ARRAY<") and sql.endswith(">"):
         inner_type = sql[len("ARRAY<") : -1]
         return pa.list_(parse_sql_type(inner_type))
 
-    if sql.startswith("STRUCT<") and sql.endswith(">"):
+    if sql.upper().startswith("STRUCT<") and sql.endswith(">"):
         inner_fields = parse_sql_fields(sql[len("STRUCT<") : -1])
         return pa.struct(inner_fields)
 

tests/system/small/bigquery/test_ai.py

@@ -94,7 +94,7 @@ def test_ai_generate_with_output_schema(session):
     result = bbq.ai.generate(
         prompt,
         endpoint="gemini-2.5-flash",
-        output_schema={"population": "INT64", "is_in_north_america": "BOOL"},
+        output_schema={"population": "INT64", "is_in_north_america": "bool"},
     )
 
     assert _contains_no_nulls(result)

tests/unit/operations/test_output_schemas.py

@@ -23,6 +23,7 @@
     [
         ("INT64", pa.int64()),
         (" INT64  ", pa.int64()),
+        ("int64", pa.int64()),
         ("FLOAT64", pa.float64()),
         ("STRING", pa.string()),
         ("BOOL", pa.bool_()),
@@ -44,7 +45,7 @@
             pa.list_(pa.struct((pa.field("x", pa.int64()), pa.field("y", pa.int64())))),
         ),
         (
-            "STRUCT<x ARRAY<FLOAT64>, y STRUCT<a BOOL, b STRING>>",
+            "STRUCT<y STRUCT<b STRING, a BOOL>, x ARRAY<FLOAT64>>",
             pa.struct(
                 (
                     pa.field("x", pa.list_(pa.float64())),
@@ -66,14 +67,13 @@ def test_parse_sql_to_pyarrow_dtype(sql, expected):
 @pytest.mark.parametrize(
     "sql",
     [
-        "int64",
         "a INT64",
         "ARRAY<>",
         "ARRAY<INT64",
         "ARRAY<x INT64>" "ARRAY<int64>" "STRUCT<>",
         "DATE",
         "STRUCT<INT64, FLOAT64>",
-        "STRUCT<x int64>",
+        "ARRAY<ARRAY<>>",
     ],
 )
 def test_parse_sql_to_pyarrow_dtype_invalid_input_raies_error(sql):
@@ -89,6 +89,10 @@ def test_parse_sql_to_pyarrow_dtype_invalid_input_raies_error(sql):
             "x INT64, y FLOAT64",
             (pa.field("x", pa.int64()), pa.field("y", pa.float64())),
         ),
+        (
+            "y FLOAT64, x INT64",
+            (pa.field("x", pa.int64()), pa.field("y", pa.float64())),
+        ),
     ],
 )
 def test_parse_sql_fields(sql, expected):