jason/google-mcp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

All Drive API operations now include supportsAllDrives=True

Browse Source
This commit is contained in:
Taylor Wilsdon
2025-11-04 10:19:37 -05:00
parent 7ac9087939
commit 24d79ab470
5 changed files with 27 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
gdocs/docs_tools.py
View File

@@ -71,7 +71,9 @@ async def search_docs(
service.files().list(
q=f"name contains '{escaped_query}' and mimeType='application/vnd.google-apps.document' and trashed=false",
pageSize=page_size,
fields="files(id, name, createdTime, modifiedTime, webViewLink)"
fields="files(id, name, createdTime, modifiedTime, webViewLink)",
supportsAllDrives=True,
includeItemsFromAllDrives=True
).execute
)
files = response.get('files', [])
@@ -110,7 +112,8 @@ async def get_doc_content(
# Step 2: Get file metadata from Drive
file_metadata = await asyncio.to_thread(
drive_service.files().get(
fileId=document_id, fields="id, name, mimeType, webViewLink"
fileId=document_id, fields="id, name, mimeType, webViewLink",
supportsAllDrives=True
).execute
)
mime_type = file_metadata.get("mimeType", "")
@@ -214,7 +217,7 @@ async def get_doc_content(
request_obj = (
drive_service.files().export_media(fileId=document_id, mimeType=effective_export_mime)
if effective_export_mime
else drive_service.files().get_media(fileId=document_id)
else drive_service.files().get_media(fileId=document_id, supportsAllDrives=True)
)
fh = io.BytesIO()
@@ -265,7 +268,9 @@ async def list_docs_in_folder(
service.files().list(
q=f"'{folder_id}' in parents and mimeType='application/vnd.google-apps.document' and trashed=false",
pageSize=page_size,
fields="files(id, name, modifiedTime, webViewLink)"
fields="files(id, name, modifiedTime, webViewLink)",
supportsAllDrives=True,
includeItemsFromAllDrives=True
).execute
)
items = rsp.get('files', [])
@@ -613,7 +618,8 @@ async def insert_doc_image(
file_metadata = await asyncio.to_thread(
drive_service.files().get(
fileId=image_source,
fields="id, name, mimeType"
fields="id, name, mimeType",
supportsAllDrives=True
).execute
)
mime_type = file_metadata.get('mimeType', '')
@@ -1080,7 +1086,8 @@ async def export_doc_to_pdf(
file_metadata = await asyncio.to_thread(
service.files().get(
fileId=document_id,
fields="id, name, mimeType, webViewLink"
fields="id, name, mimeType, webViewLink",
supportsAllDrives=True
).execute
)
except Exception as e:
@@ -1100,7 +1107,8 @@ async def export_doc_to_pdf(
try:
request_obj = service.files().export_media(
fileId=document_id,
mimeType='application/pdf'
mimeType='application/pdf',
supportsAllDrives=True
)
fh = io.BytesIO()