feat: Support custom scopes and maxQueryResultRows

README.md

@@ -48,6 +48,8 @@ Set the following environment variables before starting the Gemini CLI. These va
 ```bash
 export BIGQUERY_PROJECT="<your-gcp-project-id>"
 export BIGQUERY_LOCATION="<your-dataset-location>"  # Optional
+export BIGQUERY_SCOPES="https://www.googleapis.com/auth/bigquery,https://www.googleapis.com/auth/drive"  # Optional
+export BIGQUERY_MAX_QUERY_RESULT_ROWS=50  # Optional
 ```
 
 Ensure [Application Default Credentials](https://cloud.google.com/docs/authentication/gcloud) are available in your environment.

gemini-extension.json

@@ -23,6 +23,16 @@
       "name": "Location",
       "description": "(Optional) Location of the BigQuery resources",
       "envVar": "BIGQUERY_LOCATION"
+    },
+    {
+      "name": "Authorization Scopes",
+      "description": "(Optional) Authorization scopes for Google APIs (Dataplex, Conversational API, BigQuery), comma-separated",
+      "envVar": "BIGQUERY_SCOPES"
+    },
+    {
+      "name": "Maximum Query Result Rows",
+      "description": "(Optional) Maximum number of rows to return from BigQuery query results",
+      "envVar": "BIGQUERY_MAX_QUERY_RESULT_ROWS"
     }
   ]
-}
+}

tools.yaml

@@ -18,6 +18,8 @@ sources:
     project: ${BIGQUERY_PROJECT}
     location: ${BIGQUERY_LOCATION:}
     useClientOAuth: ${BIGQUERY_USE_CLIENT_OAUTH:false}
+    scopes: ${BIGQUERY_SCOPES:}
+    maxQueryResultRows: ${BIGQUERY_MAX_QUERY_RESULT_ROWS:50}
 
 tools:
   ask_data_insights: