Skip to content

Commit c38438e

Browse files
authored
feat: Support custom scopes and maxQueryResultRows (#77)
1 parent 8ab8d45 commit c38438e

3 files changed

Lines changed: 15 additions & 1 deletion

File tree

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,8 @@ Set the following environment variables before starting the Gemini CLI. These va
4848
```bash
4949
export BIGQUERY_PROJECT="<your-gcp-project-id>"
5050
export BIGQUERY_LOCATION="<your-dataset-location>" # Optional
51+
export BIGQUERY_SCOPES="https://www.googleapis.com/auth/bigquery,https://www.googleapis.com/auth/drive" # Optional
52+
export BIGQUERY_MAX_QUERY_RESULT_ROWS=50 # Optional
5153
```
5254

5355
Ensure [Application Default Credentials](https://cloud.google.com/docs/authentication/gcloud) are available in your environment.

gemini-extension.json

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,16 @@
2323
"name": "Location",
2424
"description": "(Optional) Location of the BigQuery resources",
2525
"envVar": "BIGQUERY_LOCATION"
26+
},
27+
{
28+
"name": "Authorization Scopes",
29+
"description": "(Optional) Authorization scopes for Google APIs (Dataplex, Conversational API, BigQuery), comma-separated",
30+
"envVar": "BIGQUERY_SCOPES"
31+
},
32+
{
33+
"name": "Maximum Query Result Rows",
34+
"description": "(Optional) Maximum number of rows to return from BigQuery query results",
35+
"envVar": "BIGQUERY_MAX_QUERY_RESULT_ROWS"
2636
}
2737
]
28-
}
38+
}

tools.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@ sources:
1818
project: ${BIGQUERY_PROJECT}
1919
location: ${BIGQUERY_LOCATION:}
2020
useClientOAuth: ${BIGQUERY_USE_CLIENT_OAUTH:false}
21+
scopes: ${BIGQUERY_SCOPES:}
22+
maxQueryResultRows: ${BIGQUERY_MAX_QUERY_RESULT_ROWS:50}
2123

2224
tools:
2325
ask_data_insights:

0 commit comments

Comments
 (0)