diff --git a/contributing/samples/bigquery/agent.py b/contributing/samples/bigquery/agent.py index 0402d617..5ff34c7c 100644 --- a/contributing/samples/bigquery/agent.py +++ b/contributing/samples/bigquery/agent.py @@ -38,7 +38,9 @@ BIGQUERY_AGENT_NAME = "adk_sample_bigquery_agent" # tool read-only) or PROTECTED (only allows writes in the anonymous dataset of a # BigQuery session) write mode. tool_config = BigQueryToolConfig( - write_mode=WriteMode.ALLOWED, application_name=BIGQUERY_AGENT_NAME + write_mode=WriteMode.ALLOWED, + application_name=BIGQUERY_AGENT_NAME, + max_query_result_rows=50, ) if CREDENTIALS_TYPE == AuthCredentialTypes.OAUTH2: