From be4865837052f7ccba6b1eab945ab8a9d6d0f51f Mon Sep 17 00:00:00 2001 From: Garrett Wu Date: Thu, 12 Feb 2026 00:07:35 +0000 Subject: [PATCH] docs: fix generate_text and generate_table input docs --- bigframes/bigquery/_operations/ai.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/bigframes/bigquery/_operations/ai.py b/bigframes/bigquery/_operations/ai.py index 7f9c3eb55f..5fe9f306d5 100644 --- a/bigframes/bigquery/_operations/ai.py +++ b/bigframes/bigquery/_operations/ai.py @@ -522,10 +522,10 @@ def generate_text( model (bigframes.ml.base.BaseEstimator or str): The model to use for text generation. data (bigframes.pandas.DataFrame or bigframes.pandas.Series): - The data to generate embeddings for. If a Series is provided, it is - treated as the 'content' column. If a DataFrame is provided, it - must contain a 'content' column, or you must rename the column you - wish to embed to 'content'. + The data to generate text for. If a Series is provided, it is + treated as the 'prompt' column. If a DataFrame is provided, it + must contain a 'prompt' column, or you must rename the column you + wish to generate text to 'prompt'. temperature (float, optional): A FLOAT64 value that is used for sampling promiscuity. The value must be in the range ``[0.0, 1.0]``. A lower temperature works well @@ -638,10 +638,10 @@ def generate_table( model (bigframes.ml.base.BaseEstimator or str): The model to use for table generation. data (bigframes.pandas.DataFrame or bigframes.pandas.Series): - The data to generate embeddings for. If a Series is provided, it is - treated as the 'content' column. If a DataFrame is provided, it - must contain a 'content' column, or you must rename the column you - wish to embed to 'content'. + The data to generate table for. If a Series is provided, it is + treated as the 'prompt' column. If a DataFrame is provided, it + must contain a 'prompt' column, or you must rename the column you + wish to generate table to 'prompt'. output_schema (str): A string defining the output schema (e.g., "col1 STRING, col2 INT64"). temperature (float, optional):