Skip to content

Commit 57b923a

Browse files
include google drive shared drive (#465)
* include google drive shared drive * update version changelog * update changelog format * bump version
1 parent 5c5fcc9 commit 57b923a

File tree

3 files changed

+23
-3
lines changed

3 files changed

+23
-3
lines changed

CHANGELOG.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,9 @@
1+
## 1.0.9
2+
3+
### Fixes
4+
5+
* **Google Drive connector also include shared drive**
6+
17
## 1.0.8
28

39
### Enhancements

unstructured_ingest/__version__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "1.0.8" # pragma: no cover
1+
__version__ = "1.0.9" # pragma: no cover

unstructured_ingest/processes/connectors/google_drive.py

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,13 @@ def verify_drive_api_enabled(client) -> None:
153153
"""
154154
try:
155155
# A very minimal call: list 1 file from the drive.
156-
client.list(spaces="drive", pageSize=1, fields="files(id)").execute()
156+
client.list(
157+
supportsAllDrives=True,
158+
includeItemsFromAllDrives=True,
159+
spaces="drive",
160+
pageSize=1,
161+
fields="files(id)",
162+
).execute()
157163
except HttpError as e:
158164
error_content = e.content.decode() if hasattr(e, "content") else ""
159165
lower_error = error_content.lower()
@@ -183,6 +189,8 @@ def count_files_recursively(files_client, folder_id: str, extensions: list[str]
183189
page_token = None
184190
while True:
185191
response = files_client.list(
192+
supportsAllDrives=True,
193+
includeItemsFromAllDrives=True,
186194
spaces="drive",
187195
q=query,
188196
fields="nextPageToken, files(id, mimeType, fileExtension)",
@@ -251,6 +259,8 @@ def precheck(self) -> None:
251259
else:
252260
# Non-recursive: check for at least one immediate non-folder child.
253261
response = client.list(
262+
supportsAllDrives=True,
263+
includeItemsFromAllDrives=True,
254264
spaces="drive",
255265
fields="files(id)",
256266
pageSize=1,
@@ -348,6 +358,8 @@ def get_paginated_results(
348358
files_response = []
349359
while not done:
350360
response: dict = files_client.list(
361+
supportsAllDrives=True,
362+
includeItemsFromAllDrives=True,
351363
spaces="drive",
352364
fields=fields_input,
353365
corpora="user",
@@ -381,7 +393,9 @@ def get_paginated_results(
381393
return files_response
382394

383395
def get_root_info(self, files_client, object_id: str) -> dict:
384-
return files_client.get(fileId=object_id, fields=",".join(self.fields)).execute()
396+
return files_client.get(
397+
supportsAllDrives=True, fileId=object_id, fields=",".join(self.fields)
398+
).execute()
385399

386400
def get_files(
387401
self,

0 commit comments

Comments
 (0)