add require scope and fmt

Canner · Feb 6, 2025 · 465fcc0 · 465fcc0
1 parent 5fff1c9
commit 465fcc0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/ibis-server/app/model/connector.py b/ibis-server/app/model/connector.py
@@ -149,6 +149,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
@@ -98,10 +98,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",
-        ])
+        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(),