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

Search with file type

Browse Source
This commit is contained in:
Francisco
2026-02-25 21:27:29 +13:00
parent 9631b9e3eb
commit 7e33626b72
2 changed files with 69 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
gdrive/drive_tools.py
View File

@@ -36,6 +36,7 @@ from gdrive.drive_helpers import (
format_permission_info,
get_drive_image_url,
resolve_drive_item,
resolve_file_type_mime,
resolve_folder_id,
validate_expiration_time,
validate_share_role,
@@ -60,6 +61,7 @@ async def search_drive_files(
drive_id: Optional[str] = None,
include_items_from_all_drives: bool = True,
corpora: Optional[str] = None,
file_type: Optional[str] = None,
) -> str:
"""
Searches for files and folders within a user's Google Drive, including shared drives.
@@ -73,12 +75,15 @@ async def search_drive_files(
corpora (Optional[str]): Bodies of items to query (e.g., 'user', 'domain', 'drive', 'allDrives').
If 'drive_id' is specified and 'corpora' is None, it defaults to 'drive'.
Otherwise, Drive API default behavior applies. Prefer 'user' or 'drive' over 'allDrives' for efficiency.
file_type (Optional[str]): Restrict results to a specific file type. Accepts a friendly
name ('folder', 'document', 'doc', 'spreadsheet', 'sheet', 'presentation', 'slides', 'form', 'drawing', 'pdf', 'shortcut') or
any raw MIME type string (e.g. 'application/pdf'). Defaults to None (all types).
Returns:
str: A formatted list of found files/folders with their details (ID, name, type, size, modified time, link).
"""
logger.info(
f"[search_drive_files] Invoked. Email: '{user_google_email}', Query: '{query}'"
f"[search_drive_files] Invoked. Email: '{user_google_email}', Query: '{query}', file_type: '{file_type}'"
)
# Check if the query looks like a structured Drive query or free text
@@ -98,6 +103,11 @@ async def search_drive_files(
f"[search_drive_files] Reformatting free text query '{query}' to '{final_query}'"
)
if file_type is not None:
mime = resolve_file_type_mime(file_type)
final_query += f" and mimeType = '{mime}'"
logger.info(f"[search_drive_files] Added mimeType filter: '{mime}'")
list_params = build_drive_list_params(
query=final_query,
page_size=page_size,
@@ -414,6 +424,7 @@ async def list_drive_items(
drive_id: Optional[str] = None,
include_items_from_all_drives: bool = True,
corpora: Optional[str] = None,
file_type: Optional[str] = None,
) -> str:
"""
Lists files and folders, supporting shared drives.
@@ -427,17 +438,27 @@ async def list_drive_items(
drive_id (Optional[str]): ID of the shared drive. If provided, the listing is scoped to this drive.
include_items_from_all_drives (bool): Whether items from all accessible shared drives should be included if `drive_id` is not set. Defaults to True.
corpora (Optional[str]): Corpus to query ('user', 'drive', 'allDrives'). If `drive_id` is set and `corpora` is None, 'drive' is used. If None and no `drive_id`, API defaults apply.
file_type (Optional[str]): Restrict results to a specific file type. Accepts a friendly
name ('folder', 'document', 'doc', 'spreadsheet', 'sheet',
'presentation', 'slides', 'form', 'drawing', 'pdf',
'shortcut') or any raw MIME type string
(e.g. 'application/pdf'). Defaults to None (all types).
Returns:
str: A formatted list of files/folders in the specified folder.
"""
logger.info(
f"[list_drive_items] Invoked. Email: '{user_google_email}', Folder ID: '{folder_id}'"
f"[list_drive_items] Invoked. Email: '{user_google_email}', Folder ID: '{folder_id}', File Type: '{file_type}'"
)
resolved_folder_id = await resolve_folder_id(service, folder_id)
final_query = f"'{resolved_folder_id}' in parents and trashed=false"
if file_type is not None:
mime = resolve_file_type_mime(file_type)
final_query += f" and mimeType = '{mime}'"
logger.info(f"[list_drive_items] Added mimeType filter: '{mime}'")
list_params = build_drive_list_params(
query=final_query,
page_size=page_size,