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

consolidate the retry mechanism into the handle_http_errors decorator, mitigates idempotency risks by limiting retries to read-only operations

Browse Source
This commit is contained in:
Taylor Wilsdon
2025-07-17 13:57:21 -04:00
parent 181ffb650d
commit f08373a8b2
9 changed files with 116 additions and 165 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
gdrive/drive_tools.py
View File

@@ -15,7 +15,7 @@ import io
import httpx
from auth.service_decorator import require_google_service
from core.utils import extract_office_xml_text, handle_http_errors, retry_on_ssl_error
from core.utils import extract_office_xml_text, handle_http_errors
from core.server import server
logger = logging.getLogger(__name__)
@@ -76,9 +76,8 @@ def _build_drive_list_params(
return list_params
@server.tool()
@handle_http_errors("search_drive_files", is_read_only=True)
@require_google_service("drive", "drive_read")
@handle_http_errors("search_drive_files")
@retry_on_ssl_error()
async def search_drive_files(
service,
user_google_email: str,
@@ -144,9 +143,8 @@ async def search_drive_files(
return text_output
@server.tool()
@handle_http_errors("get_drive_file_content", is_read_only=True)
@require_google_service("drive", "drive_read")
@handle_http_errors("get_drive_file_content")
@retry_on_ssl_error()
async def get_drive_file_content(
service,
user_google_email: str,
@@ -202,7 +200,7 @@ async def get_drive_file_content(
"application/vnd.openxmlformats-officedocument.presentationml.presentation",
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
}
if mime_type in office_mime_types:
office_text = extract_office_xml_text(file_content_bytes, mime_type)
if office_text:
@@ -235,9 +233,8 @@ async def get_drive_file_content(
@server.tool()
@handle_http_errors("list_drive_items", is_read_only=True)
@require_google_service("drive", "drive_read")
@handle_http_errors("list_drive_items")
@retry_on_ssl_error()
async def list_drive_items(
service,
user_google_email: str,
@@ -292,9 +289,8 @@ async def list_drive_items(
return text_output
@server.tool()
@require_google_service("drive", "drive_file")
@handle_http_errors("create_drive_file")
@retry_on_ssl_error()
@require_google_service("drive", "drive_file")
async def create_drive_file(
service,
user_google_email: str,