Skip to content

Commit 6a53f41

Browse files
google-genai-botcopybara-github
authored andcommitted
chore: set a maximum number of rows for BigQuery query results in the sample agent
The `BigQueryToolConfig` now sets `max_query_result_rows=50` explicitly to showcase this particular configuration. PiperOrigin-RevId: 871709345
1 parent dbd6420 commit 6a53f41

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

contributing/samples/bigquery/agent.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,9 @@
3838
# tool read-only) or PROTECTED (only allows writes in the anonymous dataset of a
3939
# BigQuery session) write mode.
4040
tool_config = BigQueryToolConfig(
41-
write_mode=WriteMode.ALLOWED, application_name=BIGQUERY_AGENT_NAME
41+
write_mode=WriteMode.ALLOWED,
42+
application_name=BIGQUERY_AGENT_NAME,
43+
max_query_result_rows=50,
4244
)
4345

4446
if CREDENTIALS_TYPE == AuthCredentialTypes.OAUTH2:

0 commit comments

Comments
 (0)