Skip to content
This repository was archived by the owner on Apr 1, 2026. It is now read-only.

Commit 89118ea

Browse files
committed
docs: correct the code sample
1 parent e03d799 commit 89118ea

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

bigframes/bigquery/ai.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,10 +43,14 @@
4343
>>> import bigframes.pandas as bpd
4444
>>> import bigframes.bigquery as bbq
4545
46-
>>> df = bpd.DataFrame({"text_input": ["Is this a positive review?", "The food was terrible."]}) # doctest: +SKIP
46+
>>> df = bpd.DataFrame({
47+
... "text_input": [
48+
... "Is this a positive review? The food was terrible.",
49+
... ],
50+
... }) # doctest: +SKIP
4751
4852
>>> # Assuming a Gemini model has been created in BigQuery as 'my_gemini_model'
49-
>>> result = bq.ai.generate_text(df["text_input"], model_name="my_gemini_model") # doctest: +SKIP
53+
>>> result = bq.ai.generate_text("my_gemini_model", df["text_input"]) # doctest: +SKIP
5054
5155
For more information on the underlying BigQuery ML syntax, see:
5256
https://cloud.google.com/bigquery/docs/reference/standard-sql/bigqueryml-syntax-ai-generate-bool

0 commit comments

Comments
 (0)