@@ -47,13 +47,13 @@ def generate_bool(
4747 ... "col_1": ["apple", "bear", "pear"],
4848 ... "col_2": ["fruit", "animal", "animal"]
4949 ... })
50- >>> bbq.ai_generate_bool ((df["col_1"], " is a ", df["col_2"]))
50+ >>> bbq.ai.generate_bool ((df["col_1"], " is a ", df["col_2"]))
5151 0 {'result': True, 'full_response': '{"candidate...
5252 1 {'result': True, 'full_response': '{"candidate...
5353 2 {'result': False, 'full_response': '{"candidat...
5454 dtype: struct<result: bool, full_response: string, status: string>[pyarrow]
5555
56- >>> bbq.ai_generate_bool ((df["col_1"], " is a ", df["col_2"])).struct.field("result")
56+ >>> bbq.ai.generate_bool ((df["col_1"], " is a ", df["col_2"])).struct.field("result")
5757 0 True
5858 1 True
5959 2 False
@@ -66,7 +66,7 @@ def generate_bool(
6666 ... }
6767 ... }
6868 ... }
69- >>> bbq.ai_generate_bool (
69+ >>> bbq.ai.generate_bool (
7070 ... (df["col_1"], " is a ", df["col_2"]),
7171 ... endpoint="gemini-2.5-pro",
7272 ... model_params=model_params,
0 commit comments