fix test

sycai · sycai · commit 99c7bfc7d70e · 2025-10-06T17:57:49.000Z
diff --git a/tests/system/small/bigquery/test_ai.py b/tests/system/small/bigquery/test_ai.py
@@ -118,7 +118,7 @@ def test_ai_generate_with_invalid_output_schema_raise_error(session):
         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": "JSON"},
         )
 
 

Original file line number	Diff line number	Diff line change
`@@ -118,7 +118,7 @@ def test_ai_generate_with_invalid_output_schema_raise_error(session):`
`118`	`118`	`bbq.ai.generate(`
`119`	`119`	`prompt,`
`120`	`120`	`endpoint="gemini-2.5-flash",`
`121`		`- output_schema={"population": "INT64", "is_in_north_america": "bool"},`
	`121`	`+ output_schema={"population": "INT64", "is_in_north_america": "JSON"},`
`122`	`122`	`)`
`123`	`123`
`124`	`124`