@@ -47,13 +47,13 @@ def generate_bool(
47
47
... "col_1": ["apple", "bear", "pear"],
48
48
... "col_2": ["fruit", "animal", "animal"]
49
49
... })
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"]))
51
51
0 {'result': True, 'full_response': '{"candidate...
52
52
1 {'result': True, 'full_response': '{"candidate...
53
53
2 {'result': False, 'full_response': '{"candidat...
54
54
dtype: struct<result: bool, full_response: string, status: string>[pyarrow]
55
55
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")
57
57
0 True
58
58
1 True
59
59
2 False
@@ -66,7 +66,7 @@ def generate_bool(
66
66
... }
67
67
... }
68
68
... }
69
- >>> bbq.ai_generate_bool (
69
+ >>> bbq.ai.generate_bool (
70
70
... (df["col_1"], " is a ", df["col_2"]),
71
71
... endpoint="gemini-2.5-pro",
72
72
... model_params=model_params,
0 commit comments