feat(bigquery): support to query BigQuery external table (#1055)

Canner · Feb 6, 2025 · 83389fd · 83389fd
1 parent 664e6af
commit 83389fd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/ibis-server/app/model/connector.py b/ibis-server/app/model/connector.py
@@ -151,6 +151,12 @@ def query(self, sql: str, limit: int) -> pd.DataFrame:
                 credentials = service_account.Credentials.from_service_account_info(
                     credits_json
                 )
+                credentials = credentials.with_scopes(
+                    [
+                        "https://www.googleapis.com/auth/drive",
+                        "https://www.googleapis.com/auth/cloud-platform",
+                    ]
+                )
                 client = bigquery.Client(credentials=credentials)
                 ibis_schema_mapper = ibis.backends.bigquery.BigQuerySchema()
                 bq_fields = client.query(sql).result()

diff --git a/ibis-server/app/model/data_source.py b/ibis-server/app/model/data_source.py
@@ -101,6 +101,12 @@ def get_bigquery_connection(info: BigQueryConnectionInfo) -> BaseBackend:
         credentials = service_account.Credentials.from_service_account_info(
             credits_json
         )
+        credentials = credentials.with_scopes(
+            [
+                "https://www.googleapis.com/auth/drive",
+                "https://www.googleapis.com/auth/cloud-platform",
+            ]
+        )
         return ibis.bigquery.connect(
             project_id=info.project_id.get_secret_value(),
             dataset_id=info.dataset_id.get_secret_value(),